61 lines
2.7 KiB
Python
61 lines
2.7 KiB
Python
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,
|
|
}) |