from flask import Blueprint, request, Response

from CompanyUser.CompanyUserAuth import verify_token, verify_report_view_auth
from file.file_obj import TfseFile

file_route = Blueprint('file', __name__)


@file_route.route('/pdf', methods=['GET'])
@verify_token
@verify_report_view_auth
def get_file(**kwargs):
    """
    获取pdf文件
    """
    tfse_file = TfseFile()
    tfse_file.file_type = request.args.get('file_type')
    tfse_file.file_id = request.args.get('file_id')
    pdf = tfse_file.get_pdf_file()
    if pdf is not None:
        response = Response(pdf, content_type='application/pdf')
        return response
    else:
        return {"info": "文件不存在"}, 200