config.py 871 B

123456789101112131415161718192021
  1. import os
  2. import sys
  3. sys.path.append(os.path.dirname(os.path.dirname(__file__)))
  4. from sqlalchemy import UniqueConstraint
  5. from loguru import logger
  6. from sqlmodel import Field, SQLModel
  7. # 定义数据库模型,不推荐使用 __tablename__ = '"UserInfo"' 来定义包含大写的表名字,
  8. # 因为可能会导致与其他数据库系统不兼容,而且表查询的时候需要额外注意表格名的大小写
  9. class UserOAuthToken(SQLModel):
  10. access_token:str = Field(default=None, primary_key=True)
  11. refresh_expires_in:str = Field(default=None, )
  12. refresh_token:str = Field(default=1, )
  13. __table_args__ = (UniqueConstraint('open_id', 'name', name='uq_open_id_name'),)
  14. user = UserOAuthToken()
  15. HOST = '::'
  16. con:UniqueConstraint = [col.name for col in UserOAuthToken.__table__.constraints['uq_open_id_name'].columns]
  17. logger.info(con)
  18. # print(HOST)