from flask import Blueprint, request
from common.security.APIAuth import api_secret
from macro.Macro import Macro

macro_route = Blueprint('macro', __name__)


@macro_route.route('/upload_macroscopic_report', methods=['POST'])
@api_secret
def upload_report():
    """
    根据行业名称上传行业宏观报告
    Parameters:
        -
    Returns:
        info: 上传结果
    """
    try:
        macro = Macro()
        macro.industry = request.form['industry']
        file = request.files['file']
        result = macro.upload_macroscopic_report(file)
        if result:
            return {"info": "上传成功", "result": {"报告fid": result}}, 200
        else:
            return {"info": "上传失败"}, 210
    except KeyError:
        return {"info": "键值错误"}, 400
    except TypeError:
        return {"info": "参数错误"}, 400


@macro_route.route('/search_macroscopic_report', methods=['POST'])
@api_secret
def search_report():
    """
    根据行业名称查询行业宏观报告
    Parameters:
        -
    Returns:
        info: 查询结果
    """
    try:
        macro = Macro()
        macro.industry = request.json['industry']
        time = request.json['time']
        skip = request.json['skip']
        limit = request.json['limit']
        result = macro.search_macroscopic_report(time, skip, limit)
        if result:
            return {"info": "查询成功", "result": result}, 200
        else:
            return {"info": "查询失败"}, 210
    except KeyError:
        return {"info": "键值错误"}, 400
    except TypeError:
        return {"info": "参数错误"}, 400


@macro_route.route('/upload_macro_data', methods=['POST'])
@api_secret
def upload_data():
    """
    上传行业宏观数据
    Parameters:
        -
    Returns:
        info: 上传结果
    """
    try:
        macro = Macro()
        req = request.json
        result = macro.upload_macroscopic_data(req)
        if result:
            return {"info": "上传成功"}, 200
        else:
            return {"info": "上传失败"}, 210
    except KeyError:
        return {"info": "键值错误"}, 400
    except TypeError:
        return {"info": "参数错误"}, 400