From ca36cadb4ae08832204fe6f4b0850b1d7bc8366a Mon Sep 17 00:00:00 2001
From: Marty Oehme <marty.oehme@gmail.com>
Date: Thu, 22 Dec 2022 18:33:33 +0100
Subject: [PATCH] Add option to enable writing to notes

---
 extract/extract.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/extract/extract.py b/extract/extract.py
index a265f11..452eb4d 100644
--- a/extract/extract.py
+++ b/extract/extract.py
@@ -45,13 +45,20 @@ class ExtractPlugin(PapersPlugin):
         """Allow the usage of the pubs git subcommand"""
         # TODO option for quiet/loud mode.
         # TODO option for ignoring missing documents or erroring.
-        # TODO option for writing to stdout or notes
         extract_parser = subparsers.add_parser(self.name, help=self.description)
         extract_parser.add_argument(
             "citekeys",
             nargs=argparse.REMAINDER,
             help="citekey(s) of the documents to extract from",
         )
+        # TODO option for writing to stdout or notes
+        extract_parser.add_argument(
+            "-w",
+            "--write",
+            help="write to individual notes instead of standard out. CAREFUL: OVERWRITES NOTES CURRENTLY",
+            action='store_true',
+            default=None,
+        )
         extract_parser.set_defaults(func=self.command)
 
     def command(self, conf, args):
@@ -63,7 +70,10 @@ class ExtractPlugin(PapersPlugin):
             return
         papers = self.gather_papers(citekeys)
         all_annotations = self.extract(papers)
-        self.to_stdout(all_annotations)
+        if args.write:
+            self.to_notes(conf, all_annotations, args.edit)
+        else:
+            self.to_stdout(all_annotations)
 
     def extract(self, papers):
         papers_annotated = []