from flask import Blueprint, request, Response

from user.user_auth import verify_token, verify_report_view_auth
from file.file_obj import TfseFile

file_route = Blueprint('file', __name__)


@file_route.route('/get_company_report', methods=['GET'])
@verify_token
@verify_report_view_auth
def get_company_report(**kwargs):
    """
    查看公司评价报告
    Parameters:
        -
    Returns:
        response pdf二进制文件流
    """
    tfse_file = TfseFile()
    tfse_file.fid = request.args.get('file_id')
    response = Response(tfse_file.get_general_report(), content_type='application/pdf')
    return response


@file_route.route('/get_certification', methods=['GET'])
@verify_token
@verify_report_view_auth
def get_certification_route(**kwargs):
    """
    查看公司评价结果证书
    Parameters:
        -
    Returns:
        response pdf二进制文件流
    """
    tfse_file = TfseFile()
    tfse_file.fid = request.args.get('file_id')
    response = Response(tfse_file.get_general_certification(), content_type='application/pdf')
    return response