anndl: Print raw names in choice list.
[utils.git] / anndl
... / ...
CommitLineData
1#!/usr/bin/python3
2
3import sys, os, io, getopt, ann, subprocess, tempfile
4pj = os.path.join
5
6def opentty():
7 try:
8 return io.TextIOWrapper(io.FileIO(os.open("/dev/tty", os.O_RDWR | os.O_NOCTTY), "r+"))
9 except OSError as exc:
10 sys.stderr.write("anndl: terminal not available for interactive input: %s\n" % (exc.strerror))
11 sys.exit(1)
12
13def choose(opts):
14 tty.write("%d matches:\n" % (len(opts)))
15 for i, anime in enumerate(opts):
16 tty.write("%d. %s\n" % (i + 1, anime.rawname))
17 tty.write("> ")
18 tty.flush()
19 resp = tty.readline().strip()
20 if resp == "":
21 return opts[0]
22 if not resp.isdigit():
23 sys.stderr.write("anndl: %s: not a number\n" % resp)
24 sys.exit(1)
25 n = int(resp) - 1
26 if not 0 <= n < len(opts):
27 sys.stderr.write("anndl: %s: no such option\n" % resp)
28 sys.exit(1)
29 return opts[n]
30
31def prompt(line, default=""):
32 tty.write(line)
33 if line[-1:] != " ":
34 if default != "":
35 tty.write(" [%s]" % (default))
36 tty.write(": ")
37 tty.flush()
38 resp = tty.readline().strip()
39 if resp == "":
40 return default
41 return resp
42
43def edit(string):
44 fd, path = tempfile.mkstemp()
45 try:
46 with os.fdopen(fd, "w") as fp:
47 fp.write(string)
48 subprocess.run([os.getenv("EDITOR"), path])
49 with open(path, "r") as fp:
50 return fp.read()
51 finally:
52 os.unlink(path)
53
54def usage(out):
55 out.write("usage: anndl [-h] SERIES\n")
56
57def main(argv):
58 global tty
59
60 opts, args = getopt.getopt(argv[1:], "h")
61 for o, a in opts:
62 if o == "-h":
63 usage(sys.stdout)
64 sys.exit(0)
65 if len(args) < 1:
66 usage(sys.stderr)
67 sys.exit(1)
68 name = args[0]
69 tty = opentty()
70
71 match = ann.getlist(name)
72 if len(match) == 0:
73 sys.stderr.write("anndl: could not find %s\n" % (name))
74 sys.exit(1)
75 elif len(match) == 1:
76 get = match[0]
77 elif len(match) < 10:
78 get = choose(match)
79 else:
80 sys.stderr.write("anndl: %s matches %i series\n" % (name, len(match)))
81 sys.exit(1)
82
83 if get.eps is None:
84 tty.write("could not find number of eps\n")
85 if prompt("accept? ")[:1].lower() != "y":
86 sys.exit(1)
87 name = prompt("canonical name", get.name)
88 sexpr = "\"" + r"L~%s[^0-9]*0*%%i([^0-9]*(\[[^]]*\]|\([^\)]*\)|v2))*[^0-9]*\.(avi|ogm|mkv|mp4)$" + "\""
89 sexpr = sexpr % (name.lower().replace(" ", ".*"))
90 tty.write("sexpr: %s\n" % (sexpr))
91 if prompt("accept? ")[:1].lower() != "y":
92 sexpr = edit(sexpr)
93
94 root = pj(os.getenv("HOME"), "dc/autodl/cur", name)
95 dl = pj(root, ".autodl")
96
97 try:
98 os.mkdir(root)
99 except OSError as exc:
100 sys.stderr.write("anndl: %s: %s\n" % (root, exc.strerror))
101 sys.exit(1)
102 os.mkdir(dl)
103 with open(pj(dl, "disabled"), "w") as fp:
104 pass
105 with open(pj(root, "nextep"), "w") as fp:
106 fp.write("1\n")
107 with open(pj(dl, "uarg"), "w") as fp:
108 if get.eps is None:
109 melen = 2
110 else:
111 melen = len(str(get.eps))
112 fp.write("rename:auto:%s:%i:move:../autodl/cur/%s\n" % (name, melen, name))
113 with open(pj(dl, "sexpr"), "w") as fp:
114 fp.write("%s\n" % sexpr)
115 with open(pj(dl, "curep"), "w") as fp:
116 fp.write("1\n")
117 if get.eps is not None:
118 with open(pj(dl, "maxep"), "w") as fp:
119 fp.write("%i\n" % get.eps)
120 if prompt("begin? ")[:1].lower() == "y":
121 os.unlink(pj(dl, "disabled"))
122
123if __name__ == "__main__":
124 try:
125 main(sys.argv)
126 except KeyboardInterrupt:
127 sys.exit(1)