from fastapi import APIRouter, Depends
from app.models.database import get_db
from app.api.deps import get_admin_user
from app.models.user import UserDB
from app.models.order import ShippingOrderDB, OrderStatus
from app.models.community import CommunityDB
from sqlalchemy.orm import Session
from datetime import datetime, timedelta
from app.core.response import success_response, error_response, ResponseModel
router = APIRouter()

# 获取基础信息
# 总用户数， 今日新增用户数， 总配送订单数， 今日新增配送订单数
# 小区数量， 今日新增小区数

@router.get("")
async def get_dashboard_info(
    db: Session = Depends(get_db)
):
    # 获取接口包括：
    # 总用户数， 今日新增用户数，昨日新增用户数， 总配送订单数， 今日新增配送订单数，昨日新增配送订单数
    # 小区数量， 今日新增小区数，昨日新增小区数

    # 总用户数
    total_user_count = db.query(UserDB).count()

    # 今日新增用户数
    today_user_count = db.query(UserDB).filter(UserDB.create_time >= datetime.now().date()).count()

    # 昨日新增用户数
    yesterday_user_count = db.query(UserDB).filter(UserDB.create_time >= datetime.now().date() - timedelta(days=1)).count()

    # 总配送订单数
    total_order_count = db.query(ShippingOrderDB).filter(ShippingOrderDB.status == OrderStatus.COMPLETED).count()

    # 昨日订单数
    yesterday_order_count = db.query(ShippingOrderDB).filter(ShippingOrderDB.completed_time >= datetime.now().date() - timedelta(days=1), ShippingOrderDB.status == OrderStatus.COMPLETED).count()
    
    # 今日配送订单数
    today_order_count = db.query(ShippingOrderDB).filter(ShippingOrderDB.completed_time >= datetime.now().date(), ShippingOrderDB.status == OrderStatus.COMPLETED).count()
    
    # 小区数量
    total_community_count = db.query(CommunityDB).count()
    
    # 昨日新增小区数
    yesterday_community_count = db.query(CommunityDB).filter(CommunityDB.create_time >= datetime.now().date() - timedelta(days=1)).count()

    # 今日新增小区数
    today_community_count = db.query(CommunityDB).filter(CommunityDB.create_time >= datetime.now().date()).count()

    return success_response(data={
        "total_user_count": total_user_count,
        "today_user_count": today_user_count,
        "yesterday_user_count": yesterday_user_count,
        "total_order_count": total_order_count,
        "today_order_count": today_order_count,
        "yesterday_order_count": yesterday_order_count,
        "total_community_count": total_community_count,
        "today_community_count": today_community_count,
        "yesterday_community_count": yesterday_community_count,
    })