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

Format file

Browse source
This commit is contained in:
Marty Oehme 2022-12-22 20:32:06 +01:00
parent 627233a750
commit 223771f02d
Signed by: Marty
GPG key ID: 73BA40D5AFAF49C9
1 changed files with 8 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
extract/extract.py
View file

@ -14,7 +14,6 @@ from pubs.utils import resolve_citekey_list
from pubs.content import write_file from pubs.content import write_file
class ExtractPlugin(PapersPlugin): class ExtractPlugin(PapersPlugin):
"""Make the pubs repository also a git repository. """Make the pubs repository also a git repository.
@ -58,14 +57,14 @@ class ExtractPlugin(PapersPlugin):
"-w", "-w",
"--write", "--write",
help="write to individual notes instead of standard out. CAREFUL: OVERWRITES NOTES CURRENTLY", help="write to individual notes instead of standard out. CAREFUL: OVERWRITES NOTES CURRENTLY",
action='store_true', action="store_true",
default=None, default=None,
) )
extract_parser.add_argument( extract_parser.add_argument(
"-e", "-e",
"--edit", "--edit",
help="open each note in editor for manual editing after extracting annotations to it", help="open each note in editor for manual editing after extracting annotations to it",
action='store_true', action="store_true",
default=False, default=False,
) )
extract_parser.set_defaults(func=self.command) extract_parser.set_defaults(func=self.command)
@ -125,10 +124,10 @@ class ExtractPlugin(PapersPlugin):
paper = contents[0] paper = contents[0]
annotations = contents[1] annotations = contents[1]
if annotations: if annotations:
output+=f"{paper.citekey}\n" output += f"{paper.citekey}\n"
for annot in annotations: for annot in annotations:
output+=f'> "{annot}"\n' output += f'> "{annot}"\n'
output+="\n" output += "\n"
print(output) print(output)
def _to_notes(self, annotated_papers, note_extension="txt", edit=False): def _to_notes(self, annotated_papers, note_extension="txt", edit=False):
@ -136,13 +135,11 @@ class ExtractPlugin(PapersPlugin):
paper = contents[0] paper = contents[0]
annotations = contents[1] annotations = contents[1]
if annotations: if annotations:
notepath = self.broker.real_notepath( notepath = self.broker.real_notepath(paper.citekey, note_extension)
paper.citekey, note_extension
)
output = "# Annotations\n\n" output = "# Annotations\n\n"
for annotation in annotations: for annotation in annotations:
output+=f"> {annotation}\n\n" output += f"> {annotation}\n\n"
write_file(notepath, output, 'w') write_file(notepath, output, "w")
if edit is True: if edit is True:
self.ui.edit_file(notepath, temporary=False) self.ui.edit_file(notepath, temporary=False)
NoteEvent(paper.citekey).send() NoteEvent(paper.citekey).send()
@ -157,4 +154,3 @@ def modify_event(event):
if all_annotations[0][1]: if all_annotations[0][1]:
plg._to_notes(all_annotations, plg.conf["main"]["note_extension"]) plg._to_notes(all_annotations, plg.conf["main"]["note_extension"])
plg.ui.info(f"Imported {event.citekey} annotations.") plg.ui.info(f"Imported {event.citekey} annotations.")