增加诗词收藏功能

create_poetry_favorites_table.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+创建古诗词收藏表
+"""
+
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from src.flask_prompt_master import create_app, db
+from src.flask_prompt_master.models.poetry_favorites import PoetryFavorite
+
+def create_poetry_favorites_table():
+    """创建古诗词收藏表"""
+    app = create_app()
+    
+    with app.app_context():
+        try:
+            # 创建表
+            db.create_all()
+            print("✅ 古诗词收藏表创建成功！")
+            
+            # 检查表是否存在
+            from sqlalchemy import inspect
+            inspector = inspect(db.engine)
+            if 'poetry_favorites' in inspector.get_table_names():
+                print("✅ 表 'poetry_favorites' 已存在")
+            else:
+                print("❌ 表 'poetry_favorites' 创建失败")
+                
+        except Exception as e:
+            print(f"❌ 创建表失败: {e}")
+            return False
+    
+    return True
+
+if __name__ == '__main__':
+    print("开始创建古诗词收藏表...")
+    success = create_poetry_favorites_table()
+    if success:
+        print("🎉 古诗词收藏功能数据库初始化完成！")
+    else:
+        print("💥 数据库初始化失败！")
+        sys.exit(1)