78 lines
3.1 KiB
Python
78 lines
3.1 KiB
Python
import datetime
|
|
import math
|
|
import pandas as pd
|
|
from fastapi import APIRouter, Depends, UploadFile, File, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
|
|
from APPData.Crud import CompanyCrud, BalanceCrud
|
|
from Utils.DataBase.MongoHelperUtils import get_mongodb
|
|
from Utils.DataBase.SqlAlchemyUtils import get_db
|
|
from Utils.TycApi.CommonUtil import CommonUtils
|
|
|
|
router = APIRouter(
|
|
tags=["资产负债表"],
|
|
prefix="/api/balance_sheet"
|
|
)
|
|
|
|
|
|
@router.post("/upload", summary="上传数据")
|
|
async def func01(name: str, excel: UploadFile = File(...), 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
|
|
contents = await excel.read()
|
|
data = pd.read_excel(contents, skiprows=3)
|
|
# 行数
|
|
row = data.shape[0]
|
|
# 列数
|
|
col = data.shape[1]
|
|
# 财报数据
|
|
for num in range(1, col):
|
|
finance_dict = dict()
|
|
finance_dict['企业ID'] = cid
|
|
finance_dict['更新时间'] = CommonUtils.get_current_time()
|
|
for row_num in range(row):
|
|
try:
|
|
if isinstance(data.iloc[row_num, 0], float):
|
|
finance_dict['截止日期'] = data.iloc[row_num, num].strftime("%Y-%m-%d")
|
|
continue
|
|
if isinstance(data.iloc[row_num, num], datetime.datetime):
|
|
finance_dict[data.iloc[row_num, 0].strip()] = data.iloc[row_num, num].strftime("%Y-%m-%d")
|
|
else:
|
|
finance_dict[data.iloc[row_num, 0].strip()] = data.iloc[row_num, num]
|
|
except AttributeError:
|
|
if isinstance(data.iloc[row_num, num], datetime.datetime):
|
|
try:
|
|
finance_dict[data.iloc[row_num, 0].strip()] = data.iloc[row_num, num].strftime("%Y-%m-%d")
|
|
except AttributeError:
|
|
finance_dict[data.iloc[row_num, 0]] = data.iloc[row_num, num].strftime("%Y-%m-%d")
|
|
else:
|
|
finance_dict[data.iloc[row_num, 0]] = data.iloc[row_num, num]
|
|
|
|
for key, value in finance_dict.items():
|
|
try:
|
|
if math.isnan(value):
|
|
finance_dict[key] = None
|
|
except TypeError:
|
|
continue
|
|
mongodb = get_mongodb()
|
|
BalanceCrud.create_balance_by_id(db=db, mongodb=mongodb, _id=cid, reportdate=finance_dict['截止日期'],
|
|
condition={"企业ID": cid, "截止日期": finance_dict['截止日期']}, content=finance_dict)
|
|
mongodb.client.close()
|
|
return {"info": "Success"}
|
|
|
|
|
|
@router.post("/view", summary="查看数据")
|
|
def func02(cid: str, db: Session = Depends(get_db)):
|
|
mongodb = get_mongodb()
|
|
# 查询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 = BalanceCrud.find_balance_by_id(db=db, mongodb=mongodb, _id=cid)
|
|
return data
|