123 lines
6.1 KiB
Python
123 lines
6.1 KiB
Python
from sqlalchemy import Column, Integer, Text, String, Float, DateTime, ForeignKey, func
|
||
from sqlalchemy.ext.declarative import declarative_base
|
||
from sqlalchemy.orm import relationship, Mapped
|
||
from typing import List, Optional
|
||
|
||
# 基础模型基类（如果已有全局 Base 可直接复用）
|
||
ResourceBase = declarative_base()
|
||
|
||
class U3C3(ResourceBase):
|
||
"""对应 sqlite 中的 u3c3 表"""
|
||
__tablename__ = "u3c3" # 表名与原表保持一致
|
||
|
||
# 字段定义（严格映射原表结构）
|
||
id = Column(Integer, primary_key=True, autoincrement=True, comment="主键ID")
|
||
sites = Column(Text, comment="站点名称")
|
||
category = Column(Text, comment="分类")
|
||
title = Column(Text, comment="标题")
|
||
# 关键修改：给 String 加长度（如 512，根据实际链接长度调整）
|
||
url = Column(String(512), unique=True, comment="资源链接（唯一）")
|
||
torrent_url = Column(Text, comment="种子链接")
|
||
magnet_url = Column(Text, comment="磁力链接")
|
||
size_text = Column(Text, comment="大小文本描述")
|
||
size_gb = Column(Float, comment="大小（GB）")
|
||
update_date = Column(Text, comment="更新日期")
|
||
# 补充：MySQL 中建议用 func.now() 替代 func.datetime(...)，兼容性更好
|
||
created_at = Column(DateTime, default=func.now(), comment="创建时间（本地时间）")
|
||
updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间（本地时间）")
|
||
|
||
def __repr__(self):
|
||
"""打印实例时显示的信息"""
|
||
return f"<U3c3(id={self.id}, title='{self.title[:20]}...')>"
|
||
|
||
class Sis(ResourceBase):
|
||
"""对应 MySQL 中的 sis 表"""
|
||
__tablename__ = "sis" # 表名与原表保持一致
|
||
|
||
# 字段定义（严格映射原表结构）
|
||
id = Column(Integer, primary_key=True, autoincrement=True, comment="主键ID")
|
||
plate_name = Column(Text, comment="板块名称")
|
||
title = Column(Text, comment="标题")
|
||
url = Column(String(512), unique=True, comment="资源链接（唯一）")
|
||
size_text = Column(Text, comment="大小文本描述")
|
||
size_gb = Column(Float, comment="大小（GB）")
|
||
update_date = Column(Text, comment="更新日期")
|
||
# 补充：MySQL 中建议用 func.now() 替代 func.datetime(...)，兼容性更好
|
||
created_at = Column(DateTime, default=func.now(), comment="创建时间（本地时间）")
|
||
updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间（本地时间）")
|
||
|
||
def __repr__(self):
|
||
return f"<Sis(id={self.id}, title='{self.title[:20]}...')>"
|
||
|
||
|
||
class ClmIndex(ResourceBase):
|
||
"""对应 MySQL 中的 clm_index 表"""
|
||
__tablename__ = "clm_index" # 表名与原表保持一致
|
||
|
||
# 字段定义（严格映射原表结构）
|
||
id = Column(Integer, primary_key=True, autoincrement=True, comment="主键ID")
|
||
category = Column(Text, comment="分类")
|
||
title = Column(Text, comment="标题")
|
||
href = Column(String(512), unique=True, comment="资源链接（唯一）")
|
||
magnet_href = Column(Text, comment="磁力链接")
|
||
size_text = Column(Text, comment="大小文本描述")
|
||
size_gb = Column(Float, comment="大小（GB）")
|
||
heat = Column(Integer, default=0, comment="热度")
|
||
add_date = Column(Text, comment="添加日期")
|
||
last_down_date = Column(Text, comment="最后下载日期")
|
||
created_at = Column(DateTime, default=func.now(), comment="创建时间（本地时间）")
|
||
updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间（本地时间）")
|
||
|
||
# 关系定义：用 Mapped[List["ClmKeywordsIndex"]] 替代 List["ClmKeywordsIndex"]
|
||
clm_keywords_index: Mapped[List["ClmKeywordsIndex"]] = relationship(
|
||
"ClmKeywordsIndex", back_populates="index"
|
||
)
|
||
|
||
|
||
def __repr__(self):
|
||
return f"<ClmIndex(id={self.id}, title='{self.title[:20]}...')>"
|
||
|
||
|
||
class ClmKeywords(ResourceBase):
|
||
"""对应 MySQL 中的 clm_keywords 表"""
|
||
__tablename__ = "clm_keywords" # 表名与原表保持一致
|
||
|
||
# 字段定义（严格映射原表结构）
|
||
id = Column(Integer, primary_key=True, autoincrement=True, comment="主键ID")
|
||
words = Column(String(512), unique=True, comment="关键词（唯一）")
|
||
groups = Column(Text, comment="关键词分组")
|
||
tags = Column(Text, comment="标签")
|
||
index_count = Column(Integer, default=0, comment="关联索引数量")
|
||
created_at = Column(DateTime, default=func.now(), comment="创建时间（本地时间）")
|
||
updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间（本地时间）")
|
||
|
||
# 关系定义：用 Mapped[List["ClmKeywordsIndex"]] 替代 List["ClmKeywordsIndex"]
|
||
clm_keywords_index: Mapped[List["ClmKeywordsIndex"]] = relationship(
|
||
"ClmKeywordsIndex", back_populates="words"
|
||
)
|
||
|
||
def __repr__(self):
|
||
return f"<ClmKeywords(id={self.id}, words='{self.words[:20]}...')>"
|
||
|
||
|
||
|
||
class ClmKeywordsIndex(ResourceBase):
|
||
"""对应 MySQL 中的 clm_keywords_index 表（关联表）"""
|
||
__tablename__ = "clm_keywords_index" # 表名与原表保持一致
|
||
|
||
# 字段定义（严格映射原表结构）
|
||
id = Column(Integer, primary_key=True, autoincrement=True, comment="主键ID")
|
||
words_id = Column(Integer, ForeignKey("clm_keywords.id"), comment="关键词ID（外键）")
|
||
index_id = Column(Integer, ForeignKey("clm_index.id"), comment="索引ID（外键）")
|
||
wid_iid = Column(String(255), unique=True, comment="关键词与索引的关联标识")
|
||
tags = Column(Text, comment="关联标签")
|
||
created_at = Column(DateTime, default=func.now(), comment="创建时间（本地时间）")
|
||
updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间（本地时间）")
|
||
|
||
# 关系定义：用 Mapped 包装单个对象类型
|
||
index: Mapped["ClmIndex"] = relationship("ClmIndex", back_populates="clm_keywords_index")
|
||
words: Mapped["ClmKeywords"] = relationship("ClmKeywords", back_populates="clm_keywords_index")
|
||
|
||
def __repr__(self):
|
||
return f"<ClmKeywordsIndex(id={self.id}, words_id={self.words_id}, index_id={self.index_id})>"
|