#!/usr/bin/env python3
"""检查用户角色信息"""

import requests

def check_user_role():
    """检查用户角色"""
    base_url = "http://localhost:8001/api/v1"
    
    # 登录
    print("步骤1: 登录")
    login_data = {
        "username": "admin",
        "password": "admin123"
    }
    
    try:
        response = requests.post(f"{base_url}/users/login", json=login_data)
        print(f"状态码: {response.status_code}")
        
        if response.status_code != 200:
            print(f"登录失败: {response.text}")
            return
        
        data = response.json()
        access_token = data.get('access_token')
        print(f"登录成功!")
        
        # 获取用户信息
        print("\n步骤2: 获取用户信息")
        headers = {"Authorization": f"Bearer {access_token}"}
        user_response = requests.get(f"{base_url}/users/me", headers=headers)
        print(f"状态码: {user_response.status_code}")
        
        if user_response.status_code == 200:
            user_data = user_response.json()
            print(f"\n用户信息:")
            print(f"  用户名: {user_data.get('username', 'N/A')}")
            print(f"  邮箱: {user_data.get('email', 'N/A')}")
            print(f"  角色ID: {user_data.get('role_id', 'N/A')}")
            print(f"  角色名称: {user_data.get('role_name', 'N/A')}")
            print(f"  角色对象: {user_data.get('role', 'N/A')}")
            
            # 检查是否是管理员
            role_name = user_data.get('role_name')
            if role_name == 'admin':
                print(f"\n✅ 用户是管理员，应该显示后台管理页面")
            else:
                print(f"\n❌ 用户不是管理员，角色名称是: {role_name}")
        else:
            print(f"获取用户信息失败: {user_response.text}")
            
    except Exception as e:
        print(f"错误: {e}")

if __name__ == "__main__":
    check_user_role()