server/base/models/user_model.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@File    :   user_model.py
@Time    :   2024/08/31
@Project :   https://github.com/PeterH0323/Streamer-Sales
@Author  :   HinGwenWong
@Version :   1.0
@Desc    :   用户信息数据结构
"""

from datetime import datetime
from ipaddress import IPv4Address
from pydantic import BaseModel
from sqlmodel import Field, SQLModel


# =======================================================
#                      基本模型
# =======================================================
class TokenItem(BaseModel):
    access_token: str
    token_type: str


class UserBaseInfo(BaseModel):
    user_id: int | None = Field(default=None, primary_key=True, unique=True)
    username: str = Field(index=True, unique=True)
    email: str | None = None
    avatar: str | None = None
    create_time: datetime = datetime.now()


# =======================================================
#                      数据库模型
# =======================================================
class UserInfo(UserBaseInfo, SQLModel, table=True):

    __tablename__ = "user_info"

    hashed_password: str
    ip_address: IPv4Address | None = None
    delete: bool = False