添加注册登录功能

test_template_selection.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+"""
+测试模板选择功能
+"""
+from src.flask_prompt_master import create_app, db
+from src.flask_prompt_master.models.models import PromptTemplate
+
+def test_template_selection():
+    """测试模板选择功能"""
+    app = create_app()
+    
+    with app.app_context():
+        print("=" * 50)
+        print("模板选择功能测试")
+        print("=" * 50)
+        
+        try:
+            # 获取所有模板
+            templates = PromptTemplate.query.all()
+            print(f"📋 总模板数: {len(templates)}")
+            
+            if not templates:
+                print("❌ 没有找到任何模板")
+                return
+            
+            # 显示模板信息
+            print("\n📝 模板列表:")
+            for i, template in enumerate(templates, 1):
+                print(f"   {i}. ID: {template.id}")
+                print(f"      名称: {template.name}")
+                print(f"      分类: {template.category}")
+                print(f"      是否默认: {template.is_default}")
+                print()
+            
+            # 查找默认模板
+            default_template = PromptTemplate.query.filter_by(is_default=True).first()
+            if default_template:
+                print(f"✅ 找到默认模板: {default_template.name} (ID: {default_template.id})")
+            else:
+                print("⚠️  没有找到默认模板")
+            
+            # 查找第一个模板
+            first_template = templates[0]
+            print(f"✅ 第一个模板: {first_template.name} (ID: {first_template.id})")
+            
+            print("\n" + "=" * 50)
+            print("✅ 模板选择功能测试完成！")
+            print("=" * 50)
+            
+        except Exception as e:
+            print(f"❌ 测试过程中出现错误: {str(e)}")
+
+if __name__ == '__main__':
+    test_template_selection()