automanga: Added option to sort tags-lists by mtime.
[automanga.git] / automanga
1 #!/usr/bin/python3
2
3 import sys, getopt, pathlib
4 import manga.lib, manga.reader, manga.profile, manga.local
5 from gi.repository import Gtk, Gdk, GObject
6
7 def usage(out):
8     out.write("usage: automanga [-h] [-p PROFILE] {DIRECTORY|-l LIBRARY ID|-a ALIAS}\n")
9     out.write("       automanga -a ALIAS=LIBRARY:ID\n")
10     out.write("       automanga -t {DIRECTORY|-l LIBRARY ID|-a ALIAS} [-]TAG...\n")
11     out.write("       automanga -{s|S} LIBRARY NAME\n")
12     out.write("       automanga -L TAG [-o ORDER]\n")
13
14 libname = None
15 search = None
16 profile = ""
17 alias = None
18 settags = False
19 listtag = None
20 listorder = None
21 opts, args = getopt.getopt(sys.argv[1:], "hl:sSp:a:tL:o:")
22 for o, a in opts:
23     if o == "-h":
24         usage(sys.stdout)
25         sys.exit(0)
26     elif o == "-l":
27         libname = a
28     elif o == "-a":
29         alias = a
30     elif o == "-s":
31         search = "name"
32     elif o == "-S":
33         search = "match"
34     elif o == "-p":
35         if a == "":
36             profile = None
37         else:
38             profile = a
39     elif o == "-t":
40         settags = True
41     elif o == "-L":
42         listtag = a
43     elif o == "-o":
44         listorder = a
45 if profile == "":
46     try:
47         profile = manga.profile.profile.last()
48     except KeyError:
49         profile = manga.profile.profile.byname("default")
50 elif profile is not None:
51     profile = manga.profile.profile.byname(profile)
52
53 def main():
54     global libname
55
56     if search:
57         if len(args) < 2:
58             usage(sys.stderr)
59             sys.exit(1)
60         try:
61             lib = manga.lib.findlib(args[0])
62         except ImportError:
63             sys.stderr.write("automanga: no such library: %s\n" % args[0])
64             sys.exit(1)
65         try:
66             if search == "name":
67                 results = lib.byname(args[1])
68             elif search == "match":
69                 results = lib.search(args[1])
70                 if hasattr(results, "__length_hint__"):
71                     sys.stderr.write("%i results:\n" % results.__length_hint__())
72         except NotImplementedError:
73             sys.stderr.write("automanga: %s does not support searching\n" % args[0])
74             sys.exit(1)
75         for mng in results:
76             sys.stdout.write("\"%s\": %s\n" % (mng.name, mng.id))
77         return
78
79     if listtag is not None:
80         if profile is not None:
81             results = []
82             for mprof in profile.bytag(listtag):
83                 try:
84                     mng = mprof.open()
85                 except KeyError:
86                     sys.stderr.write("%s %s: no longer found\n" % (mprof.libnm, mprof.id))
87                     continue
88                 except Exception as exc:
89                     sys.stderr.write("%s %s: unexpected error: %s\n" % (mprof.libnm, mprof.id, str(exc)))
90                     continue
91                 line = "%s %s: \"%s\"" % (mprof.libnm, mprof.id, mng.name)
92                 if listorder is None:
93                     sys.stdout.write("%s\n" % (line,))
94                 else:
95                     if listorder == "mtime":
96                         key = -mprof.mtime()
97                     else:
98                         sys.stderr.write("automanga: undefined order: %s\n" % (listorder,))
99                         sys.exit(1)
100                     results.append((line, key))
101             if len(results) > 0:
102                 results.sort(key=lambda o: o[1])
103                 for line, key in results:
104                     sys.stdout.write("%s\n" % (line,))
105         return
106
107     if alias and (alias.find('=') > 0):
108         tgt, nm = alias.split('=', 1)
109         if not ':' in nm:
110             usage(sys.stderr)
111             sys.exit(1)
112         libname, mid = nm.split(':', 1)
113         try:
114             lib = manga.lib.findlib(libname)
115         except ImportError:
116             sys.stderr.write("automanga: no such library: %s\n" % libname)
117             sys.exit(1)
118         try:
119             mng = lib.byid(mid)
120         except KeyError as e:
121             sys.stderr.write("automanga: no such manga: %s\n" % str(e))
122             sys.exit(1)
123         profile.setalias(tgt, libname, mng.id)
124         return
125
126     if alias:
127         try:
128             libname, mid = profile.getalias(alias)
129         except KeyError:
130             sys.stderr.write("automanga: no such alias: %s\n" % alias)
131             sys.exit(1)
132         try:
133             lib = manga.lib.findlib(libname)
134         except ImportError:
135             sys.stderr.write("automanga: no such library: %s\n" % libname)
136             sys.exit(1)
137         try:
138             mng = lib.byid(mid)
139         except KeyError:
140             sys.stderr.write("automanga: manga no longer found: %s\n" % mid)
141             sys.exit(1)
142     elif libname is not None:
143         if len(args) < 1:
144             usage(sys.stderr)
145             sys.exit(1)
146         try:
147             lib = manga.lib.findlib(libname)
148         except ImportError:
149             sys.stderr.write("automanga: no such library: %s\n" % libname)
150             sys.exit(1)
151         try:
152             mng = lib.byid(args.pop(0))
153         except KeyError as e:
154             sys.stderr.write("automanga: no such manga: %s\n" % str(e))
155             sys.exit(1)
156     else:
157         libname = "local"
158         try:
159             if len(args) < 1:
160                 usage(sys.stderr)
161                 sys.exit(1)
162             mdir = args.pop(0)
163             mng = manga.local.manga(pathlib.Path(mdir))
164         except TypeError:
165             sys.stderr.write("automanga: not a valid manga directory: %s\n" % mdir)
166             sys.exit(1)
167
168     mprof = None if profile is None else profile.getmanga(libname, mng.id, True)
169     if profile is not None:
170         profile.setlast()
171
172     if settags:
173         if mprof is not None:
174             ntags = [tag for tag in args if tag[:1] != "-"]
175             rtags = [tag[1:] for tag in args if tag[:1] == "-"]
176             if ntags: mprof.tags.add(*ntags)
177             if rtags: mprof.tags.remove(*rtags)
178         return
179
180     reader = manga.reader.reader(mng, mprof)
181     reader.show()
182     Gtk.main()
183
184 try:
185     main()
186 except KeyboardInterrupt:
187     pass