fastapi - 從 main.py 導入配置

Question

我是 fastapi 的新手，到目前為止真的很棒，但我很難找到一種干凈的方法將我的應用程序配置導入另一個模塊。

編輯：我需要能夠在運行單元測試時更改配置

這是我的目錄樹：

/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". ERROR: Error loading ASGI app. Could not import module "app.main". 我猜是因為一種循環導入。 但是我不知道如何將我的配置傳遞給我的路由器？

謝謝你的幫助：）

Answer 1

直接在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")

Answer 2

我使用 FastAPI Dependency 系統讓它工作，並且按照@Kassym Dorsel 的建議，通過將 lru_cache 移動到config.py 。

唯一的缺點是我必須在每個需要設置的 function 調用中添加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():
    ...

Answer 3

唯一的缺點是我必須在每個需要設置的 function 調用中添加設置：config.Setting = Depends(config.get_setting)，這非常“繁重”。

您可以使用來自 fastapi_utils package 的基於Class 的視圖：

from fastapi import APIRouter, Depends
from fastapi_utils.cbv import cbv
from starlette import requests
from logging import Logger
from .. import config

router = APIRouter()

@cbv(router)
class MyQueryCBV:
    settings: config.Setting = Depends(config.get_setting)  # you can introduce settings dependency here

    def __init__(self, r: requests.Request):  # called for each query
        self.logger: Logger = self.settings.logger
        self.logger.warning(str(r.headers))

    @router.get('/test')
    def get_param_list(self, user_id: int)
        self.logger.warning(f"get_param_list: {user_id}")
        return self.settings

    @router.get("/test2")
    def get_param_list2(self):
        self.logger.warning(f"get_param_list2")
        return self.settings