Flask 项目使用的配置文件是 config.ini ,示例:
[flask]
FLASK_APP=run.py
FLASK_ENV=produce
FLASK_PORT=8000
SECRET_KEY=''
[local]
DEBUG=1
DATABASE_URL=mysql://root:[email protected]:3306/database_local
SQLALCHEMY_POOL_RECYCLE=3600
[test]
DEBUG=1
DATABASE_URL=mysql://root:[email protected]:3306/database_test
SQLALCHEMY_POOL_RECYCLE=3600
[produce]
DEBUG=0
DATABASE_URL=mysql://root:[email protected]:3306/database_produce
SQLALCHEMY_POOL_RECYCLE=600
flask-sqlalchemy 3.1 版本 SQLALCHEMY_POOL_RECYCLE 配置被移除了,变成如下方式:
SQLALCHEMY_ENGINE_OPTIONS = {
"pool_recycle": 600,
}
目前的解决方式是写在读取配置文件的类里面:
import configparser
import os
cf = configparser.ConfigParser()
config_dir = os.path.dirname(os.path.abspath(__file__))
cf.read(os.path.join(config_dir, "config.ini"))
class Config:
# Flask
FLASK_ENV = cf.get("flask", "FLASK_ENV")
SECRET_KEY = cf.get("flask", "SECRET_KEY")
FLASK_PORT = cf.getint("flask", "FLASK_PORT")
DEBUG = True if cf.getint(FLASK_ENV, "DEBUG") else False
# SQLALCHEMY
SQLALCHEMY_DATABASE_URI = cf.get(FLASK_ENV, "DATABASE_URL")
SQLALCHEMY_ENGINE_OPTIONS = {
"pool_recycle": 600
} # 这样就无法区分 local 、test 、produce 环境
但是,这样写会导致一个问题——local 、test 、produce 三个环境都使用了相同的 pool_recycle ,不满足需求。请问在 config.ini 里面如何根据不同的环境配置不同的 SQLALCHEMY_ENGINE_OPTIONS(或 pool_recycle)?
请各位大佬指点指点,谢谢。
1
nulIptr 278 天前 1
如果是我的话可能会把配置文件整个都改成 json 或者 yaml
想改动小的话不改配置文件,读文件的时候做个映射不得了 比如 SQLALCHEMY_ENGINE_OPTIONS = { "pool_recycle": cf.getint(FLASK_ENV, "SQLALCHEMY_POOL_RECYCLE") } |
2
linauror 278 天前 1
按 1 的配置写法,这样直接去取不行吗?把代码中的 600 直接替换为 cf.get(FLASK_ENV, "SQLALCHEMY_POOL_RECYCLE")
|
4
codists OP 感谢两位大佬的解答。
|
5
lisongeee 278 天前 1
能不能将 ini 后缀改成 toml, 然后使用 tomlkit 模块读取
然后只需要把 SQLALCHEMY_ENGINE_OPTIONS={"pool_recycle": 600,} 正常放在 SQLALCHEMY_POOL_RECYCLE=3600 的下面一行就行 |
6
lisongeee 278 天前
好像还得把原来的字符串类型如 produce 和 mysql://root 加上引号才行
|
7
LeeYD 277 天前 via Android 1
config.py
```python import os class Config: SECRET_KEY = os.environ.get('SECRET_KEY', '') SQLALCHEMY_ENGINE_OPTIONS = { "pool_recycle": 600 } class LocalConfig(Config): DEBUG = True DATABASE_URL = 'mysql://root:[email protected]:3306/database_local' SQLALCHEMY_POOL_RECYCLE = 3600 class TestConfig(Config): DEBUG = True DATABASE_URL = 'mysql://root:[email protected]:3306/database_test' SQLALCHEMY_POOL_RECYCLE = 3600 class ProductionConfig(Config): DEBUG = False DATABASE_URL = 'mysql://root:[email protected]:3306/database_produce' SQLALCHEMY_POOL_RECYCLE = 600 ``` run.py ```python from flask import Flask from config import LocalConfig, TestConfig, ProductionConfig app = Flask(__name__) env = 'production' if env == 'local': app.config.from_object(LocalConfig) elif env == 'test': app.config.from_object(TestConfig) elif env == 'production': app.config.from_object(ProductionConfig) ``` |