import requests
import json

# 创建会话以保持登录状态
session = requests.Session()

def login():
    """登录系统"""
    url = "http://localhost:5000/api/v1/auth/login"
    headers = {"Content-Type": "application/json"}
    data = {
        "username": "admin",
        "password": "admin123"
    }
    
    try:
        response = session.post(url, headers=headers, data=json.dumps(data))
        print(f"Login Status Code: {response.status_code}")
        print(f"Login Response: {response.text}")
        return response.status_code == 200
    except Exception as e:
        print(f"Login Error: {e}")
        return False

def test_create_product():
    """测试创建产品API"""
    url = "http://localhost:5000/api/v1/products"
    headers = {"Content-Type": "application/json"}
    data = {
        "product_name": "测试产品",
        "description": "这是一个测试产品"
    }
    
    try:
        response = session.post(url, headers=headers, data=json.dumps(data))
        print(f"Create Product Status Code: {response.status_code}")
        print(f"Create Product Response: {response.text}")
        return response.status_code == 200
    except Exception as e:
        print(f"Create Product Error: {e}")
        return False

def test_get_logs():
    """测试获取日志API"""
    url = "http://localhost:5000/api/v1/logs"
    try:
        response = session.get(url)
        print(f"Get Logs Status Code: {response.status_code}")
        print(f"Get Logs Response: {response.text}")
        return response.status_code == 200
    except Exception as e:
        print(f"Get Logs Error: {e}")
        return False

if __name__ == "__main__":
    # 登录
    if login():
        print("登录成功")
        # 测试创建产品
        if test_create_product():
            print("创建产品成功")
            # 测试获取日志
            test_get_logs()
        else:
            print("创建产品失败")
    else:
        print("登录失败")