143 lines
3.9 KiB
Python
143 lines
3.9 KiB
Python
import datetime
|
|
|
|
import pandas as pd
|
|
|
|
from Manage.macro.db import *
|
|
|
|
|
|
def upload_macroscopic_report_script(industry, file):
|
|
"""
|
|
根据行业保存PDF宏观报告
|
|
Parameters:
|
|
industry: 行业名称
|
|
file: pdf文件
|
|
Returns:
|
|
result: 生成结果
|
|
"""
|
|
|
|
def save_report():
|
|
"""
|
|
保存PDF宏观报告
|
|
Parameters:
|
|
-
|
|
Returns:
|
|
result: 生成结果
|
|
"""
|
|
file_name = industry + '宏观报告'
|
|
file_id = upload_report(file_name, file)
|
|
return file_id
|
|
|
|
# 保存新宏观报告pdf文件
|
|
new_fid = save_report()
|
|
# 根据行业名称查询此行业是否存在宏观报告
|
|
macrosc = find_macroscopic_report_data(industry)
|
|
if macrosc:
|
|
update_time = datetime.datetime.now().strftime('%Y-%m-%d')
|
|
# 使用新fid更新此条数据
|
|
upload_res = update_macroscopic_report_data(industry, update_time, new_fid)
|
|
if upload_res:
|
|
return new_fid
|
|
else:
|
|
return False
|
|
else:
|
|
# 不存在此条数据,就插入一条新数据
|
|
insert_data = dict()
|
|
insert_data['行业名称'] = industry
|
|
insert_data['报告名称'] = industry + '宏观报告'
|
|
insert_data['报告fid'] = new_fid
|
|
insert_data['上传日期'] = datetime.datetime.now().strftime('%Y-%m-%d')
|
|
save_res = insert_macroscopic_report_data(insert_data)
|
|
if save_res:
|
|
return new_fid
|
|
else:
|
|
return False
|
|
|
|
|
|
def search_macroscopic_report_script(industry, time, skip, limit):
|
|
"""
|
|
根据行业或上传日期查询PDF宏观报告
|
|
Parameters:
|
|
industry: 行业名称
|
|
time: 上传日期
|
|
skip: 页码
|
|
limit: 每页条数
|
|
Returns:
|
|
result: 查询结果
|
|
"""
|
|
result = find_macroscopic_report_records(industry, time, skip, limit)
|
|
if len(result['data']) > 0:
|
|
for item in result['data']:
|
|
item['报告fid'] = '/file/get_industry_report?file_id={}'.format(item['报告fid'])
|
|
return result
|
|
|
|
|
|
def upload_macroscopic_data_script(req):
|
|
"""
|
|
上传行业宏观数据
|
|
Parameters:
|
|
req: 上传数据
|
|
Returns:
|
|
result: 查询结果
|
|
"""
|
|
result = upload_industry_macroscopic_data(req)
|
|
return result
|
|
|
|
|
|
def update_macroscopic_data_script(req):
|
|
"""
|
|
编辑行业宏观数据
|
|
Parameters:
|
|
req: 编辑后数据
|
|
Returns:
|
|
result: 修改结果
|
|
"""
|
|
result = update_industry_macroscopic_data(req)
|
|
return result
|
|
|
|
|
|
def search_macroscopic_data_script(req, skip, limit):
|
|
"""
|
|
查询行业宏观数据
|
|
Parameters:
|
|
req: 查询条件
|
|
skip: 页码
|
|
limit: 每页条数
|
|
Returns:
|
|
result: 查询结果
|
|
"""
|
|
result = search_macroscopic_data(req, skip, limit)
|
|
return result
|
|
|
|
|
|
def get_macro_data():
|
|
df = pd.read_excel('./static/C13-02.xlsx')
|
|
for index, row in df.iterrows():
|
|
num = len(row)
|
|
for i in range(num-1):
|
|
req = {
|
|
"一级行业": "信息技术业",
|
|
"二级行业": row.keys()[i + 1],
|
|
"数据": row[i + 1],
|
|
"统计字段名称": "营业收入",
|
|
"统计周期": "年度",
|
|
"统计时间": row[0],
|
|
"数据单位": "亿元"
|
|
}
|
|
upload_overview_macroscopic_data(req)
|
|
|
|
|
|
def get_macro_data_over():
|
|
df = pd.read_excel('./static/C13-02.xlsx')
|
|
for index, row in df.iterrows():
|
|
num = len(row)
|
|
for i in range(num-1):
|
|
req = {
|
|
"行业名称": "制造业",
|
|
"数据": row[i + 1],
|
|
"统计字段名称": row.keys()[i + 1],
|
|
"统计周期": "年度",
|
|
"统计时间": "2020年",
|
|
"数据单位": "亿元"
|
|
}
|
|
upload_overview_macroscopic_data(req)
|