Create ingest/citations.py

ingest/citations.py

@@ -0,0 +1,16 @@
+from reportlab.platypus import SimpleDocTemplate, Paragraph
+from reportlab.lib.styles import getSampleStyleSheet
+import tempfile
+
+def bluebook_pdf(results):
+    styles = getSampleStyleSheet()
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    doc = SimpleDocTemplate(tmp.name)
+
+    content = []
+    for r in results:
+        citation = f"{r['agency']}, {r['title']}, {r['url']}."
+        content.append(Paragraph(citation, styles["Normal"]))
+
+    doc.build(content)
+    return tmp.name