-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathradarr-monitor.py
84 lines (72 loc) · 2.71 KB
/
radarr-monitor.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
import requests
import argparse
import logging
import json
logging.basicConfig(
level=logging.DEBUG,
format='%(asctime)s|%(pathname)s:%(lineno)d|%(levelname)s|%(message)s',
datefmt='%Y-%m-%d.%H-%M-%S'
)
log = logging.getLogger(__name__)
def main(args):
log.setLevel(logging.DEBUG if args.debug else logging.INFO)
api = 'http://{}:{}/radarr/api/movie/'.format(
args.host, args.port
)
movies_url = '{}?apikey={}'.format(api, args.api_key)
r = requests.get(movies_url)
r.raise_for_status()
movies = r.json()
for movie in movies:
dl = movie.get('downloaded')
monitored = movie.get('monitored')
path = movie.get('path')
if dl == monitored:
movie['monitored'] = not dl # if downloaded, unset monitored.
# put url is same as get everything since id is embedded(?)
log.debug('+: {} -> {}'.format(
movie.get('path'),
'unmonitored' if dl else 'monitored'
))
if not args.print_only:
data = json.dumps(movie)
log.debug('put-ing to {}, data={}'.format(movies_url, data))
r = requests.put(movies_url, data=data)
r.raise_for_status()
else:
log.debug('-: {:40s}; {} & {}'.format(
path.replace('/movies/', ''),
'downloaded' if dl else 'undownloaded',
'monitored' if monitored else 'unmonitored',
))
if __name__ == '__main__':
parser = argparse.ArgumentParser(
description='Movie un-monitor-or. Will unmonitor any movies listed as downloaded, and monitor all those not listed as downloaded.'
)
parser.add_argument('-D', '--debug',
help='Set DEBUG log level. Default INFO',
required=False,
action='store_true')
parser.add_argument('-?', '--print-only',
help="print what I'd do, but don't do it",
required=False,
action='store_true')
parser.add_argument('--host',
help='radarr host',
required=False,
default='localhost',
type=str)
parser.add_argument('-p', '--port',
help='radarr port',
required=False,
default='7878',
type=str)
parser.add_argument('-a', '--api-key',
help='radarr api key',
required=True,
type=str)
args = parser.parse_args()
try:
main(args)
except KeyboardInterrupt:
pass