def test_text_representation(self):
submission = Submission(requester_id=u'Alice',
queue_name=u'Wonderland',
xqueue_header=u'{}')
assert u"Submission from Alice for queue 'Wonderland'" in six.text_type(
submission)
def submit(request):
'''
Handle submissions to Xqueue from the LMS
'''
if request.method != 'POST':
transaction.commit()
return HttpResponse(
compose_reply(False, 'Queue requests should use HTTP POST'))
else:
# queue_name, xqueue_header, xqueue_body are all serialized
(request_is_valid, lms_callback_url, queue_name, xqueue_header,
xqueue_body) = _is_valid_request(request.POST)
if not request_is_valid:
log.error(
"Invalid queue submission from LMS: lms ip: {0}, request.POST: {1}"
.format(
get_request_ip(request),
request.POST,
))
transaction.commit()
return HttpResponse(
compose_reply(False, 'Queue request has invalid format'))
else:
if queue_name not in settings.XQUEUES:
transaction.commit()
return HttpResponse(
compose_reply(False, "Queue '%s' not found" % queue_name))
else:
# Limit DOS attacks by invalidating prior submissions from the
# same (user, module-id) pair as encoded in the lms_callback_url
_invalidate_prior_submissions(lms_callback_url)
# Check for file uploads
keys = dict() # For internal Xqueue use
urls = dict() # For external grader use
for filename in request.FILES.keys():
key = make_hashkey(xqueue_header + filename)
url = _upload(request.FILES[filename], queue_name, key)
keys.update({filename: key})
urls.update({filename: url})
urls_json = json.dumps(urls)
keys_json = json.dumps(keys)
if len(urls_json) > CHARFIELD_LEN_LARGE:
key = make_hashkey(xqueue_header +
json.dumps(list(request.FILES.keys())))
url = _upload_file_dict(urls, keys, queue_name, key)
keys = {"KEY_FOR_EXTERNAL_DICTS": key}
urls = {"URL_FOR_EXTERNAL_DICTS": url}
urls_json = json.dumps(urls)
keys_json = json.dumps(keys)
# Track the submission in the Submission database
submission = Submission(
requester_id=get_request_ip(request),
lms_callback_url=lms_callback_url[:128],
queue_name=queue_name,
xqueue_header=xqueue_header,
xqueue_body=xqueue_body,
s3_urls=urls_json,
s3_keys=keys_json)
submission.save()
transaction.commit(
) # Explicit commit to DB before inserting submission.id into queue
qcount = Submission.objects.get_queue_length(queue_name)
# For a successful submission, return the count of prior items
return HttpResponse(
compose_reply(success=True, content="%d" % qcount))
def test_keys(self):
keys = u'Alabama Florida Ohio Oklahoma West Virginia'
submission = Submission(s3_keys=keys)
assert submission.keys == keys