Handle PDF attachments without parent item

functions.py

@@ -108,28 +108,28 @@ def select_collection(collection, messages):
     )
 
 
-def search_attachments(id, type, key, collection, queries=[], limit=10):
-    try:
-        zot = zotero.Zotero(int(id), type.lower(), key)
-        searches = [zot.collection_items(
-            collection['key'],
-            q=q,
-            limit=limit,
-            itemType='attachment',
-            qmode='everything'
-        ) for q in queries]
-        attachments = [x for x in {item['key']: item for search in searches for item in search if item['data']
-                                   ['contentType'] == 'application/pdf'}.values()][:limit]
-
-        parents = set([a['data']['parentItem'] for a in attachments])
-
-        message = f"<div>✅ Found {len(attachments)} PDF {'attachments' if len(attachments) > 1 else 'attachment'} from {len(parents)} {'articles' if len(parents) > 1 else 'article'}.</div>" if len(
-            attachments) else "<div>❔ No results. Make sure to index your PDF attachments in Zotero.</div>"
-        return parents, attachments, message
-
-    except Exception as e:
-        message = f"<div>⚠️ Error occurred when searching in Zotero: {e}</div>"
-        return [], [], message
+# def search_attachments(id, type, key, collection, queries=[], limit=10):
+#     try:
+#         zot = zotero.Zotero(int(id), type.lower(), key)
+#         searches = [zot.collection_items(
+#             collection['key'],
+#             q=q,
+#             limit=limit,
+#             itemType='attachment',
+#             qmode='everything'
+#         ) for q in queries]
+#         attachments = [x for x in {item['key']: item for search in searches for item in search if item['data']
+#                                    ['contentType'] == 'application/pdf'}.values()][:limit]
+
+#         parents = set([a['data']['parentItem'] for a in attachments])
+
+#         message = f"<div>✅ Found {len(attachments)} PDF {'attachments' if len(attachments) > 1 else 'attachment'} from {len(parents)} {'articles' if len(parents) > 1 else 'article'}.</div>" if len(
+#             attachments) else "<div>❔ No results. Make sure to index your PDF attachments in Zotero.</div>"
+#         return parents, attachments, message
+
+#     except Exception as e:
+#         message = f"<div>⚠️ Error occurred when searching in Zotero: {e}</div>"
+#         return [], [], message
 
 
 def download_attachment(id, type, key, attachment):
@@ -211,7 +211,7 @@ def handle_submit(zot, collection_name, collections, question, messages):
         attachments = [x for x in {
             item['key']: item for search in searches for item in search if item['data']['contentType'] == 'application/pdf'}.values()][:10]
 
-        parents = set([a['data']['parentItem'] for a in attachments])
+        parents = set([a['data']['parentItem'] if 'parentItem' in a['data'] else a['key'] for a in attachments ])
         if len(attachments) > 0:
             messages.append(Message(
                 Icons.SUCCESS, f"Found {len(attachments)} PDF {'attachments' if len(attachments) > 1 else 'attachment'} from {len(parents)} {'articles' if len(parents) > 1 else 'article'}."))
@@ -251,13 +251,13 @@ def handle_submit(zot, collection_name, collections, question, messages):
         None,
     )
     for attachment in attachments:
-        parent_id = attachment["data"]["parentItem"]
+        parent_id = attachment["data"]["parentItem"] if "parentItem" in attachment["data"] else attachment["key"]        
         try:
             if parent_id in parents:
                 citation_dict[attachment["key"]] = parents[parent_id]
             else:
                 parent = zot.item(
-                    attachment["data"]["parentItem"], content="bib", style="nature")[0]
+                    parent_id, content="bib", style="nature")[0]
                 bib = f"""
                 {html.fragment_fromstring(parent).xpath("normalize-space(div[2])")}
                 <a href="{attachment['links']['alternate']['href']}" target="_blank" class="zotero-link">Open in Zotero</a>