39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
from sqlalchemy import Column, Integer, String, Sequence, UniqueConstraint, Index
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.db import db_query, db_update, Base
|
|
|
|
|
|
class UserConfig(Base):
|
|
"""
|
|
用户配置表
|
|
"""
|
|
id = Column(Integer, Sequence('id'), primary_key=True, index=True)
|
|
# 用户名
|
|
username = Column(String, index=True)
|
|
# 配置键
|
|
key = Column(String)
|
|
# 值
|
|
value = Column(String, nullable=True)
|
|
|
|
__table_args__ = (
|
|
# 用户名和配置键联合唯一
|
|
UniqueConstraint('username', 'key'),
|
|
Index('ix_userconfig_username_key', 'username', 'key'),
|
|
)
|
|
|
|
@staticmethod
|
|
@db_query
|
|
def get_by_key(db: Session, username: str, key: str):
|
|
return db.query(UserConfig) \
|
|
.filter(UserConfig.username == username) \
|
|
.filter(UserConfig.key == key) \
|
|
.first()
|
|
|
|
@db_update
|
|
def delete_by_key(self, db: Session, username: str, key: str):
|
|
userconfig = self.get_by_key(db=db, username=username, key=key)
|
|
if userconfig:
|
|
userconfig.delete(db=db, rid=userconfig.id)
|
|
return True
|