python-fastapi-mm-zhcs-yj-api/routers/api/rainfall/chemical_company.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from fastapi import APIRouter, Request, Depends, Query, HTTPException, status
from common.security import valid_access_token
from sqlalchemy.orm import Session
from sqlalchemy.sql import func
from common.auth_user import *
from sqlalchemy import  text
from pydantic import BaseModel
from common.BigDataCenterAPI import *
from database import get_db
from typing import List
from models import *
from utils import *
from utils.spatial import *
from utils.rainfall_util import *
import json
import traceback
from jobs.rainfall_conditions_job import get_stcd_data
from datetime import datetime,timedelta


router = APIRouter()


@router.get("/list")
async def get_list(
    area_name: str = Query(None),
    company_type:str = Query(None),
    keyword: str= Query(None),
    history_time:int = Query(None),
    future_time:int = Query(None),
    db: Session = Depends(get_db),
    page: int = Query(1, gt=0, description='页码'),
    pageSize: int = Query(10, gt=0, description='每页条目数量')
):
    try:
        # 计算 OFFSET 值
        offset = (page - 1) * pageSize
        # 构造基础查询
        base_sql = "SELECT * FROM sharedb.chemical_company"
        count_sql = "SELECT COUNT(*) FROM sharedb.chemical_company"

        # 添加 WHERE 条件
        conditions = []
        params = {}
        if area_name:
            conditions.append("area = :area_name")
            params['area_name'] = area_name
        if keyword:
            conditions.append("`company_name` LIKE :keyword")
            params['keyword'] = f"%{keyword}%"
        if company_type:
            conditions.append("`company_type` LIKE :company_type")
            params['company_type'] = f"%{company_type}%"

        if conditions:
            base_sql += " WHERE " + " AND ".join(conditions)
            count_sql += " WHERE " + " AND ".join(conditions)

        count_query = text(count_sql) #.bindparams(**params)

        # 执行统计查询并获取总数据量
        total = db.execute(count_query,params).scalar()

        # 添加 LIMIT 和 OFFSET
        paginated_sql = f"{base_sql} LIMIT :limit OFFSET :offset"
        params['limit'] = pageSize
        params['offset'] = offset

        # 构造查询对象
        paginated_query = text(paginated_sql) #.bindparams(**params)
        # 执行分页查询并获取结果
        result = db.execute(paginated_query,params).fetchall()
        # 将结果转换为rain_pits.py字典列表
        result_list = []
        for row in result:
            data = dict(row)
            data['have_video'] = False
            data['video_unit_indexcode'] = ''
            company_name = data['company_name']
            video_unit_info = unitName_get_video_region_info(db,company_name)
            if video_unit_info:
                data['have_video'] = True
                data['video_unit_indexcode'] = video_unit_info.indexCode
            data['weather_warning_type'],data['weather_warninglevel'] = get_weather_warning(data['area'], db)
            # data['weather_warning_type'] = '暴雨预警'
            # data['weather_warninglevel'] = '3'
            if history_time:
                real_code = get_real_code(db,data['longitude'],data['latitude'])
                rainfall = get_rainfall(real_code,history_time,db)
                data['rainfall'] = rainfall
            if future_time:
                data['rainfall'] = 0
            result_list.append(data)
        return {
            "code": 200,
            "msg": "操作成功",
            "data": result_list,
            "total": total,
            "page": page,
            "pageSize": pageSize,
            "totalPages": (total + pageSize - 1) // pageSize
        }
    except Exception as e:
        traceback.print_exc()
        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")


@router.get('/get_video_list')
async def get_video_list(
    video_unit_indexcode: str = Query(None, description='所属单位索引码'),
    db: Session = Depends(get_db),
    page: int = Query(1, gt=0, description='页码'),
    pageSize: int = Query(10, gt=0, description='每页条目数量')
):
    # 大屏左下角视频及更多视频
    try:
        query = db.query(TPVideoInfo)

        if video_unit_indexcode:
            query = query.filter(TPVideoInfo.regionPath.like(f'%{video_unit_indexcode}%'))
        else:
            return {
            "code": 200,
            "msg": "操作成功",
            "data": None}
        total_items = query.count()
        # query = query.order_by()
        videos = query.offset(pageSize * (page - 1)).limit(pageSize).all()
        data = []
        for row in videos:
            # row = row[0]
            # print(type(row),row[0])
            data.append({
                "name":row.name,
                "video_code": row.gbIndexCode,
                "status":row.status,
                "statusName":row.statusName,
                "regionPath":row.regionPath,
                "installPlace":row.installPlace,
                "cameraTypeName":row.cameraTypeName,
                "cameraType":row.cameraType
            })
        return {
            "code": 200,
            "msg": "操作成功",
            "data": data,
            "total": total_items,
            "page": page,
            "pageSize": pageSize,
            "totalPages": (total_items + pageSize - 1) // pageSize
        }
    except Exception as e:
        # 处理异常
        traceback.print_exc()
        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e))