from flask import Response

from DBHelper.MongoHelper import MongoHelper
from Objects.InProject.Other.TFSEFile import TFSEFile


class TFSEFileImpl(TFSEFile):
    """文件类实现"""

    db = MongoHelper("tfse_v0.21")

    def dict_to_return(self, **kwargs):
        return self.file_body

    def get_image(self):
        """"""

    def get_pdf(self):

        file_stream = self.db.find_file(
            "文件",
            self.file_bucket,
            self.file_id
        )

        if file_stream:
            self.file_body = Response(file_stream, content_type='application/pdf')
        else:
            self.file_body = None