examiner: Fix missing exam urls in organize

JakobGM · Jan 21, 2019 · e19435a · e19435a
1 parent d6ee614
commit e19435a
Showing 1 changed file with 15 additions and 12 deletions.
diff --git a/examiner/models.py b/examiner/models.py
@@ -134,18 +134,21 @@ def organize(self):
                 docinfo.language or 'Ukjent',
                 [],
             )
-            try:
-                url = DocumentInfoSource.objects.filter(
-                    document_info=docinfo,
-                    verified_by__isnull=False,
-                ).first().pdf.hosted_at.filter(dead_link=False).first()
-            except (AttributeError, DocumentInfoSource.DoesNotExist):
-                url = DocumentInfoSource.objects.filter(
-                    document_info=docinfo,
-                ).first().pdf.hosted_at.filter(dead_link=False).first()
-
-            if url not in urls:
-                urls.append(url)
+            for pdf in docinfo.pdfs.all():
+                url = pdf.hosted_at.filter(dead_link=False).first()
+                if url not in urls:
+                    urls.append(url)
+
+            # TODO: Make this type of logic work
+            # try:
+            #     url = DocumentInfoSource.objects.filter(
+            #         document_info=docinfo,
+            #         verified_by__isnull=False,
+            #     ).first().pdf.hosted_at.filter(dead_link=False).first()
+            # except (AttributeError, DocumentInfoSource.DoesNotExist):
+            #     url = DocumentInfoSource.objects.filter(
+            #         document_info=docinfo,
+            #     ).first().pdf.hosted_at.filter(dead_link=False).first()
 
         return organization