From 60996be71bcf03f4f3347984eb956f5540f79f66 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Thu, 16 Apr 2026 17:05:57 +0800
Subject: [PATCH] fix local db initialization model registration

---
 app/db/init.py            | 3 +++
 app/db/models/__init__.py | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/app/db/init.py b/app/db/init.py
index 46db4c11..66a21733 100644
--- a/app/db/init.py
+++ b/app/db/init.py
@@ -10,6 +10,9 @@ def init_db():
     """
     初始化数据库
     """
+    # 确保所有模型都已注册到 Base.metadata 中
+    import app.db.models  # noqa: F401
+
     # 全量建表
     Base.metadata.create_all(bind=Engine) # noqa
 
diff --git a/app/db/models/__init__.py b/app/db/models/__init__.py
index a7bb7cd8..9f772c02 100644
--- a/app/db/models/__init__.py
+++ b/app/db/models/__init__.py
@@ -1,10 +1,14 @@
 from .downloadhistory import DownloadHistory, DownloadFiles
 from .mediaserver import MediaServerItem
+from .message import Message
 from .passkey import PassKey
 from .plugindata import PluginData
 from .site import Site
 from .siteicon import SiteIcon
+from .sitestatistic import SiteStatistic
+from .siteuserdata import SiteUserData
 from .subscribe import Subscribe
+from .subscribehistory import SubscribeHistory
 from .systemconfig import SystemConfig
 from .transferhistory import TransferHistory
 from .user import User