import pandas as pd
import numpy as np
from datetime import datetime, date
import re

def calculate_workdays_from_csv(file_path):
    """
    从混乱格式的CSV中提取日期并计算工作日天数
    """
    try:
        df = pd.read_csv(file_path, header=None, encoding='utf-8')
    except UnicodeDecodeError:
        df = pd.read_csv(file_path, header=None, encoding='latin1')
    
    today = date.today()
    result = {}
    
    for index, row in df.iterrows():
        raw_date = str(row[1]).strip()
        parsed_date = None
        
        # 尝试多种日期格式
        date_formats = [
            '%Y-%m-%d',           # 2024-01-15
            '%m/%d/%Y',           # 01/15/2024
            '%b %d, %Y',          # Jan 15, 2024
            '%B %d, %Y',          # January 15, 2024
            '%d-%m-%Y',           # 15-01-2024
        ]
        
        for fmt in date_formats:
            try:
                parsed_date = datetime.strptime(raw_date, fmt).date()
                break
            except ValueError:
                continue
        
        if parsed_date is None:
            # 使用正则表达式尝试提取
            match = re.search(r'(\d{4})-(\d{2})-(\d{2})', raw_date)
            if match:
                parsed_date = date(int(match.group(1)), int(match.group(2)), int(match.group(3)))
            else:
                result[index] = None
                continue
        
        # 计算工作日天数（使用numpy的busday_count）
        workdays = np.busday_count(parsed_date, today)
        result[index] = int(workdays)
    
    return result

# 使用AI生成基础代码，但人工审查关键逻辑
def transfer_funds(from_account, to_account, amount):
    """
    资金转账函数 - 核心业务逻辑必须人工审查
    """
    # AI生成的代码（需要人工验证）
    if amount <= 0:
        raise ValueError("转账金额必须大于0")
    
    if from_account.balance < amount:
        raise ValueError("余额不足")
    
    # 关键事务逻辑 - 必须手动编写以确保原子性
    with transaction.atomic():
        from_account.balance -= amount
        to_account.balance += amount
        TransactionLog.objects.create(
            from_account=from_account,
            to_account=to_account,
            amount=amount,
            status='completed'
        )

from fastapi import FastAPI, HTTPException, Depends, Query
from pydantic import BaseModel, EmailStr, Field
from typing import List, Optional
from datetime import datetime

app = FastAPI(title="评论管理系统")

# 数据模型
class CommentCreate(BaseModel):
    user_name: str = Field(..., min_length=1, max_length=50)
    email: EmailStr
    content: str = Field(..., min_length=1, max_length=1000)
    rating: int = Field(..., ge=1, le=5)

class CommentResponse(BaseModel):
    id: int
    user_name: str
    email: str
    content: str
    rating: int
    created_at: datetime
    is_approved: bool

# 模拟数据库
comments_db = []
comment_id_counter = 1

@app.post("/comments", response_model=CommentResponse, status_code=201)
async def create_comment(comment: CommentCreate):
    global comment_id_counter
    new_comment = {
        "id": comment_id_counter,
        "user_name": comment.user_name,
        "email": comment.email,
        "content": comment.content,
        "rating": comment.rating,
        "created_at": datetime.now(),
        "is_approved": False  # 需要审核
    }
    comments_db.append(new_comment)
    comment_id_counter += 1
    return new_comment

@app.get("/comments", response_model=List[CommentResponse])
async def list_comments(
    skip: int = Query(0, ge=0),
    limit: int = Query(10, ge=1, le=100),
    min_rating: Optional[int] = Query(None, ge=1, le=5)
):
    result = comments_db
    if min_rating:
        result = [c for c in result if c["rating"] >= min_rating]
    return result[skip:skip + limit]

@app.get("/comments/{comment_id}", response_model=CommentResponse)
async def get_comment(comment_id: int):
    for comment in comments_db:
        if comment["id"] == comment_id:
            return comment
    raise HTTPException(status_code=404, detail="评论不存在")