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