"""
初始化测试数据 - 创建测试用户和任务类别
"""
from storage.database.supabase_client import get_supabase_client

def init_test_data():
    """初始化测试数据"""
    client = get_supabase_client()
    
    # 创建测试用户
    print("创建测试用户...")
    user_data = {
        'name': '张三',
        'phone': '13800138000',
        'room_number': 'A栋1001',
        'user_type': 'owner',
        'is_active': True
    }
    
    result = client.table('users').insert(user_data).execute()
    if result.data:
        user_id = result.data[0]['id']
        print(f"✅ 测试用户创建成功，ID: {user_id}")
    else:
        print("❌ 用户创建失败")
        return
    
    # 创建任务类别
    print("\n创建任务类别...")
    categories = [
        {'name': '快递代取', 'description': '帮忙取快递、拿外卖', 'icon': 'package', 'priority': 1, 'keywords': ['快递', '取快递', '拿快递', '外卖', '取外卖']},
        {'name': '车辆挪移', 'description': '挪车、车位咨询', 'icon': 'car', 'priority': 2, 'keywords': ['挪车', '移车', '车位', '停车']},
        {'name': '设施报修', 'description': '电梯故障、水电维修、公共设施损坏', 'icon': 'repair', 'priority': 3, 'keywords': ['报修', '维修', '电梯', '水电', '故障', '损坏']},
        {'name': '清洁服务', 'description': '公共区域清洁、垃圾处理', 'icon': 'clean', 'priority': 4, 'keywords': ['清洁', '打扫', '垃圾', '卫生']},
        {'name': '安全服务', 'description': '可疑人员、噪音投诉', 'icon': 'security', 'priority': 5, 'keywords': ['安全', '可疑', '噪音', '投诉']},
        {'name': '其他咨询', 'description': '物业费查询、入住办理等', 'icon': 'info', 'priority': 6, 'keywords': ['咨询', '物业费', '入住']}
    ]
    
    for category in categories:
        result = client.table('task_categories').insert(category).execute()
        if result.data:
            print(f"✅ 类别 '{category['name']}' 创建成功")
        else:
            print(f"❌ 类别 '{category['name']}' 创建失败")
    
    print("\n初始化完成！")
    return user_id

if __name__ == "__main__":
    init_test_data()