tfse-admin-api-v0.2/Manage/macro/scripts.py

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)