# -*- 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()