summaryrefslogtreecommitdiff
path: root/bibtool/cli.py
blob: 43d6c2177a0c618395a27c06db952fdd3d81393a (plain)
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
# 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_DIR = click.Path(exists=True, file_okay=False, dir_okay=True)


@click.group()
def cli():
    pass


@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)
def _import(filename, directory, delete):
    bibtex_data = bibtool.extract.get_bibtex_data(filename)

    if not bibtex_data.entries:
        raise Exception('Did not find any Bibtex entry.')
    if len(bibtex_data.entries) > 1:
        raise Exception('Found more than one Bibtex entries.')

    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 = os.path.join(directory, entry['ID'] + '.bib')
    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))