pydantic 模式/模型的 python 類型提示

Question

我試圖在我的代碼中為每個 function 指定一個類型提示。 如何為等待任何 pydantic 模式（模型）的 function 指定類型提示？

這是我的代碼：

def hash_password(password: str, schema = None) -> str:
    if schema:
        del schema.password if hasattr(schema, 'password') else None
    return config.pwd_context.hash(password)

示例 model：

my_schema = schemas.UserBase(full_name='a b c')  # just a text
type(my_schema)  # <class 'app.schemas.UserBase'>

Answer 1

您可以從模塊中導入您的架構： from app.schemas import UserBase 。 然后將這個 object 輸入到您的 function 中：

def your_function() -> UserBase:
    ...

如果我理解正確，這就是你想要做的。 如果它不符合您的需求，請告訴我！