97 lines
3.9 KiB
Python
97 lines
3.9 KiB
Python
from crc import session
|
|
from crc.api.common import ApiError
|
|
from crc.models.file import FileSchema, FileType
|
|
from crc.services.spec_file_service import SpecFileService
|
|
|
|
from flask import send_file
|
|
|
|
import io
|
|
import connexion
|
|
|
|
from crc.services.workflow_spec_service import WorkflowSpecService
|
|
|
|
def get_files(spec_id, include_libraries=False):
|
|
if spec_id is None:
|
|
raise ApiError(code='missing_spec_id',
|
|
message='Please specify the workflow_spec_id.')
|
|
workflow_spec_service = WorkflowSpecService()
|
|
workflow_spec = workflow_spec_service.get_spec(spec_id)
|
|
if workflow_spec is None:
|
|
raise ApiError(code='unknown_spec',
|
|
message=f'Unknown Spec: {spec_id}')
|
|
files = SpecFileService.get_files(workflow_spec,
|
|
include_libraries=include_libraries)
|
|
return FileSchema(many=True).dump(files)
|
|
|
|
|
|
def get_file(spec_id, file_name):
|
|
workflow_spec_service = WorkflowSpecService()
|
|
workflow_spec = workflow_spec_service.get_spec(spec_id)
|
|
files = SpecFileService.get_files(workflow_spec, file_name)
|
|
if len(files) == 0:
|
|
raise ApiError(code='unknown file',
|
|
message=f'No information exists for file {file_name}'
|
|
f' it does not exist in workflow {spec_id}.', status_code=404)
|
|
return FileSchema().dump(files[0])
|
|
|
|
|
|
def add_file(spec_id):
|
|
workflow_spec_service = WorkflowSpecService()
|
|
workflow_spec = workflow_spec_service.get_spec(spec_id)
|
|
file = connexion.request.files['file']
|
|
file = SpecFileService.add_file(workflow_spec, file.filename, file.stream.read())
|
|
if not workflow_spec.primary_process_id and file.type == FileType.bpmn:
|
|
SpecFileService.set_primary_bpmn(workflow_spec, file.name)
|
|
workflow_spec_service.update_spec(workflow_spec)
|
|
return FileSchema().dump(file)
|
|
|
|
|
|
def update(spec_id, file_name, is_primary):
|
|
workflow_spec_service = WorkflowSpecService()
|
|
workflow_spec = workflow_spec_service.get_spec(spec_id)
|
|
files = SpecFileService.get_files(workflow_spec, file_name)
|
|
if len(files) < 1:
|
|
raise ApiError(code='unknown file',
|
|
message=f'No information exists for file {file_name}'
|
|
f' it does not exist in workflow {spec_id}.')
|
|
file = files[0]
|
|
if is_primary:
|
|
SpecFileService.set_primary_bpmn(workflow_spec, file_name)
|
|
workflow_spec_service.update_spec(workflow_spec)
|
|
return FileSchema().dump(file)
|
|
|
|
|
|
def update_data(spec_id, file_name):
|
|
workflow_spec_service = WorkflowSpecService()
|
|
workflow_spec_model = workflow_spec_service.get_spec(spec_id)
|
|
if workflow_spec_model is None:
|
|
raise ApiError(code='missing_spec',
|
|
message=f'The workflow spec for id {spec_id} does not exist.')
|
|
file_data = connexion.request.files['file']
|
|
file = SpecFileService.update_file(workflow_spec_model, file_name, file_data.stream.read())
|
|
return FileSchema().dump(file)
|
|
|
|
|
|
def get_data(spec_id, file_name):
|
|
workflow_spec_service = WorkflowSpecService()
|
|
workflow_spec = workflow_spec_service.get_spec(spec_id)
|
|
file_data = SpecFileService.get_data(workflow_spec, file_name)
|
|
if file_data is not None:
|
|
file_info = SpecFileService.get_files(workflow_spec, file_name)[0]
|
|
return send_file(
|
|
io.BytesIO(file_data),
|
|
attachment_filename=file_name,
|
|
mimetype=file_info.content_type,
|
|
cache_timeout=-1 # Don't cache these files on the browser.
|
|
)
|
|
else:
|
|
raise ApiError(code='missing_data_model',
|
|
message=f'The data model for file {file_name}'
|
|
f' does not exist in workflow {spec_id}.')
|
|
|
|
|
|
def delete(spec_id, file_name):
|
|
workflow_spec_service = WorkflowSpecService()
|
|
workflow_spec = workflow_spec_service.get_spec(spec_id)
|
|
SpecFileService.delete_file(workflow_spec, file_name)
|