jiachenlong/backend/app/routers/yichens.py

from fastapi import APIRouter, Depends, Query
from sqlalchemy import func, text
from sqlalchemy.orm import Session
from pydantic import BaseModel
from typing import Optional, List
from datetime import datetime, date
from app.core.coolbot_db import get_coolbot_db

router = APIRouter(prefix="/api/yichens", tags=["一尘看板"])

# ============ 数据模型 ============
class YichensPostStats(BaseModel):
    total_posts: int
    total_deals: int  # 出售
    total_wants: int  # 求购
    total_replies: int
    total_views: int
    avg_price: Optional[float]

class CategoryStat(BaseModel):
    category: str
    count: int

class PostItem(BaseModel):
    post_id: str
    title: str
    category: Optional[str]
    post_type: str
    price: Optional[float]
    author_username: str
    post_time: str
    reply_count: int
    view_count: int
    url: Optional[str]
    content: Optional[str]

class UserStat(BaseModel):
    total_users: int
    new_users_today: int
    sellers: int

class UserItem(BaseModel):
    user_id: str
    username: str
    avatar_url: Optional[str]
    content: Optional[str]
    credit_level: Optional[str]
    credit_score: Optional[int]
    post_count: int
    is_seller: bool
    registration_date: Optional[str]

# ============ 统计接口 ============
@router.get("/stats/posts", response_model=YichensPostStats)
def get_post_stats(days: int = Query(30, ge=1, le=365), db: Session = Depends(get_coolbot_db)):
    """获取帖子统计"""
    result = db.execute(text("""
        SELECT 
            COUNT(*) as total_posts,
            COUNT(*) FILTER (WHERE post_type = 'deal') as total_deals,
            COUNT(*) FILTER (WHERE post_type = 'want') as total_wants,
            COALESCE(SUM(reply_count), 0) as total_replies,
            COALESCE(SUM(view_count), 0) as total_views,
            AVG(price) as avg_price
        FROM yichens_posts 
        WHERE post_time >= NOW() - INTERVAL '1 day' * :days
    """), {"days": days}).fetchone()
    
    return YichensPostStats(
        total_posts=result[0] or 0,
        total_deals=result[1] or 0,
        total_wants=result[2] or 0,
        total_replies=result[3] or 0,
        total_views=result[4] or 0,
        avg_price=float(result[5]) if result[5] else None
    )

@router.get("/stats/categories", response_model=List[CategoryStat])
def get_category_stats(days: int = Query(30, ge=1, le=365), db: Session = Depends(get_coolbot_db)):
    """按分类统计帖子数量"""
    results = db.execute(text("""
        SELECT category, COUNT(*) as count
        FROM yichens_posts
        WHERE post_time >= NOW() - INTERVAL '1 day' * :days
        GROUP BY category
        ORDER BY count DESC
    """), {"days": days}).fetchall()
    
    return [CategoryStat(category=r[0] or "未知", count=r[1]) for r in results]

@router.get("/stats/users", response_model=UserStat)
def get_user_stats(db: Session = Depends(get_coolbot_db)):
    """获取用户统计"""
    result = db.execute(text("""
        SELECT 
            COUNT(*) as total_users,
            COUNT(*) FILTER (WHERE created_at::date = CURRENT_DATE) as new_users_today,
            COUNT(*) FILTER (WHERE is_seller = true) as sellers
        FROM yichens_users
    """)).fetchone()
    
    return UserStat(
        total_users=result[0] or 0,
        new_users_today=result[1] or 0,
        sellers=result[2] or 0
    )

@router.get("/posts")
def get_posts(
    limit: int = Query(20, ge=1, le=500),
    offset: int = Query(0, ge=0),
    category: Optional[str] = None,
    post_type: Optional[str] = None,
    keyword: Optional[str] = None,
    db: Session = Depends(get_coolbot_db)
):
    """获取帖子列表 - 支持全局搜索，返回总数和分页信息"""
    # 构建WHERE条件
    where_clauses = ["1=1"]
    params = {"limit": limit, "offset": offset}
    
    if category:
        where_clauses.append("category = :category")
        params["category"] = category
    
    if post_type:
        where_clauses.append("post_type = :post_type")
        params["post_type"] = post_type
    
    # 全局搜索
    if keyword:
        where_clauses.append("(title ILIKE :keyword OR content ILIKE :keyword OR category ILIKE :keyword OR author_username ILIKE :keyword)")
        params["keyword"] = f"%{keyword}%"
    
    where_sql = " AND ".join(where_clauses)
    
    # 查询总数
    count_query = f"SELECT COUNT(*) FROM yichens_posts WHERE {where_sql}"
    total_count = db.execute(text(count_query), {k: v for k, v in params.items() if k not in ['limit', 'offset']}).scalar() or 0
    
    # 查询数据
    data_query = f"""
        SELECT post_id, title, content, category, post_type, price, 
               author_username, post_time, reply_count, view_count, url
        FROM yichens_posts
        WHERE {where_sql}
        ORDER BY post_time DESC LIMIT :limit OFFSET :offset
    """
    results = db.execute(text(data_query), params).fetchall()
    
    posts = [PostItem(
        post_id=r[0],
        title=r[1] or "",
        content=r[2] or "",
        category=r[3],
        post_type=r[4] or "",
        price=float(r[5]) if r[5] else None,
        author_username=r[6] or "",
        post_time=str(r[7]) if r[7] else "",
        reply_count=r[8] or 0,
        view_count=r[9] or 0,
        url=r[10]
    ) for r in results]
    
    return {
        "posts": posts,
        "total": total_count,
        "page": offset // limit + 1,
        "page_size": limit
    }

@router.get("/users", response_model=List[UserItem])
def get_users(
    limit: int = Query(20, ge=1, le=500),
    offset: int = Query(0, ge=0),
    is_seller: Optional[bool] = None,
    db: Session = Depends(get_coolbot_db)
):
    """获取用户列表"""
    query = """
        SELECT user_id, username, avatar_url, credit_level, credit_score,
               post_count, is_seller, registration_date
        FROM yichens_users
        WHERE 1=1
    """
    params = {"limit": limit, "offset": offset}
    
    if is_seller is not None:
        query += " AND is_seller = :is_seller"
        params["is_seller"] = is_seller
    
    query += " ORDER BY post_count DESC LIMIT :limit OFFSET :offset"
    
    results = db.execute(text(query), params).fetchall()
    
    return [UserItem(
        user_id=r[0],
        username=r[1] or "",
        avatar_url=r[2],
        credit_level=r[3],
        credit_score=r[4],
        post_count=r[5] or 0,
        is_seller=r[6] or False,
        registration_date=str(r[7]) if r[7] else None
    ) for r in results]


@router.get("/stats/today")
async def get_today_stats(db: Session = Depends(get_coolbot_db)):
    """获取今日新增帖子统计"""
    query = """
        SELECT 
            COUNT(*) as total,
            SUM(CASE WHEN post_type = 'deal' THEN 1 ELSE 0 END) as deals,
            SUM(CASE WHEN post_type = 'want' THEN 1 ELSE 0 END) as wants,
            SUM(CASE WHEN post_type = 'normal' THEN 1 ELSE 0 END) as others,
            SUM(CASE WHEN category LIKE '%龙%' THEN 1 ELSE 0 END) as dragons,
            SUM(CASE WHEN category LIKE '%马%' THEN 1 ELSE 0 END) as horses,
            SUM(CASE WHEN category LIKE '%蛇%' THEN 1 ELSE 0 END) as snakes,
            SUM(CASE WHEN title LIKE '%天马%' OR content LIKE '%天马%' THEN 1 ELSE 0 END) as tianma
        FROM yichens_posts
        WHERE post_time >= CURRENT_DATE
    """
    result = db.execute(text(query)).fetchone()
    return {
        "total": result[0] or 0,
        "deals": result[1] or 0,
        "wants": result[2] or 0,
        "others": result[3] or 0,
        "dragons": result[4] or 0,
        "horses": result[5] or 0,
        "snakes": result[6] or 0,
        "tianma": result[7] or 0
    }

@router.get("/stats/hour")
async def get_hour_stats(db: Session = Depends(get_coolbot_db)):
    """获取近一个小时新增帖子统计"""
    query = """
        SELECT 
            COUNT(*) as total,
            SUM(CASE WHEN post_type = 'deal' THEN 1 ELSE 0 END) as deals,
            SUM(CASE WHEN post_type = 'want' THEN 1 ELSE 0 END) as wants,
            SUM(CASE WHEN post_type = 'normal' THEN 1 ELSE 0 END) as others,
            SUM(CASE WHEN category LIKE '%龙%' THEN 1 ELSE 0 END) as dragons,
            SUM(CASE WHEN category LIKE '%马%' THEN 1 ELSE 0 END) as horses,
            SUM(CASE WHEN category LIKE '%蛇%' THEN 1 ELSE 0 END) as snakes
        FROM yichens_posts
        WHERE post_time >= NOW() - INTERVAL '1 hour'
    """
    result = db.execute(text(query)).fetchone()
    return {
        "total": result[0] or 0,
        "deals": result[1] or 0,
        "wants": result[2] or 0,
        "others": result[3] or 0,
        "dragons": result[3] or 0,
        "horses": result[4] or 0,
        "snakes": result[5] or 0
    }

@router.get("/stats/today-category")
async def get_today_category_stats(db: Session = Depends(get_coolbot_db)):
    """获取今日帖子分类统计"""
    query = """
        SELECT category, COUNT(*) as count
        FROM yichens_posts
        WHERE post_time >= CURRENT_DATE
        GROUP BY category
        ORDER BY count DESC
    """
    results = db.execute(text(query)).fetchall()
    return [{"category": r[0] or "未分类", "count": r[1]} for r in results]


@router.get("/stats/dragons-today")
def get_dragons_stats_today(
    db: Session = Depends(get_coolbot_db)
):
    """获取今日龙钞详细统计数据（按号码分类）- 就高不就低"""
    from sqlalchemy import text
    
    # 1. 带4：包含"带4"、"带四"、"通货"
    dai4 = db.execute(text("""
        SELECT 
            COUNT(*) as total,
            SUM(CASE WHEN post_type = 'deal' THEN 1 ELSE 0 END) as deals,
            SUM(CASE WHEN post_type = 'want' THEN 1 ELSE 0 END) as wants
        FROM yichens_posts
        WHERE post_time >= CURRENT_DATE
            AND category LIKE '%龙%'
            AND (
                content LIKE '%带4%' OR title LIKE '%带4%'
                OR content LIKE '%带四%' OR title LIKE '%带四%'
                OR content LIKE '%通货%' OR title LIKE '%通货%'
            )
    """)).fetchone()
    
    # 2. 无4：包含"无4"、"无四"，排除"无47"、"无247"、"无四七"、"永恒"
    wu4 = db.execute(text("""
        SELECT 
            COUNT(*) as total,
            SUM(CASE WHEN post_type = 'deal' THEN 1 ELSE 0 END) as deals,
            SUM(CASE WHEN post_type = 'want' THEN 1 ELSE 0 END) as wants
        FROM yichens_posts
        WHERE post_time >= CURRENT_DATE
            AND category LIKE '%龙%'
            AND (
                content LIKE '%无4%' OR title LIKE '%无4%'
                OR content LIKE '%无四%' OR title LIKE '%无四%'
            )
            AND content NOT LIKE '%无47%' AND title NOT LIKE '%无47%'
            AND content NOT LIKE '%无247%' AND title NOT LIKE '%无247%'
            AND content NOT LIKE '%永恒%' AND title NOT LIKE '%永恒%'
            AND content NOT LIKE '%无四七%' AND title NOT LIKE '%无四七%'
    """)).fetchone()
    
    # 3. 无47：包含"无47"、"永恒"、"无四七"，排除"无247"
    wu47 = db.execute(text("""
        SELECT 
            COUNT(*) as total,
            SUM(CASE WHEN post_type = 'deal' THEN 1 ELSE 0 END) as deals,
            SUM(CASE WHEN post_type = 'want' THEN 1 ELSE 0 END) as wants
        FROM yichens_posts
        WHERE post_time >= CURRENT_DATE
            AND category LIKE '%龙%'
            AND (
                content LIKE '%无47%' OR title LIKE '%无47%'
                OR content LIKE '%永恒%' OR title LIKE '%永恒%'
                OR content LIKE '%无四七%' OR title LIKE '%无四七%'
            )
            AND content NOT LIKE '%无247%' AND title NOT LIKE '%无247%'
    """)).fetchone()
    
    # 4. 无247：包含"无247"、"天马"、"金山"，排除"无347"
    wu247 = db.execute(text("""
        SELECT 
            COUNT(*) as total,
            SUM(CASE WHEN post_type = 'deal' THEN 1 ELSE 0 END) as deals,
            SUM(CASE WHEN post_type = 'want' THEN 1 ELSE 0 END) as wants
        FROM yichens_posts
        WHERE post_time >= CURRENT_DATE
            AND category LIKE '%龙%'
            AND (
                content LIKE '%无247%' OR title LIKE '%无247%'
                OR content LIKE '%天马%' OR title LIKE '%天马%'
                OR content LIKE '%金山%' OR title LIKE '%金山%'
            )
            AND content NOT LIKE '%无347%' AND title NOT LIKE '%无347%'
    """)).fetchone()
    
    # 5. 无347：包含"无347"、"钻石"、"金马"、"魅力"、"朦胧"
    wu347 = db.execute(text("""
        SELECT 
            COUNT(*) as total,
            SUM(CASE WHEN post_type = 'deal' THEN 1 ELSE 0 END) as deals,
            SUM(CASE WHEN post_type = 'want' THEN 1 ELSE 0 END) as wants
        FROM yichens_posts
        WHERE post_time >= CURRENT_DATE
            AND category LIKE '%龙%'
            AND (
                content LIKE '%无347%' OR title LIKE '%无347%'
                OR content LIKE '%钻石%' OR title LIKE '%钻石%'
                OR content LIKE '%金马%' OR title LIKE '%金马%'
                OR content LIKE '%魅力%' OR title LIKE '%魅力%'
                OR content LIKE '%朦胧%' OR title LIKE '%朦胧%'
            )
    """)).fetchone()
    
    return {
        "dai4": {"total": dai4[0] or 0, "deals": dai4[1] or 0, "wants": dai4[2] or 0},
        "wu4": {"total": wu4[0] or 0, "deals": wu4[1] or 0, "wants": wu4[2] or 0},
        "wu47": {"total": wu47[0] or 0, "deals": wu47[1] or 0, "wants": wu47[2] or 0},
        "wu247": {"total": wu247[0] or 0, "deals": wu247[1] or 0, "wants": wu247[2] or 0},
        "wu347": {"total": wu347[0] or 0, "deals": wu347[1] or 0, "wants": wu347[2] or 0}
    }