mirror of
https://github.com/sartography/cr-connect-workflow.git
synced 2025-02-22 12:48:25 +00:00
4ec6e403f5
2. Reference Files and Spec Files are written to disk, they do not exist in the database at all.
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
from crc import session
|
|
from crc.api.common import ApiError
|
|
from crc.api.file import to_file_api
|
|
from crc.models.file import FileModel, FileSchema, CONTENT_TYPES
|
|
from crc.services.reference_file_service import ReferenceFileService
|
|
|
|
from flask import send_file
|
|
|
|
import io
|
|
import connexion
|
|
|
|
|
|
def get_reference_files():
|
|
"""Gets a list of all reference files"""
|
|
files = ReferenceFileService.get_reference_files()
|
|
return FileSchema(many=True).dump(files)
|
|
|
|
|
|
def get_reference_file_data(name):
|
|
file_extension = ReferenceFileService.get_extension(name)
|
|
content_type = CONTENT_TYPES[file_extension]
|
|
file_data = ReferenceFileService().get_data(name)
|
|
return send_file(
|
|
io.BytesIO(file_data),
|
|
attachment_filename=name,
|
|
mimetype=content_type,
|
|
cache_timeout=-1 # Don't cache these files on the browser.
|
|
)
|
|
|
|
def get_reference_file_info(name):
|
|
"""Return metadata for a reference file"""
|
|
return FileSchema().dump(ReferenceFileService.get_reference_file(name))
|
|
|
|
|
|
def update_reference_file_data(name):
|
|
"""Uses the file service to manage reference-files. They will be used in script tasks to compute values."""
|
|
if 'file' not in connexion.request.files:
|
|
raise ApiError('invalid_file',
|
|
'Expected a file named "file" in the multipart form request', status_code=400)
|
|
|
|
file = connexion.request.files['file']
|
|
name_extension = ReferenceFileService.get_extension(name)
|
|
file_extension = ReferenceFileService.get_extension(file.filename)
|
|
if name_extension != file_extension:
|
|
raise ApiError('invalid_file_type',
|
|
"The file you uploaded has an extension '%s', but it should have an extension of '%s' " %
|
|
(file_extension, name_extension))
|
|
|
|
return_file = ReferenceFileService.update_reference_file(file_name=name, binary_data=file.stream.read())
|
|
return FileSchema().dump(return_file)
|
|
|
|
def add_reference_file():
|
|
file = connexion.request.files['file']
|
|
file_model = ReferenceFileService.add_reference_file(file.filename, file.stream.read())
|
|
return FileSchema().dump(file_model)
|
|
|
|
|
|
def delete_reference_file(name):
|
|
ReferenceFileService().delete(name)
|
|
|
|
|