mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-07 00:16:57 +08:00
refactor(db): 修复异步连接池无界增长,并完成 SQLAlchemy 2.0 迁移与分层归位 (#6320)
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from sqlalchemy import Column, String, select
|
||||
from typing import Optional
|
||||
from sqlalchemy import String, select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy.orm import Mapped, Session, mapped_column
|
||||
|
||||
from app.db import db_query, Base, get_id_column, async_db_query
|
||||
|
||||
@@ -11,18 +12,18 @@ class SiteIcon(Base):
|
||||
"""
|
||||
id = get_id_column()
|
||||
# 站点名称
|
||||
name = Column(String, nullable=False)
|
||||
name: Mapped[str] = mapped_column(String, nullable=False)
|
||||
# 域名Key
|
||||
domain = Column(String, index=True)
|
||||
domain: Mapped[Optional[str]] = mapped_column(String, index=True)
|
||||
# 图标地址
|
||||
url = Column(String, nullable=False)
|
||||
url: Mapped[str] = mapped_column(String, nullable=False)
|
||||
# 图标Base64
|
||||
base64 = Column(String)
|
||||
base64: Mapped[Optional[str]] = mapped_column(String)
|
||||
|
||||
@classmethod
|
||||
@db_query
|
||||
def get_by_domain(cls, db: Session, domain: str):
|
||||
return db.query(cls).filter(cls.domain == domain).first()
|
||||
return db.execute(select(cls).where(cls.domain == domain)).scalars().first()
|
||||
|
||||
@classmethod
|
||||
@async_db_query
|
||||
|
||||
Reference in New Issue
Block a user