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

Question

I'm trying to specify a type hinting for every function in my code. How I can specify the type hinting for a function which waiting for any pydantic schema (model)?

Here is my code:

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)

Example model:

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

Answer 1

You could import your schema from the module: from app.schemas import UserBase . And then type hint this object to your function:

def your_function() -> UserBase:
    ...

If I understood correctly, this is what you wanted to do. If it does not fit on your needs, let me know!