mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 20:17:13 +08:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from typing import Any, Optional
|
|
from sqlalchemy import String, UniqueConstraint, JSON, select
|
|
from sqlalchemy.orm import Mapped, Session, mapped_column
|
|
|
|
from app.db.base import get_id_column, Base
|
|
|
|
|
|
class UserConfig(Base):
|
|
"""
|
|
用户配置表
|
|
"""
|
|
id = get_id_column()
|
|
# 用户名
|
|
username: Mapped[Optional[str]] = mapped_column(String)
|
|
# 配置键
|
|
key: Mapped[Optional[str]] = mapped_column(String)
|
|
# 值
|
|
value: Mapped[Optional[Any]] = mapped_column(JSON)
|
|
|
|
__table_args__ = (
|
|
# 用户名和配置键联合唯一
|
|
UniqueConstraint('username', 'key'),
|
|
)
|
|
|
|
@classmethod
|
|
def get_by_key(cls, db: Session, username: str, key: str):
|
|
"""在调用方 Session 中查询用户配置。"""
|
|
return db.execute(
|
|
select(cls).where(cls.username == username, cls.key == key)
|
|
).scalars().first()
|
|
|
|
def delete_by_key(self, db: Session, username: str, key: str):
|
|
"""在调用方持有的事务中暂存指定用户配置删除。"""
|
|
userconfig = self.get_by_key(db=db, username=username, key=key)
|
|
if userconfig:
|
|
db.delete(userconfig)
|
|
return True
|