Add task_spec_name to FileService.add_workflow_file, and script that calls it.

This commit is contained in:
mike cullerton 2021-08-26 10:40:47 -04:00
parent 4c41011299
commit 506157d58a
2 changed files with 5 additions and 2 deletions

View File

@ -38,6 +38,7 @@ Takes two arguments:
file_name = args[0]
irb_doc_code = args[1]
FileService.add_workflow_file(workflow_id=workflow_id,
task_spec_name=task.get_name(),
name=file_name,
content_type=CONTENT_TYPES['docx'],
binary_data=final_document_stream.read(),

View File

@ -85,16 +85,18 @@ class FileService(object):
@staticmethod
def add_workflow_file(workflow_id, irb_doc_code, name, content_type, binary_data):
def add_workflow_file(workflow_id, irb_doc_code, task_spec_name, name, content_type, binary_data):
file_model = session.query(FileModel)\
.filter(FileModel.workflow_id == workflow_id)\
.filter(FileModel.name == name)\
.filter(FileModel.name == name) \
.filter(FileModel.task_spec == task_spec_name) \
.filter(FileModel.irb_doc_code == irb_doc_code).first()
if not file_model:
file_model = FileModel(
workflow_id=workflow_id,
name=name,
task_spec=task_spec_name,
irb_doc_code=irb_doc_code
)
return FileService.update_file(file_model, binary_data, content_type)