1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
# Copyright (C) 2018 Robin Krahl <robin.krahl@ireas.org>
# SPDX-License-Identifier: MIT
import os
import os.path
import shutil
import bibtexparser
import bibtool.extract
import click
TYPE_FILE = click.Path(exists=True, file_okay=True, dir_okay=False)
TYPE_FILENAME = click.Path(exists=False, file_okay=True, dir_okay=False)
TYPE_DIR = click.Path(exists=True, file_okay=False, dir_okay=True)
def _get_bibtex_filename(bibtex_data, directory, output=None):
assert bibtex_data.entries
entry = bibtex_data.entries[0]
if not output:
output = os.path.join(directory, entry['ID'] + '.bib')
elif not os.path.isabs(output):
output = os.path.join(directory, output)
return output
@click.group()
def cli():
pass
@cli.command('extract')
@click.argument('filename', type=TYPE_FILE)
@click.option('--all/--single', '-a/-s', 'print_all', default=False)
def _extract(filename, print_all):
for doi in bibtool.extract.get_doi_generator(filename):
print(doi)
if not print_all:
return
@cli.command('fetch')
@click.argument('doi')
def _fetch(doi):
bibtex_data = bibtool.extract.get_bibtex_data(doi)
print(bibtexparser.dumps(bibtex_data))
@cli.command('import')
@click.argument('filename', type=TYPE_FILE)
@click.option('--directory', type=TYPE_DIR, default=os.getcwd())
@click.option('--delete/--no-delete', default=False)
@click.option('--doi', type=str, default=None)
def _import(filename, directory, delete, doi):
doi_generator = bibtool.extract.get_doi_generator(filename)
try:
doi = next(doi_generator)
except StopIteration:
raise Exception('Could not extract a DOI from {}'.format(filename))
bibtex_data = bibtool.extract.get_bibtex_data(doi)
entry = bibtex_data.entries[0]
click.echo('Found one Bibtex entry: ' + entry['ID'])
if click.confirm('Do you want to edit the entry?'):
edited_data = click.edit(bibtexparser.dumps(bibtex_data),
extension='.bib', require_save=None)
if not edited_data:
click.echo('Empty Bibtex data. Aborting.')
return
bibtex_data = bibtexparser.loads(edited_data)
if len(bibtex_data.entries) != 1:
raise Exception('The edited data must contain exactly one '
'Bibtex entry.')
entry = bibtex_data.entries[0]
click.confirm('Add {} to the repository?'.format(entry['ID']),
default=True, abort=True)
bibfilename = _get_bibtex_filename(bibtex_data, directory)
outfileext = os.path.splitext(filename)[1]
outfilename = os.path.join(directory, entry['ID'] + outfileext)
if os.path.exists(bibfilename) or os.path.exists(outfilename):
click.confirm('There is already a document with this ID in the '
'repository. Continue anyway?', abort=True)
with open(bibfilename, 'w') as f:
bibtexparser.dump(bibtex_data, f)
shutil.copy(filename, outfilename)
click.echo('Added {} to the repository.'.format(entry['ID']))
if delete:
os.remove(filename)
click.echo('Deleted {}.'.format(filename))
|
