changes 接口重写
This commit is contained in:
parent
b9ab8761ed
commit
1868637993
|
@ -0,0 +1,10 @@
|
||||||
|
def upsert_data(mongodb, data):
|
||||||
|
mongodb.upsert_data("WR_DataBase_v2_test", 'C12.2_区域经济',
|
||||||
|
{"省/直辖市": data['省/直辖市'], "地区": data['地区'], "年份": data["年份"]}, data)
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
def find_data(mongodb, condition):
|
||||||
|
data = mongodb.find_all_data("WR_DataBase_v2_test", 'C12.2_区域经济',
|
||||||
|
{"省/直辖市": condition.provinces, "地区": condition.region, "年份": condition.year}, [])
|
||||||
|
return data
|
|
@ -3,6 +3,8 @@ import math
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
from fastapi import APIRouter, Depends, UploadFile, File, HTTPException
|
from fastapi import APIRouter, Depends, UploadFile, File, HTTPException
|
||||||
from sqlalchemy.orm import Session
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
|
from APPData.Crud import MacroDataCrud
|
||||||
from APPData.Schemas import CompanySchemas
|
from APPData.Schemas import CompanySchemas
|
||||||
from Utils.DataBase.MongoHelperUtils import get_mongodb
|
from Utils.DataBase.MongoHelperUtils import get_mongodb
|
||||||
from Utils.DataBase.SqlAlchemyUtils import get_db
|
from Utils.DataBase.SqlAlchemyUtils import get_db
|
||||||
|
@ -29,14 +31,12 @@ async def func01(region: str, excel: UploadFile = File(...), db: Session = Depen
|
||||||
df_line[key] = None
|
df_line[key] = None
|
||||||
except TypeError:
|
except TypeError:
|
||||||
continue
|
continue
|
||||||
mongodb.upsert_data("WR_DataBase_v2_test", 'C12.2_区域经济',
|
MacroDataCrud.upsert_data(mongodb=mongodb, data=df_line)
|
||||||
{"省/直辖市": region, "地区": df_line['地区'], "年份": df_line["年份"]}, df_line)
|
|
||||||
return {"info": "Success"}
|
return {"info": "Success"}
|
||||||
|
|
||||||
|
|
||||||
@router.post("/view", summary="查看数据")
|
@router.post("/view", summary="查看数据")
|
||||||
def func03(req_body: CompanySchemas.MacroSearch, db: Session = Depends(get_db)):
|
def func03(req_body: CompanySchemas.MacroSearch):
|
||||||
mongodb = get_mongodb()
|
mongodb = get_mongodb()
|
||||||
data = mongodb.find_all_data("WR_DataBase_v2_test", 'C12.2_区域经济',
|
data = MacroDataCrud.find_data(mongodb=mongodb, condition=req_body)
|
||||||
{"省/直辖市": req_body.provinces, "地区": req_body.region, "年份": req_body.year}, [])
|
|
||||||
return data
|
return data
|
||||||
|
|
Loading…
Reference in New Issue