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'})