def prep_db():
"""
Prep database for test.
"""
clean_db()
# Add User
user = CRITsUser.create_user(username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL)
user.save()
# Add Source
add_new_source(TSRC, TUSER_NAME)
# Add Data Type
add_new_signature_type(TDT, TUSER_NAME)
def new_signature_type(request):
"""
Add a new Signature datatype to CRITs. Should be an AJAX POST.
:param request: Django request object (Required)
:type request: :class:`django.http.HttpRequest`
:returns: :class:`django.http.HttpResponse`
"""
if request.method == 'POST' and request.is_ajax():
form = NewSignatureTypeForm(request.POST)
analyst = request.user.username
if form.is_valid():
result = add_new_signature_type(form.cleaned_data['data_type'],
analyst)
if result:
message = {'message': '<div>Signature Type added successfully!</div>',
'success': True}
else:
message = {'message': '<div>Signature Type addition failed!</div>',
'success': False}
else:
message = {'form': form.as_table()}
return HttpResponse(json.dumps(message),
content_type="application/json")
return render_to_response('error.html',
{'error':'Expected AJAX POST'})
def new_signature_type(request):
"""
Add a new Signature datatype to CRITs. Should be an AJAX POST.
:param request: Django request object (Required)
:type request: :class:`django.http.HttpRequest`
:returns: :class:`django.http.HttpResponse`
"""
if request.method == 'POST' and request.is_ajax():
form = NewSignatureTypeForm(request.POST)
analyst = request.user.username
if form.is_valid():
result = add_new_signature_type(form.cleaned_data['data_type'],
analyst)
if result:
message = {
'message': '<div>Signature Type added successfully!</div>',
'success': True
}
else:
message = {
'message': '<div>Signature Type addition failed!</div>',
'success': False
}
else:
message = {'form': form.as_table()}
return HttpResponse(json.dumps(message),
content_type="application/json")
return render_to_response('error.html', {'error': 'Expected AJAX POST'})
def new_signature_type(request):
"""
Add a new Signature datatype to CRITs. Should be an AJAX POST.
:param request: Django request object (Required)
:type request: :class:`django.http.HttpRequest`
:returns: :class:`django.http.HttpResponse`
"""
if request.method == 'POST' and request.is_ajax():
form = NewSignatureTypeForm(request.POST)
user = request.user
if form.is_valid():
if user.has_access_to(GeneralACL.ADD_NEW_SIGNATURE_TYPE):
result = add_new_signature_type(form.cleaned_data['data_type'],
user.username)
if result:
message = {'message': '<div>Signature Type added successfully!</div>',
'success': True}
else:
message = {'message': '<div>Signature Type addition failed!</div>',
'success': False}
else:
message = {'message': '<div>User does not have permission to add signature type!</div>',
'success': False}
else:
message = {'form': form.as_table()}
return HttpResponse(json.dumps(message),
content_type="application/json")
return render(request, 'error.html',
{'error':'Expected AJAX POST'})
def new_signature_type(request):
"""
Add a new Signature datatype to CRITs. Should be an AJAX POST.
:param request: Django request object (Required)
:type request: :class:`django.http.HttpRequest`
:returns: :class:`django.http.HttpResponse`
"""
if request.method == 'POST' and request.is_ajax():
form = NewSignatureTypeForm(request.POST)
user = request.user
if form.is_valid():
if user.has_access_to(GeneralACL.ADD_NEW_SIGNATURE_TYPE):
result = add_new_signature_type(form.cleaned_data['data_type'],
user.username)
if result:
message = {
'message':
'<div>Signature Type added successfully!</div>',
'success': True
}
else:
message = {
'message':
'<div>Signature Type addition failed!</div>',
'success': False
}
else:
message = {
'message':
'<div>User does not have permission to add signature type!</div>',
'success': False
}
else:
message = {'form': form.as_table()}
return HttpResponse(json.dumps(message),
content_type="application/json")
return render(request, 'error.html', {'error': 'Expected AJAX POST'})
