Commit | Line | Data |
---|---|---|
3cc7937c | 1 | #!/usr/bin/python3 |
e26514bb FT |
2 | |
3 | import sys, getopt | |
012c4cae | 4 | import manga.lib, manga.reader, manga.profile, manga.local |
9f1721f1 | 5 | from gi.repository import Gtk, Gdk, GObject |
e26514bb | 6 | |
9f1721f1 FT |
7 | GObject.threads_init() |
8 | Gdk.threads_init() | |
e26514bb FT |
9 | |
10 | def usage(out): | |
271d68da FT |
11 | out.write("usage: automanga [-h] [-p PROFILE] {DIRECTORY|-l LIBRARY ID|-a ALIAS}\n") |
12 | out.write(" automanga -a ALIAS=LIBRARY:ID\n") | |
5c000565 | 13 | out.write(" automanga -t {DIRECTORY|-l LIBRARY ID|-a ALIAS} [-]TAG...\n") |
ffd12e71 | 14 | out.write(" automanga -{s|S} LIBRARY NAME\n") |
5c000565 | 15 | out.write(" automanga -L TAG\n") |
e26514bb FT |
16 | |
17 | libname = None | |
ffd12e71 | 18 | search = None |
012c4cae | 19 | profile = "" |
271d68da | 20 | alias = None |
5c000565 FT |
21 | settags = False |
22 | listtag = None | |
23 | opts, args = getopt.getopt(sys.argv[1:], "hl:sSp:a:tL:") | |
e26514bb FT |
24 | for o, a in opts: |
25 | if o == "-h": | |
26 | usage(sys.stdout) | |
27 | sys.exit(0) | |
28 | elif o == "-l": | |
29 | libname = a | |
271d68da FT |
30 | elif o == "-a": |
31 | alias = a | |
e26514bb | 32 | elif o == "-s": |
ffd12e71 FT |
33 | search = "name" |
34 | elif o == "-S": | |
35 | search = "match" | |
012c4cae FT |
36 | elif o == "-p": |
37 | if a == "": | |
38 | profile = None | |
39 | else: | |
40 | profile = a | |
5c000565 FT |
41 | elif o == "-t": |
42 | settags = True | |
43 | elif o == "-L": | |
44 | listtag = a | |
012c4cae FT |
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) | |
e26514bb | 52 | |
299d3f50 | 53 | def main(): |
41a034f6 FT |
54 | global libname |
55 | ||
299d3f50 FT |
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 | |
e26514bb | 78 | |
5c000565 FT |
79 | if listtag is not None: |
80 | if profile is not None: | |
81 | for mprof in profile.bytag(listtag): | |
1e28222b FT |
82 | try: |
83 | mng = mprof.open() | |
84 | except KeyError: | |
85 | sys.stderr.write("%s %s: no longer found\n" % (mprof.libnm, mprof.id)) | |
86 | continue | |
5c000565 FT |
87 | sys.stdout.write("%s %s: \"%s\"\n" % (mprof.libnm, mprof.id, mng.name)) |
88 | return | |
89 | ||
299d3f50 | 90 | if alias and (alias.find('=') > 0): |
41a034f6 | 91 | tgt, nm = alias.split('=', 1) |
299d3f50 FT |
92 | if not ':' in nm: |
93 | usage(sys.stderr) | |
94 | sys.exit(1) | |
95 | libname, mid = nm.split(':', 1) | |
96 | try: | |
97 | lib = manga.lib.findlib(libname) | |
98 | except ImportError: | |
99 | sys.stderr.write("automanga: no such library: %s\n" % libname) | |
100 | sys.exit(1) | |
101 | try: | |
102 | mng = lib.byid(mid) | |
d39faa3f FT |
103 | except KeyError as e: |
104 | sys.stderr.write("automanga: no such manga: %s\n" % str(e)) | |
299d3f50 | 105 | sys.exit(1) |
41a034f6 | 106 | profile.setalias(tgt, libname, mng.id) |
299d3f50 | 107 | return |
271d68da | 108 | |
299d3f50 FT |
109 | if alias: |
110 | try: | |
111 | libname, mid = profile.getalias(alias) | |
112 | except KeyError: | |
43bddc57 | 113 | sys.stderr.write("automanga: no such alias: %s\n" % alias) |
299d3f50 FT |
114 | sys.exit(1) |
115 | try: | |
116 | lib = manga.lib.findlib(libname) | |
117 | except ImportError: | |
118 | sys.stderr.write("automanga: no such library: %s\n" % libname) | |
119 | sys.exit(1) | |
120 | try: | |
121 | mng = lib.byid(mid) | |
122 | except KeyError: | |
123 | sys.stderr.write("automanga: manga no longer found: %s\n" % mid) | |
124 | sys.exit(1) | |
125 | elif libname is not None: | |
126 | if len(args) < 1: | |
127 | usage(sys.stderr) | |
128 | sys.exit(1) | |
129 | try: | |
130 | lib = manga.lib.findlib(libname) | |
131 | except ImportError: | |
132 | sys.stderr.write("automanga: no such library: %s\n" % libname) | |
133 | sys.exit(1) | |
134 | try: | |
5c000565 | 135 | mng = lib.byid(args.pop(0)) |
d39faa3f FT |
136 | except KeyError as e: |
137 | sys.stderr.write("automanga: no such manga: %s\n" % str(e)) | |
299d3f50 FT |
138 | sys.exit(1) |
139 | else: | |
140 | libname = "local" | |
141 | try: | |
142 | if len(args) > 0: | |
5c000565 | 143 | mdir = args.pop(0) |
299d3f50 FT |
144 | else: |
145 | mdir = "." | |
146 | mng = manga.local.manga(mdir) | |
147 | except TypeError: | |
148 | sys.stderr.write("automanga: not a valid manga directory: %s\n" % mdir) | |
149 | sys.exit(1) | |
150 | ||
151 | mprof = None if profile is None else profile.getmanga(libname, mng.id, True) | |
152 | if profile is not None: | |
153 | profile.setlast() | |
e26514bb | 154 | |
5c000565 FT |
155 | if settags: |
156 | if mprof is not None: | |
157 | ntags = [tag for tag in args if tag[:1] != "-"] | |
158 | rtags = [tag[1:] for tag in args if tag[:1] == "-"] | |
5c000565 FT |
159 | if ntags: mprof.tags.add(*ntags) |
160 | if rtags: mprof.tags.remove(*rtags) | |
161 | return | |
162 | ||
299d3f50 FT |
163 | reader = manga.reader.reader(mng, mprof) |
164 | reader.show() | |
9f1721f1 | 165 | Gtk.main() |
012c4cae | 166 | |
299d3f50 FT |
167 | try: |
168 | main() | |
169 | except KeyboardInterrupt: | |
170 | pass |