dataplatform/APPData/Router/DishonestRouter.py

import requests
from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session

from APPData.Crud import CompanyCrud
from Utils.DBHelper.MongoDBInstance import MG_DB_TEST, MG_DB_TYC
from Utils.DataBase.SqlAlchemyUtils import get_db

from Utils.TycApi.CommonUtil import CommonUtils
from Utils.TycApi.Configure import DB_SFFX
from Utils.TycApi.Dishonest import get_tyc_data

router = APIRouter(
    tags=["失信人"],
    prefix="/api/dishonest"
)


@router.get("/pull", summary="拉取数据")
def func01(name: str, db: Session = Depends(get_db)):
    # 根据name查询企业是否存在
    data = CompanyCrud.get_company_by_name(db=db, name=name)
    if not data:
        # 不能存在新建企业再返回cid
        data = CompanyCrud.create_company_by_name(db=db, name=name)
    cid = data.id
    # 查询数据库是否有此条数据
    search_data = MG_DB_TEST.find_all_data(
        "WR_DataBase_v2_test",
        "C9.5_失信人",
        {"企业ID": cid},
        []
    )
    if search_data:
        return {"info": "Data already exists", "Datail": search_data}
    # 拉取天眼查数据到mongodb
    tyc_res = get_tyc_data(name)
    if tyc_res != 'success':
        raise HTTPException(status_code=401, detail=tyc_res)
    # 查询天眼查数据库
    tyc_data = MG_DB_TYC.find_single_column(
        DB_SFFX,
        '失信人',
        {"企业名称": name},
        '失信人'
    )
    if not tyc_data:
        raise HTTPException(status_code=401, detail='Data query failed')
    try:
        record = tyc_data['result']
        for item in record:
            data = dict()
            data['企业ID'] = cid
            data['案号'] = item['casecode']
            data['发布日期'] = CommonUtils.time_stamp(item['publishdate'])
            data['失信行为'] = item['disrupttypename']
            data['履行情况'] = item['performance']
            data['执行依据文号'] = item['gistid']
            data['执行法院'] = item['courtname']
            data['涉案金额(元)'] = item['duty']
            data['立案日期'] = CommonUtils.time_stamp(item['regdate'])
            data['更新时间'] = CommonUtils.get_current_time()
            # 保存天眼查数据到企业数据库
            MG_DB_TEST.upsert_single_data(
                "WR_DataBase_v2_test",
                "C9.5_失信人",
                {"企业ID": cid, "执行依据文号": data['执行依据文号']},
                data
            )
    except KeyError:
        pass
    return {"info": "Pull succeeded"}


@router.post("/update", summary="更新数据")
def func02(cid: str, db: Session = Depends(get_db)):
    # 查询mysql数据库中的_id
    db_item = CompanyCrud.get_company_by_id(db, _id=cid)
    if not db_item:
        raise HTTPException(status_code=204, detail="Company Not Found")
        # 拉取天眼查数据到mongodb
    name = db_item.name
    # 拉取天眼查数据到mongodb
    tyc_res = get_tyc_data(name)
    if tyc_res != 'success':
        raise HTTPException(status_code=401, detail=tyc_res)
    # 查询天眼查数据库
    tyc_data = MG_DB_TYC.find_single_column(
        DB_SFFX,
        '失信人',
        {"企业名称": name},
        '失信人'
    )
    if not tyc_data:
        raise HTTPException(status_code=401, detail='Data query failed')
    try:
        record = tyc_data['result']
        for item in record:
            data = dict()
            data['企业ID'] = cid
            data['案号'] = item['casecode']
            data['发布日期'] = CommonUtils.time_stamp(item['publishdate'])
            data['失信行为'] = item['disrupttypename']
            data['履行情况'] = item['performance']
            data['执行依据文号'] = item['gistid']
            data['执行法院'] = item['courtname']
            data['涉案金额(元)'] = item['duty']
            data['立案日期'] = CommonUtils.time_stamp(item['regdate'])
            data['更新时间'] = CommonUtils.get_current_time()
            # 保存天眼查数据到企业数据库
            MG_DB_TEST.upsert_single_data(
                "WR_DataBase_v2_test",
                "C9.5_失信人",
                {"企业ID": cid, "执行依据文号": data['执行依据文号']},
                data
            )
    except KeyError:
        pass
    return {"info": "Update succeeded"}


@router.get("/view", summary="查看数据")
def func03(cid: str, db: Session = Depends(get_db)):
    # 查询mysql数据库中的_id
    db_item = CompanyCrud.get_company_by_id(db, _id=cid)
    if not db_item:
        raise HTTPException(status_code=204, detail="Company Not Found")
    # 根据_id查询mongodb数据库
    data = MG_DB_TEST.find_all_data(
        "WR_DataBase_v2_test",
        "C9.5_失信人",
        {"企业ID": cid},
        []
    )

    return data