# -*- coding: utf-8 -*-
"""
检查user表的实际结构
"""
from src.flask_prompt_master import create_app, db
from sqlalchemy import inspect

def check_user_table():
    """检查user表结构"""
    app = create_app()
    
    with app.app_context():
        print("=" * 60)
        print("检查user表结构")
        print("=" * 60)
        
        try:
            # 获取表结构
            inspector = inspect(db.engine)
            columns = inspector.get_columns('user')
            
            print("📋 user表字段列表:")
            for column in columns:
                print("   {}: {}".format(column['name'], column['type']))
                if column.get('nullable') is False:
                    print("      (NOT NULL)")
                if column.get('primary_key'):
                    print("      (PRIMARY KEY)")
                if column.get('unique'):
                    print("      (UNIQUE)")
            
            # 检查是否有数据
            result = db.session.execute("SELECT COUNT(*) as count FROM user").fetchone()
            print("\n📊 user表记录数: {}".format(result[0]))
            
            if result[0] > 0:
                # 显示前几条记录
                print("\n📝 前3条记录:")
                records = db.session.execute("SELECT * FROM user LIMIT 3").fetchall()
                for i, record in enumerate(records, 1):
                    print("   记录{}: {}".format(i, dict(record)))
            
        except Exception as e:
            print("❌ 检查表结构失败: {}".format(str(e)))

if __name__ == '__main__':
    check_user_table()