import pandas as pd
import requests
import json
from datetime import datetime

# Excel 文件路径
EXCEL_FILE = "/home/xsl/code/xsl_node/docs/example.xls"

# API 端点
API_URL = "http://localhost:8000/api/projects"

# xsl 用户的 token
TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJ4c2wiLCJleHAiOjE3Njk0MjIxOTh9.5f_hiybhIRl-biiLETczZwLr9E-LcAxhi57Cc7ikEyI"

# 读取 Excel 文件，跳过前 4 行表头
df = pd.read_excel(EXCEL_FILE, skiprows=4)

# 处理数据并导入项目
for index, row in df.iterrows():
    # 跳过空行
    if pd.isna(row.iloc[0]):
        continue
    
    # 构建项目数据
    project_data = {
        "name": str(row.iloc[2]) if pd.notna(row.iloc[2]) else f"Project {index+1}",
        "description": str(row.iloc[3]) if pd.notna(row.iloc[3]) else "",
        "department": "Marketing",
        "budget": int(row.iloc[4]) if pd.notna(row.iloc[4]) else None,
        "status": "pending"
    }
    
    # 发送 POST 请求创建项目
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {TOKEN}"
    }
    
    response = requests.post(API_URL, headers=headers, data=json.dumps(project_data))
    
    if response.status_code == 200:
        print(f"Project '{project_data['name']}' created successfully!")
    else:
        print(f"Failed to create project '{project_data['name']}': {response.status_code} - {response.text}")

print("\nProject import completed!")