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)
|
|
# 用户ID
|
|
user_id = Column(Integer, index=True)
|
|
# 配置键
|
|
key = Column(String)
|
|
# 值
|
|
value = Column(String, nullable=True)
|
|
|
|
__table_args__ = (
|
|
# 用户ID和配置键联合唯一
|
|
UniqueConstraint('user_id', 'key'),
|
|
Index('ix_userconfig_userid_key', 'user_id', 'key'),
|
|
)
|
|
|
|
@staticmethod
|
|
@db_query
|
|
def get_by_key(db: Session, user_id: int, key: str):
|
|
return db.query(UserConfig) \
|
|
.filter(UserConfig.user_id == user_id) \
|
|
.filter(UserConfig.key == key) \
|
|
.first()
|
|
|
|
@db_update
|
|
def delete_by_key(self, db: Session, user_id: int, key: str):
|
|
userconfig = self.get_by_key(db, user_id, key)
|
|
if userconfig:
|
|
userconfig.delete(db, userconfig.id)
|
|
return True
|