def obj_create(self, bundle, **kwargs):
"""
Handles creating service result entries through the API.
:param bundle: Bundle containing the service results to add.
:type bundle: Tastypie Bundle object.
:returns: Bundle object.
:raises BadRequest: If necessary data is not provided or creation fails.
"""
analyst = bundle.request.user.username
object_type = bundle.data.get('object_type', None)
object_id = bundle.data.get('object_id', None)
analysis_id = bundle.data.get('analysis_id', None)
result = bundle.data.get('result', None)
result_type = bundle.data.get('result_type', None)
result_subtype = bundle.data.get('result_subtype', None)
log_message = bundle.data.get('log_message', None)
log_level = bundle.data.get('log_level', 'info')
status = bundle.data.get('status', None)
finish = bundle.data.get('finish', False)
success = True
message = ""
if not object_type or not object_id or not analysis_id:
raise BadRequest(
'Need an object type, object id, and analysis id.')
if result:
if not result_type or not result_subtype:
raise BadRequest(
'When adding a result, also need type and subtype')
result = add_result(object_type, object_id, analysis_id, result,
result_type, result_subtype, analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
if log_message:
result = add_log(object_type, object_id, analysis_id, log_message,
log_level, analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
if finish:
result = finish_task(object_type, object_id, analysis_id, status,
analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
if not success:
raise BadRequest(message)
else:
return bundle
def obj_create(self, bundle, **kwargs):
"""
Handles creating service result entries through the API.
:param bundle: Bundle containing the service results to add.
:type bundle: Tastypie Bundle object.
:returns: Bundle object.
:raises BadRequest: If necessary data is not provided or creation fails.
"""
analyst = bundle.request.user.username
object_type = bundle.data.get('object_type', None)
object_id = bundle.data.get('object_id', None)
analysis_id = bundle.data.get('analysis_id', None)
result = bundle.data.get('result', None)
result_type = bundle.data.get('result_type', None)
result_subtype = bundle.data.get('result_subtype', None)
log_message = bundle.data.get('log_message', None)
log_level = bundle.data.get('log_level', 'info')
status = bundle.data.get('status', None)
finish = bundle.data.get('finish', False)
success = True
message = ""
if not object_type or not object_id or not analysis_id:
raise BadRequest('Need an object type, object id, and analysis id.')
if result:
if not result_type or not result_subtype:
raise BadRequest('When adding a result, also need type and subtype')
result = add_result(object_type, object_id, analysis_id,
result, result_type, result_subtype, analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
if log_message:
result = add_log(object_type, object_id, analysis_id,
log_message, log_level, analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
if finish:
result = finish_task(object_type, object_id, analysis_id,
status, analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
if not success:
raise BadRequest(message)
else:
return bundle
def obj_create(self, bundle, **kwargs):
"""
Handles creating service result entries through the API.
:param bundle: Bundle containing the service results to add.
:type bundle: Tastypie Bundle object.
:returns: HttpResponse.
"""
analyst = bundle.request.user.username
object_type = bundle.data.get('object_type', None)
object_id = bundle.data.get('object_id', None)
analysis_id = bundle.data.get('analysis_id', None)
result = bundle.data.get('result', None)
result_type = bundle.data.get('result_type', None)
result_subtype = bundle.data.get('result_subtype', None)
log_message = bundle.data.get('log_message', None)
log_level = bundle.data.get('log_level', 'info')
status = bundle.data.get('status', None)
finish = bundle.data.get('finish', False)
success = True
message = ""
content = {'return_code': 1, 'type': object_type}
if not object_type or not object_id or not analysis_id:
content[
'message'] = 'Need an object type, object id, and analysis id.'
self.crits_response(content)
if result:
if not result_type or not result_subtype:
content[
'message'] = 'When adding a result, also need type and subtype'
self.crits_response(content)
result = add_result(object_type, object_id, analysis_id, result,
result_type, result_subtype, analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
if log_message:
result = add_log(object_type, object_id, analysis_id, log_message,
log_level, analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
if finish:
result = finish_task(object_type, object_id, analysis_id, status,
analyst)
if not result['success']:
message += ", %s" % result['message']
success = False
content['message'] = message
content['id'] = object_id
rname = self.resource_name_from_type(object_type)
url = reverse('api_dispatch_detail',
kwargs={
'resource_name': rname,
'api_name': 'v1',
'pk': object_id
})
content['url'] = url
if success:
content['return_code'] = 0
self.crits_response(content)
def obj_create(self, bundle, **kwargs):
"""
Handles creating service result entries through the API.
:param bundle: Bundle containing the service results to add.
:type bundle: Tastypie Bundle object.
:returns: HttpResponse.
"""
user = bundle.request.user
object_type = bundle.data.get('object_type', None)
object_id = bundle.data.get('object_id', None)
analysis_id = bundle.data.get('analysis_id', None)
result = bundle.data.get('result', None)
result_type = bundle.data.get('result_type', None)
result_subtype = bundle.data.get('result_subtype', None)
result_is_batch = bundle.data.get('result_is_batch', False)
log_message = bundle.data.get('log_message', None)
log_level = bundle.data.get('log_level', 'info')
status = bundle.data.get('status', None)
finish = bundle.data.get('finish', False)
success = True
message = ""
content = {'return_code': 1,
'type': object_type}
if not object_type or not object_id or not analysis_id:
content['message'] = 'Need an object type, object id, and analysis id.'
self.crits_response(content)
if result:
if not result_type or not result_subtype:
content['message'] = 'When adding a result, also need type and subtype'
self.crits_response(content)
if not result_is_batch:
result = add_result(object_type, object_id, analysis_id,
result, result_type, result_subtype, user)
else:
result_list = json.loads(result)
result_type_list = json.loads(result_type)
result_subtype_list = json.loads(result_subtype)
if not (len(result_list) == len(result_type_list) == len(result_subtype_list)):
content['message'] = 'When adding results in batch result, result_type, and result_subtype must have the same length!'
self.crits_response(content)
result = add_results(object_type, object_id, analysis_id, result_list,
result_type_list, result_subtype_list, user)
if not result['success']:
message += ", %s" % result['message']
success = False
if log_message:
result = add_log(object_type, object_id, analysis_id,
log_message, log_level, user)
if not result['success']:
message += ", %s" % result['message']
success = False
if finish:
result = finish_task(object_type, object_id, analysis_id,
status, user)
if not result['success']:
message += ", %s" % result['message']
success = False
content['message'] = message
content['id'] = object_id
rname = self.resource_name_from_type(object_type)
url = reverse('api_dispatch_detail',
kwargs={'resource_name': rname,
'api_name': 'v1',
'pk': object_id})
content['url'] = url
if success:
content['return_code'] = 0
self.crits_response(content)
