More changes fixing the re-import scan functionality.

Author: CCoffie

dojo/test/views.py

@@ -438,19 +438,20 @@ def re_import_scan_results(request, tid):
                                                       )
 
                     if len(find) == 1:
-                        if find[0].mitigated:
+                        find = find[0]
+                        if find.mitigated:
                             # it was once fixed, but now back
-                            find[0].mitigated = None
-                            find[0].mitigated_by = None
-                            find[0].active = True
-                            find[0].verified = verified
-                            find[0].save()
+                            find.mitigated = None
+                            find.mitigated_by = None
+                            find.active = True
+                            find.verified = verified
+                            find.save()
                             note = Notes(entry="Re-activated by %s re-upload." % scan_type,
                                          author=request.user)
                             note.save()
-                            find[0].notes.add(note)
+                            find.notes.add(note)
                             reactivated_count += 1
-                        new_items.append(find[0].id)
+                        new_items.append(find.id)
                     else:
                         item.test = t
                         item.date = t.target_start
@@ -466,7 +467,7 @@ def re_import_scan_results(request, tid):
 
                         if hasattr(item, 'unsaved_req_resp') and len(item.unsaved_req_resp) > 0:
                             for req_resp in item.unsaved_req_resp:
-                                burp_rr = BurpRawRequestResponse(finding=item,
+                                burp_rr = BurpRawRequestResponse(finding=find,
                                                                  burpRequestBase64=req_resp["req"],
                                                                  burpResponseBase64=req_resp["resp"],
                                                                  )
@@ -489,7 +490,7 @@ def re_import_scan_results(request, tid):
                                                                          query=endpoint.query,
                                                                          fragment=endpoint.fragment,
                                                                          product=t.engagement.product)
-                            item.endpoints.add(ep)
+                            find.endpoints.add(ep)
 
                         if item.unsaved_tags is not None:
                             find.tags = item.unsaved_tags