from flask import Blueprint, request

from Modules.AdminUser.UserAuthUtils import verify_token
from Modules.File.FileImpl import FileImpl
from Utils.ErrorUtil import AttrCheckError, APIReturnError
from Utils.RouteUtil import RouteParamsCheck

file_route = Blueprint('file', __name__)


@file_route.route('/credit_report', methods=['GET'])
@verify_token
def get_rating_report_route(**kwargs):
    """获取综信报告"""
    try:
        RouteParamsCheck(req=request.args, params=['file_id']).required()
        file_id = request.args.get('file_id')
        impl = FileImpl()
        impl.file_bucket = '报告PDF'
        impl.file_id = file_id
        impl.get_pdf()
        return impl.dict_to_return()
    except APIReturnError as e:
        return {"info": e.__str__()}, e.status_code


@file_route.route('/create_company', methods=['GET'])
@verify_token
def get_template_route(**kwargs):
    """获取综信报告"""
    impl = FileImpl()
    impl.get_company_excel()
    return impl.dict_to_return()