from flask import Response

from Utils.ObjUtil import SpecObject
from Utils.ValidateUtil import ValidateAttr


class TFSEFile(SpecObject):
    """文件类"""

    file_id = ValidateAttr(field='file_id', type=str)
    file_bucket = ValidateAttr(field='file_bucket', in_list=["综信报告", "综信证书", "ESG报告", "ESG证书", "行业分析报告"])
    file_body = ValidateAttr(field='file_body', type=Response, default=None)

    fields_map = {
        "file_id": "文件ID",
        "file_bucket": "文件桶",
        "file_body": "文件体"
    }

    def get_image(self):
        """获取图片"""

    def get_pdf(self):
        """获取PDF"""