From e3aacc4b15d39e3d0ff620e8eeabc7d4048d260a Mon Sep 17 00:00:00 2001
From: Marty Oehme <marty.oehme@gmail.com>
Date: Thu, 22 Dec 2022 23:21:13 +0100
Subject: [PATCH] Format file

---
 extract/extract.py | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/extract/extract.py b/extract/extract.py
index 221f0b5..2d652ed 100644
--- a/extract/extract.py
+++ b/extract/extract.py
@@ -40,9 +40,15 @@ class ExtractPlugin(PapersPlugin):
         # or `:: {annotation} :: {page} ::`
         # 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: ")
+        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"""
@@ -133,12 +139,16 @@ 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, self.highlight_prefix, self.note_prefix)
+                    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
 
-    def _retrieve_annotation_content(self, page, annotation, highlight_prefix = "> ", note_prefix = "Note: "):
+    def _retrieve_annotation_content(
+        self, page, annotation, highlight_prefix="> ", note_prefix="Note: "
+    ):
         """Gets the text content of an annotation.
 
         Returns the actual content of an annotation. Sometimes
@@ -151,7 +161,7 @@ class ExtractPlugin(PapersPlugin):
         written = page.get_textbox(annotation.rect).replace("\n", " ")
 
         # highlight with selection in note
-        if Levenshtein.ratio(content,written) > self.minimum_similarity:
+        if Levenshtein.ratio(content, written) > self.minimum_similarity:
             return f"{highlight_prefix}{content}"
         # an independent note, not a highlight
         elif content and not written:
@@ -175,7 +185,7 @@ class ExtractPlugin(PapersPlugin):
             if annotations:
                 output += f"------ {paper.citekey} ------\n\n"
                 for annot in annotations:
-                    output += f'{annot}\n\n'
+                    output += f"{annot}\n\n"
                 output += "\n"
         print(output)