def testChangingConfidence(self):
relationship1 = self.campaign1.relationships[0]
relationship2 = self.campaign2.relationships[0]
self.assertEqual(relationship1.rel_confidence, TRELATIONSHIP_CONFIDENCE)
self.assertEqual(relationship2.rel_confidence, TRELATIONSHIP_CONFIDENCE)
update_relationship_confidences(left_class=self.campaign1,
right_class=self.campaign2,
rel_type=TRELATIONSHIP_TYPE,
analyst=self.user2.username,
new_confidence=TRELATIONSHIP_NEW_CONFIDENCE)
self.assertEqual(relationship1.rel_confidence, TRELATIONSHIP_NEW_CONFIDENCE)
self.assertEqual(relationship2.rel_confidence, TRELATIONSHIP_NEW_CONFIDENCE)
def testChangingConfidence(self):
relationship1 = self.campaign1.relationships[0]
relationship2 = self.campaign2.relationships[0]
self.assertEqual(relationship1.rel_confidence,
TRELATIONSHIP_CONFIDENCE)
self.assertEqual(relationship2.rel_confidence,
TRELATIONSHIP_CONFIDENCE)
update_relationship_confidences(
left_class=self.campaign1,
right_class=self.campaign2,
rel_type=TRELATIONSHIP_TYPE,
analyst=self.user2.username,
new_confidence=TRELATIONSHIP_NEW_CONFIDENCE)
self.assertEqual(relationship1.rel_confidence,
TRELATIONSHIP_NEW_CONFIDENCE)
self.assertEqual(relationship2.rel_confidence,
TRELATIONSHIP_NEW_CONFIDENCE)
def update_relationship_confidence(request):
"""
Update relationship confidence. 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():
new_confidence = request.POST['new_confidence']
acls = get_acl_object(request.POST['my_type'])
user = request.user
if new_confidence not in ('unknown', 'low', 'medium', 'high'):
result = {'success': False, 'message': 'Unknown confidence level.'}
return HttpResponse(json.dumps(result),
content_type="application/json")
elif not user.has_access_to(acl.RELATIONSHIPS_EDIT):
result = {
'success': False,
'message':
'User does not have permission to edit relationship.'
}
else:
result = update_relationship_confidences(
left_type=request.POST['my_type'],
left_id=request.POST['my_value'],
right_type=request.POST['reverse_type'],
right_id=request.POST['dest_id'],
rel_type=request.POST['forward_relationship'],
rel_date=request.POST['relationship_date'],
analyst=request.user.username,
new_confidence=new_confidence)
if result['success']:
message = "Successfully updated relationship: %s" % result[
'message']
result = {'success': True, 'message': message}
else:
message = "Error updating relationship: %s" % result['message']
result = {'success': False, 'message': message}
return HttpResponse(json.dumps(result),
content_type="application/json")
else:
error = "Expected AJAX POST"
return render(request, "error.html", {"error": error})
def update_relationship_confidence(request):
"""
Update relationship confidence. 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():
new_confidence = request.POST['new_confidence']
acls = get_acl_object(request.POST['my_type'])
user = request.user
if new_confidence not in ('unknown', 'low', 'medium', 'high'):
result = {'success': False,
'message': 'Unknown confidence level.'}
return HttpResponse(json.dumps(result), content_type="application/json")
elif not user.has_access_to(acl.RELATIONSHIPS_EDIT):
result = {'success': False,
'message': 'User does not have permission to edit relationship.'}
else:
result = update_relationship_confidences(left_type=request.POST['my_type'],
left_id=request.POST['my_value'],
right_type=request.POST['reverse_type'],
right_id=request.POST['dest_id'],
rel_type=request.POST['forward_relationship'],
rel_date=request.POST['relationship_date'],
analyst=request.user.username,
new_confidence=new_confidence)
if result['success']:
message = "Successfully updated relationship: %s" % result['message']
result = {'success': True, 'message': message}
else:
message = "Error updating relationship: %s" % result['message']
result = {'success': False, 'message': message}
return HttpResponse(json.dumps(result), content_type="application/json")
else:
error = "Expected AJAX POST"
return render_to_response("error.html",
{"error" : error },
RequestContext(request))
def update_relationship_confidence(request):
"""
Update relationship confidence. 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():
new_confidence = request.POST['new_confidence']
if new_confidence not in ('unknown', 'low', 'medium', 'high'):
result = {'success': False, 'message': 'Unknown confidence level.'}
return HttpResponse(json.dumps(result),
mimetype="application/json")
else:
results = update_relationship_confidences(
left_type=request.POST['my_type'],
left_id=request.POST['my_value'],
right_type=request.POST['reverse_type'],
right_id=request.POST['dest_id'],
rel_type=request.POST['forward_relationship'],
rel_date=request.POST['relationship_date'],
analyst=request.user.username,
new_confidence=new_confidence)
if results['success']:
message = "Successfully updated relationship: %s" % results[
'message']
result = {'success': True, 'message': message}
else:
message = "Error updating relationship: %s" % results['message']
result = {'success': False, 'message': message}
return HttpResponse(json.dumps(result), mimetype="application/json")
else:
error = "Expected AJAX POST"
return render_to_response("error.html", {"error": error},
RequestContext(request))
