fix plugins

app/db/init.py

@@ -1,3 +1,6 @@
+import importlib
+from pathlib import Path
+
 from alembic.command import upgrade
 from alembic.config import Config
 
@@ -13,6 +16,10 @@ def init_db():
     """
     初始化数据库
     """
+    # 导入模块，避免建表缺失
+    for module in Path(__file__).with_name("models").glob("*.py"):
+        importlib.import_module(f"app.db.models.{module.stem}")
+    # 全量建表
     Base.metadata.create_all(bind=Engine)
     # 初始化超级管理员
     _db = SessionLocal()

app/db/models/plugin.py

@@ -0,0 +1,22 @@
+from sqlalchemy import Column, Integer, String, Sequence
+from sqlalchemy.orm import Session
+
+from app.db.models import Base
+
+
+class PluginData(Base):
+    """
+    插件数据表
+    """
+    id = Column(Integer, Sequence('id'), primary_key=True, index=True)
+    plugin_id = Column(String, nullable=False, index=True)
+    key = Column(String, index=True, nullable=False)
+    value = Column(String)
+
+    @staticmethod
+    def get_plugin_data(db: Session, plugin_id: str):
+        return db.query(PluginData).filter(PluginData.plugin_id == plugin_id).all()
+
+    @staticmethod
+    def get_plugin_data_by_key(db: Session, plugin_id: str, key: str):
+        return db.query(PluginData.value).filter(PluginData.plugin_id == plugin_id, PluginData.key == key).first()

app/db/models/site.py

@@ -7,6 +7,9 @@ from app.db.models import Base
 
 
 class Site(Base):
+    """
+    站点表
+    """
     id = Column(Integer, Sequence('id'), primary_key=True, index=True)
     name = Column(String, nullable=False)
     domain = Column(String, index=True)

app/db/models/subscribe.py

@@ -5,6 +5,9 @@ from app.db.models import Base
 
 
 class Subscribe(Base):
+    """
+    订阅表
+    """
     id = Column(Integer, Sequence('id'), primary_key=True, index=True)
     name = Column(String, nullable=False, index=True)
     year = Column(String)

app/db/models/systemconfig.py

@@ -5,6 +5,9 @@ from app.db.models import Base
 
 
 class SystemConfig(Base):
+    """
+    配置表
+    """
     id = Column(Integer, Sequence('id'), primary_key=True, index=True)
     key = Column(String, index=True)
     value = Column(String, nullable=True)

app/db/models/user.py

@@ -6,6 +6,9 @@ from app.db.models import Base
 
 
 class User(Base):
+    """
+    用户表
+    """
     id = Column(Integer, Sequence('id'), primary_key=True, index=True)
     full_name = Column(String, index=True)
     email = Column(String, unique=True, index=True, nullable=False)