Marty/papis-marvin
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Extract book comparison into function

Browse source
This commit is contained in:
Marty Oehme 2023-02-21 14:47:57 +01:00
parent d78c5dd64a
commit 2db5ce4cac
Signed by: Marty
GPG key ID: EDBF2ED917B2EF6A
1 changed files with 13 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
papis-marvin
View file

@ -12,8 +12,6 @@ import papis.database
import re import re
import subprocess import subprocess
# use argument passed to command as file or default file here
def format_entry(row) -> str: def format_entry(row) -> str:
text = f"> {row['HighlightText']}" text = f"> {row['HighlightText']}"
@ -26,6 +24,17 @@ def format_entry(row) -> str:
return text return text
_old_title = ""
def is_same_book(title):
global _old_title
same = _old_title == title
_old_title = title
if same:
return True
return False
def main(fpath, db): def main(fpath, db):
notes = {} notes = {}
with open(fpath) as f: with open(fpath) as f:
@ -34,15 +43,12 @@ def main(fpath, db):
csv = csv.DictReader(f) csv = csv.DictReader(f)
title_strip_pattern = re.compile(r"([^\s\w]|_)+") title_strip_pattern = re.compile(r"([^\s\w]|_)+")
old_title = ""
note_file = "" note_file = ""
for row in csv: for row in csv:
title_stripped = title_strip_pattern.sub("", row["Title"]) title_stripped = title_strip_pattern.sub("", row["Title"])
# switch to next book # switch to next book
if old_title != row["Title"]: if not is_same_book(row["Title"]):
old_title = row["Title"]
documents = db.query( documents = db.query(
f"author:({row['Author']}) title:({title_stripped})" f"author:({row['Author']}) title:({title_stripped})"
) )
@ -89,6 +95,7 @@ def main(fpath, db):
if __name__ == "__main__": if __name__ == "__main__":
# use argument passed to command as file or default file here
fpath = ( fpath = (
sys.argv[1] sys.argv[1]
if len(sys.argv) > 1 if len(sys.argv) > 1