aboutsummaryrefslogtreecommitdiff
path: root/lesana/command.py
blob: 6f3fefe34ffe5283e716fe827070ea1096ad5d31 (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
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
import logging
import os
import subprocess

import gadona
from . import Collection, Entry


class New(gadona.Command):
    name = 'new'
    arguments = [
        (['--collection', '-c'], dict(
            help='The collection to work on (default .)'
            )),
        ]

    def main(self):
        collection = Collection(self.settings.collection)
        new_entry = Entry(collection)
        collection.save_entries([new_entry])
        filepath = os.path.join(
            collection.itemdir,
            new_entry.fname
            )
        try:
            subprocess.call(['sensible-editor', filepath])
        except FileNotFoundError as e:
            logging.warning(
                "Could not open new file with editor: {}".format(str(e))
                )
        else:
            collection.update_cache([filepath])
        print(new_entry.fname)


class Index(gadona.Command):
    name = 'index'
    arguments = [
        (['--collection', '-c'], dict(
            help='The collection to work on (default .)'
            )),
        (['files'], dict(
            help='List of files to index (default: everything)',
            default=None,
            nargs='*'
            )),
        ]

    def main(self):
        collection = Collection(self.settings.collection)
        if self.settings.files:
            files = (os.path.basename(f) for f in self.settings.files)
        else:
            files = None
        indexed = collection.update_cache(fnames=files)
        print("Found and indexed {} entries".format(indexed))


class Search(gadona.Command):
    name = 'search'
    arguments = [
        (['--collection', '-c'], dict(
            help='The collection to work on (default .)'
            )),
        (['--template', '-t'], dict(
            help='Am',
            )),
        (['--offset'], dict(
            )),
        (['--pagesize'], dict(
            )),
        (['query'], dict(
            help='Xapian query to search in the collection',
            nargs='+'
            )),
        ]

    def main(self):
        # TODO: implement "searching" for everything
        if self.settings.offset:
            logging.warning(
                "offset exposes an internal knob and MAY BE" +
                " REMOVED from a future release of lesana"
                )
        if self.settings.pagesize:
            logging.warning(
                "pagesize exposes an internal knob and MAY BE" +
                " REMOVED from a future release of lesana"
                )
        offset = self.settings.offset or 0
        pagesize = self.settings.pagesize or 12
        collection = Collection(self.settings.collection)
        #TODO: pass the entries to a proper template 
        for entry in collection.search(
            ' '.join(self.settings.query),
            offset,
            pagesize):
            print(entry.fname)