67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
from flask import g
|
|
|
|
from crc.api.common import ApiError
|
|
from crc.services.data_store_service import DataStoreBase
|
|
from crc.scripts.script import Script
|
|
from crc.services.document_service import DocumentService
|
|
from crc.services.file_service import FileService
|
|
|
|
|
|
class FileDataSet(Script, DataStoreBase):
|
|
def get_description(self):
|
|
return """Sets data the data store - takes three keyword arguments arguments: 'file_id' and 'key' and 'value'"""
|
|
|
|
def do_task_validate_only(self, task, study_id, workflow_id, *args, **kwargs):
|
|
if self.validate_kw_args(**kwargs):
|
|
myargs = [kwargs['key'],kwargs['value']]
|
|
fileid = kwargs['file_id']
|
|
del(kwargs['file_id'])
|
|
return True
|
|
|
|
def validate_kw_args(self, **kwargs):
|
|
if kwargs.get('key', None) is None:
|
|
raise ApiError(code="missing_argument",
|
|
message=f"The 'file_data_get' script requires a keyword argument of 'key'")
|
|
if kwargs.get('file_id', None) is None:
|
|
raise ApiError(code="missing_argument",
|
|
message=f"The 'file_data_get' script requires a keyword argument of 'file_id'")
|
|
if kwargs.get('value', None) is None:
|
|
raise ApiError(code="missing_argument",
|
|
message=f"The 'file_data_get' script requires a keyword argument of 'value'")
|
|
|
|
if kwargs['key'] == 'irb_code' and not DocumentService.is_allowed_document(kwargs.get('value')):
|
|
raise ApiError("invalid_form_field_key",
|
|
"When setting an irb_code, the form field id must match a known document in the "
|
|
"irb_docunents.xslx reference file. This code is not found in that file '%s'" %
|
|
kwargs.get('value'))
|
|
|
|
return True
|
|
|
|
|
|
def do_task(self, task, study_id, workflow_id, *args, **kwargs):
|
|
if self.validate_kw_args(**kwargs):
|
|
myargs = [kwargs['key'],kwargs['value']]
|
|
|
|
try:
|
|
fileid = int(kwargs['file_id'])
|
|
except:
|
|
raise ApiError("invalid_file_id",
|
|
"Attempting to update DataStore for an invalid fileid '%s'" % kwargs['file_id'])
|
|
|
|
del(kwargs['file_id'])
|
|
if kwargs['key'] == 'irb_code':
|
|
irb_doc_code = kwargs['value']
|
|
FileService.update_irb_code(fileid,irb_doc_code)
|
|
|
|
|
|
return self.set_data_common(task.id,
|
|
None,
|
|
None,
|
|
workflow_id,
|
|
None,
|
|
'file_data_set',
|
|
fileid,
|
|
*myargs,
|
|
**kwargs)
|
|
|