.gitignore

@@ -1 +0,0 @@
-testfile.csv

papis-marvin

@@ -5,28 +5,13 @@
 # the iOS application 'Marvin Reader'. In the app, export your
 # annotations as 'csv' format and then point the script to the
 # resulting file.
-# https://git.martyoeh.me/Marty/papis-marvin
-import os
 import sys
-import re
-import logging
-from typing import Dict
+from typing import List
 import papis.api
-import papis.pick
-import papis.format
-import papis.commands.edit
 import papis.commands.list
-import papis.commands.add
-import papis.notes
-import papis.config
 import papis.database
-import isbnlib
-import papis.isbn
-
-logger = logging.getLogger("marvin")
-logger.setLevel(logging.DEBUG)
-
-DEFAULT_CSV_PATH = "/home/marty/Nextcloud/Personal/Backups/Journal.csv"
+import re
+import subprocess
 
 
 def main(fpath, db):
@@ -39,74 +24,59 @@ def main(fpath, db):
     write_to_files(notes)
 
 
-def get_all_annotations(db, csv) -> Dict:
+def get_all_annotations(db, csv):
     notes = {}
     note_file = ""
     for row in csv:
+        title_stripped = strip_book_title(row["Title"])
+
         # switch to next book
         if not is_same_book(row["Title"]):
-            doc = get_document(db, row["Author"], row["Title"])
-            if not doc:
+            documents = get_documents(db, row["Author"], title_stripped)
+            if not documents:
+                print(
+                    f"No papis entry found for Marvin entry - {row['Author']}: {title_stripped}.\nPlease manually create."
+                )
                 continue
-            note_file = get_notefile(db, doc)
+            note_file = get_notefile(documents)
+            if not note_file:
+                print(
+                    f"Found reference entry but no note file for - {row['Author']}: {title_stripped}."
+                )
+                manual_papis_entry()
 
         text = format_entry(row)
 
         if note_file and text:
-            if note_file not in notes.keys():
+            if not note_file in notes.keys():
                 notes[note_file] = []
             notes[note_file].append(text)
     return notes
 
 
-def get_document(db, author, title):
-    res = query_document(db, author, title)
-    if not res:
-        add_to_database(author, title)
-        res = query_document(db, author, title)
-    if not res:
-        logger.warning(f"Nothing found for {author}: {title}.\nPlease create manually.")
+# TODO Implement manual note creation
+def manual_papis_entry():
+    if input(f"Create note file now? [y/N] ") == "y":
+        # i think instead we need to run the papis.edit cmd with note bool true and maybe editor set to none?
+        output = subprocess.run(
+            ["papis", "edit", "-n"], capture_output=True, shell=True
+        )
+        print(output)
+        print("NOT IMPLEMENTED: Please create note file manually.")
         return
-    return res
-
-
-# TODO warn user/ let him pick with picker if multiple docs found
-def query_document(db, author, title):
-    title = strip_string(title)
-    for query in [f"author:({author}) title:({title})"]:
-        print(f"query: {query}")
-        res = db.query(query)
-        if len(res) >= 1:
-            return res[0]
-
-
-def add_to_database(author, title, confirm=True, edit=False):
-    logger.info(f"Searching - '{title} {author}'")
-    data = None
-    try:
-        data = papis.isbn.get_data(f"{title}")
-    except isbnlib.ISBNLibException as e:
-        logger.error(e)
     else:
-        logger.warning(f"Found: {data}")
-    if data:
-        papis_data = papis.isbn.data_to_papis(data[0])
-        papis.commands.add.run([], data=papis_data, confirm=confirm, edit=edit)
+        return
 
 
-def get_notefile(db, document) -> str | None:
-    if not document.has("notes"):
-        notes_name = papis.config.getstring("notes-name")
-        document["notes"] = papis.format.format(notes_name, document)
-        document.save()
-        db.update(document)
+def get_documents(db, author, title) -> List:
+    return db.query(f"author:({author}) title:({title})")
 
-    notes_path = os.path.join(str(document.get_main_folder()), document["notes"])
 
-    if not os.path.exists(notes_path):
-        # TODO reimplement logger: logger.debug("Creating '%s'", notes_path)
-        papis.notes.notes_path_ensured(document)
-    return notes_path
+def get_notefile(document) -> str | None:
+    note_file = papis.commands.list.run(notes=True, documents=document)
+    if not note_file:
+        return
+    return str(note_file[0])
 
 
 # TODO implement custom formatting (akin to pubs-extract)
@@ -134,30 +104,34 @@ def is_same_book(title):
     return False
 
 
-def write_to_files(notes: Dict):
+def write_to_files(notes):
     # write to notes
     for f, entries in notes.items():
         if f:
             with open(f, "a") as note:
-                logger.info(f"Editing {f}...")
+                print(f"Editing {f}...")
                 num_added = 0
                 for entry in entries:
                     with open(f) as noteread:
                         if entry not in noteread.read():
                             note.write(f"{entry}\n\n")
                             num_added += 1
-                logger.info(f"Added {num_added} entries to it.")
+                print(f"Added {num_added} entries to it.")
 
 
-strip_pattern = re.compile(r"([^\s\w]|_)+\w*")
+title_strip_pattern = re.compile(r"([^\s\w]|_)+")
 
 
-def strip_string(title) -> str:
-    return strip_pattern.sub("", title)
+def strip_book_title(title) -> str:
+    return title_strip_pattern.sub("", title)
 
 
 if __name__ == "__main__":
     # use argument passed to command as file or default file here
-    fpath = sys.argv[1] if len(sys.argv) > 1 else DEFAULT_CSV_PATH
+    fpath = (
+        sys.argv[1]
+        if len(sys.argv) > 1
+        else "/home/marty/Nextcloud/Personal/Backups/Journal.csv"
+    )
 
     main(fpath, papis.database.get())