anndl: Fixed a couple of bugs.
[utils.git] / anndl
1 #!/usr/bin/python3
2
3 import sys, os, io, getopt, ann, subprocess, tempfile
4 pj = os.path.join
5
6 def 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
13 def 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.name))
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
31 def 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
43 def 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
54 def usage(out):
55     out.write("usage: anndl [-h] SERIES\n")
56
57 def 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
123 if __name__ == "__main__":
124     try:
125         main(sys.argv)
126     except KeyboardInterrupt:
127         sys.exit(1)