fastapi -从main.py导入配置
fastapi -从main.py导入配置
提问于 2020-07-08 01:21:33
我刚接触fastapi，到目前为止，它真的很棒，但我很难找到一种干净的方法来将我的应用配置导入到另一个模块中。
编辑:我需要能够在运行单元测试时更改配置
下面是我的dir树：
/app
| __init__.py
| /router
| | __init__.py
| | my_router.py
| /test
| | test_api.py
| config.py
| main.py这是我的main.py文件：
from functools import lru_cache
from fastapi import FastAPI
from .router import my_router
from . import config
app = FastAPI()
app.include_router(
my_router.router,
prefix="/r",
tags=["my-router"],
)
@lru_cache()
def get_setting():
return config.Settings(admin_email="admin@domain.com")
@app.get('/')
def hello():
return 'Hello world'这是router.py
from fastapi import APIRouter
from ..main import get_setting
router = APIRouter()
@router.get('/test')
def get_param_list(user_id: int):
config = get_setting()
return 'Import Ok'下面是配置文件
from pydantic import BaseSettings
class Settings(BaseSettings):
param_folder: str = "param"
result_folder: str = "output"
class Config:
env_prefix = "APP_"然后运行uvicorn app.main:app --reload，我得到了：ERROR: Error loading ASGI app. Could not import module "app.main".，我猜是因为一种循环导入。但是我不知道如何将我的配置传递给我的路由器？
感谢您的帮助:)
回答 2
Stack Overflow用户
回答已采纳
发布于 2020-07-08 03:05:40
直接在config.py中设置lru缓存怎么样？
from functools import lru_cache
from pydantic import BaseSettings
class Settings(BaseSettings):
admin_email: str = "admin@example.com"
param_folder: str = "param"
result_folder: str = "output"
class Config:
env_prefix = "APP_"
@lru_cache()
def get_setting():
return Settings()和my_router.py
from fastapi import APIRouter, Depends
from ..config import Settings, get_setting
router = APIRouter()
@router.get('/test')
def get_param_list(config: Settings = Depends(get_setting)):
return config和test.py
from fastapi.testclient import TestClient
from . import config, main
client = TestClient(main.app)
def get_settings_override():
return config.Settings(admin_email="testing_admin@example.com")
main.app.dependency_overrides[config.get_settings] = get_settings_override
def test_app():
response = client.get("/r/test")
data = response.json()
assert data == config.Settings(admin_email="testing_admin@example.com")Stack Overflow用户
发布于 2020-07-08 19:39:45
我使用FastAPI依赖系统，并按照@Kassym Dorsel的建议，通过将lru_cache移动到config.py来使其正常工作。
唯一的缺点是，我必须将相当“繁重”的setting: config.Setting = Depends(config.get_setting)添加到每个需要该设置的函数调用中。
下面是我是如何做到的：
config.py文件：
from pydantic import BaseSettings
class Settings(BaseSettings):
param_folder: str = "param"
result_folder: str = "output"
class Config:
env_prefix = "APP_"
@lru_cache()
def get_setting():
return config.Settings(admin_email="admin@domain.com")main.py文件：
from functools import lru_cache
from fastapi import FastAPI
from .router import my_router
app = FastAPI()
app.include_router(
my_router.router,
prefix="/r",
tags=["my-router"],
)
@app.get('/')
def hello():
return 'Hello world'router.py文件：
from fastapi import APIRouter, Depends
from .. import config
router = APIRouter()
@router.get('/test')
def get_param_list(user_id: int, setting: config.Setting = Depends(config.get_setting)):
return setting这样，我就可以使用test_api.py中的dependency_overrides来更改测试的配置：
from fastapi.testclient import TestClient
from .. import config, server
client = TestClient(server.app)
TEST_PARAM_FOLDER = 'server/test/param'
TEST_RESULT_FOLDER = 'server/test/result'
def get_setting_override():
return config.Setting(param_folder=TEST_PARAM_FOLDER, result_folder=TEST_RESULT_FOLDER)
server.app.dependency_overrides[config.get_setting] = get_setting_override
def test_1():
...页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/62780646
复制相关文章
相似问题
腾讯云开发者
