Marty/pubs-extract
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Activity

Add configurable note and quote prefixes

Browse source
This commit is contained in:
Marty Oehme 2022-12-22 23:09:43 +01:00
parent f103cc51b5
commit 8f01b93de2
Signed by: Marty
GPG key ID: 73BA40D5AFAF49C9
2 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
extract/extract.py
View file

@ -41,6 +41,8 @@ class ExtractPlugin(PapersPlugin):
# and so on
self.on_import = conf["plugins"].get("extract", {}).get("on_import", False)
self.minimum_similarity = float(conf["plugins"].get("extract", {}).get("minimum_similarity", 0.75))
self.highlight_prefix = conf["plugins"].get("extract", {}).get("quote_prefix", "> ")
self.note_prefix = conf["plugins"].get("extract", {}).get("note_prefix", "Note: ")
def update_parser(self, subparsers, conf):
"""Allow the usage of the pubs extract subcommand"""
@ -131,7 +133,7 @@ class ExtractPlugin(PapersPlugin):
with fitz.Document(filename) as doc:
for page in doc:
for annot in page.annots():
content = self._retrieve_annotation_content(page, annot)
content = self._retrieve_annotation_content(page, annot, self.highlight_prefix, self.note_prefix)
if content:
annotations.append(f"[{(page.number or 0) + 1}] {content}")
return annotations