-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
executable file
·53 lines (45 loc) · 1.61 KB
/
main.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
#!/usr/bin/env python2.7
import json
import requests
import os
import argparse
from slack import Slack, SlackMessage
from radarr import RadarrApi
from tmdb import TmdbApi
def _argparse():
parser = argparse.ArgumentParser(
description='Radarr Custom Script : perform Slack rich notification'
)
parser.add_argument(
'--webhook-url', '-wu',
help='Slack webhook url'
)
parser.add_argument(
'--radarr-url', '-re',
help='Radarr API endpoint : https://xxxx/api'
)
parser.add_argument(
'--radarr-key', '-rk',
help='Radarr API key, find it on Radarr > Settings > General'
)
parser.add_argument(
'--tmdb-key', '-tk',
help='TMDB API Key, register app on tmdb to obtain API Key'
)
args = parser.parse_args()
return args
args = _argparse()
radarr = RadarrApi(args.radarr_url, args.radarr_key)
radarr.unmonitorMovieIfNeeded(os.environ.get("radarr_movie_id"), os.environ.get("radarr_eventtype"))
radarr.loadData(os.environ.get("radarr_download_id"))
link = "https://www.themoviedb.org/movie/"+os.environ.get("radarr_movie_tmdbid", "")
tmdb = TmdbApi(args.tmdb_key)
tmdb.loadMovieData(radarr.tmdbId)
tmdb.downloadMovieProductionImage()
message = SlackMessage(args.webhook_url)
message.package("*"+ os.environ.get("radarr_movie_title", "") +"* ("+ radarr.year +") ["+ os.environ.get("radarr_moviefile_quality", "") +"]")
message.constructor("`" +radarr.indexer+"` _"+os.environ.get("radarr_moviefile_releasegroup", "")+"_ ("+radarr.sizeOnDisk+")")
message.release(radarr.releaseTitle)
message.link(link)
message.save()
message.notify()