tfse-app-api-v0.2/Implements/Others/TFSEFileImpl.py

30 lines
639 B
Python

from flask import Response
from DBHelper.MongoHelper import MongoHelper
from ObjectsCommon.Others.FECRFile import FECRFile
class TFSEFileImpl(FECRFile):
"""文件类实现"""
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