FastAPI 读取不同环境的配置文件

配置文件

config-dev.ini

# -*- coding: utf-8 -*-
# 数据库配置
[pgsql]
host=192.168.1.2
port=5432
dbname=test
user=root
passwd=xxxxxxxx
 
# Redis配置
[redis]
host=192.168.1.2
port=6379
password=xxxxxxxxx
db=0

读取配置

global_config.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
# @time    : 2023/2/3 14:00
# @author  : pugongying
# @description : 全局参数配置
 
import configparser
import os
import sys
from functools import lru_cache
 
# 设置 目录在项目跟目录下
cache_dir_name = 'cache'
base_path = os.path.dirname(__file__)
idx = base_path.index('app' + os.sep + 'config')
work_dir = base_path[:idx]
 
params = sys.argv
 
 
class ConfigSetting(object):
 
    def __init__(self, config_file='config-dev.ini'):
        self._path = os.path.join(work_dir + 'config', config_file)
        if not os.path.exists(self._path):
            raise FileNotFoundError("No such file: %s" % config_file)
        self._config = configparser.ConfigParser()
        self._config.read(self._path, encoding='utf-8')
 
    def get_work_dir(self):
        return work_dir
 
    def get_cache_dir(self):
        cache_dir_path = work_dir + cache_dir_name
        if not os.path.exists(cache_dir_path):
            # 目录不存在，创建目录
            os.mkdir(cache_dir_path)
        return cache_dir_path
 
    def get(self, section, name, strip_blank=True, strip_quote=True):
        s = self._config.get(section, name)
        if strip_blank:
            s = s.strip()
        if strip_quote:
            s = s.strip('"').strip("'")
 
        return s
 
    def getboolean(self, section, name):
        return self._config.getboolean(section, name)
 
    def getint(self, section, name):
        return self._config.getint(section, name)
 
 
class GlobalConfig(object):
    CONFIG_SETTING = ConfigSetting('config-%s.ini' % (
        params[1] if len(params) > 1 and params[1] in ['dev', 'test', 'pre', 'prod', 'local'] else 'dev'))
 
    # 工作目录
    WORK_DIR = CONFIG_SETTING.get_work_dir()
    # 缓存目录
    CACHE_DIR = CONFIG_SETTING.get_cache_dir()
 
    # PgSQL配置
    PgSQL_HOST = CONFIG_SETTING.get('pgsql', 'host')
    PgSQL_PORT = CONFIG_SETTING.get('pgsql', 'port')
    PGSQL_DBNAME = CONFIG_SETTING.get('pgsql', 'dbname')
    PGSQL_USER = CONFIG_SETTING.get('pgsql', 'user')
    PGSQL_PASSWD = CONFIG_SETTING.get('pgsql', 'passwd').replace('@', '%40') # 处理密码中有“@”特殊字符
 
    # Redis配置
    REDIS_HOST = CONFIG_SETTING.get('redis', 'host')
    REDIS_PORT = CONFIG_SETTING.get('redis', 'port')
    REDIS_PASSWORD = CONFIG_SETTING.get('redis', 'password')
    REDIS_DB = CONFIG_SETTING.getint('redis', 'db')
 
 
@lru_cache()
def get_configs():
    return GlobalConfig()
 
 
configs = get_configs()

使用配置

database.py

from app.config import configs
 
# pgsql 数据库url
SQLALCHEMY_DATABASE_URL = "postgresql://{}:{}@{}:{}/{}".format(
    configs.PGSQL_USER,
    configs.PGSQL_PASSWD,
    configs.PgSQL_HOST,
    configs.PgSQL_PORT,
    configs.PGSQL_DBNAME
)  
 
print("SQLALCHEMY_DATABASE_URL=", SQLALCHEMY_DATABASE_URL)
 
# 数据库迁移配置
engine = create_engine(SQLALCHEMY_DATABASE_URL, pool_pre_ping=True, pool_size=50, pool_timeout=30, pool_recycle=1)
SessionLocal = sessionmaker(autocommit=False, autoflush=True, bind=engine)
session = scoped_session(SessionLocal)
 
Base = declarative_base()
 
 
def get_db():
    db = session
    try:
        yield db
    finally:
        db.remove()