Tests

laws-africa · Jan 25, 2024 · c72d88a · c72d88a
1 parent c39ba95
commit c72d88a
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/peachjam/views/generic_views.py b/peachjam/views/generic_views.py
@@ -276,16 +276,17 @@ def get_notices(self):
 
     def prefix_images(self, document):
         """Rewrite image URLs so that we can server them correctly."""
-        root = html.fromstring(document.content_html)
+        if document.content_html and not document.content_html_is_akn:
+            root = html.fromstring(document.content_html)
 
-        for img in root.xpath(".//img[@src]"):
-            src = img.attrib["src"]
-            if not src.startswith("/") and not src.startswith("data:"):
-                img.attrib["src"] = (
-                    document.expression_frbr_uri + "/media/" + img.attrib["src"]
-                )
+            for img in root.xpath(".//img[@src]"):
+                src = img.attrib["src"]
+                if not src.startswith("/") and not src.startswith("data:"):
+                    img.attrib["src"] = (
+                        document.expression_frbr_uri + "/media/" + img.attrib["src"]
+                    )
 
-        document.content_html = html.tostring(root, encoding="unicode")
+            document.content_html = html.tostring(root, encoding="unicode")
 
 
 class CSRFTokenView(View):