diff options
Diffstat (limited to 'modules/trakt.py')
| -rw-r--r-- | modules/trakt.py | 29 |
1 files changed, 16 insertions, 13 deletions
diff --git a/modules/trakt.py b/modules/trakt.py index b3c36d3d..86d3916b 100644 --- a/modules/trakt.py +++ b/modules/trakt.py @@ -1,32 +1,35 @@ -#--require-config trakt-api-key +# --require-config trakt-api-key import Utils URL_TRAKT = "https://api-v2launch.trakt.tv/users/%s/watching" URL_TRAKTSLUG = "https://trakt.tv/%s/%s" + class Module(object): def __init__(self, bot): self.bot = bot bot.events.on("received").on("command").on("nowwatching", - "nw").hook(self.now_watching, - help="Get what you or another user is now watching " - "on trakt.tv", usage="[username]") + "nw").hook(self.now_watching, + help="Get what you or another user is now watching " + "on trakt.tv", + usage="[username]") bot.events.on("postboot").on("configure").on("set" - ).assure_call(setting="trakt", help="Set username on trakt.tv") + ).assure_call( + setting="trakt", help="Set username on trakt.tv") def now_watching(self, event): if event["args"]: username = event["args_split"][0] else: username = event["user"].get_setting("trakt", - event["user"].nickname) + event["user"].nickname) page = Utils.get_url(URL_TRAKT % username, headers={ "Content-Type": "application/json", "trakt-api-version": "2", "trakt-api-key": - self.bot.config["trakt-api-key"]}, json=True, - code=True) + self.bot.config["trakt-api-key"]}, json=True, + code=True) if page[0]: code, page = page if code == 200: @@ -37,8 +40,8 @@ class Module(object): slug = page["movie"]["ids"]["slug"] event["stdout"].write( "%s is now watching %s (%s) %s" % ( - username, title, year, - URL_TRAKTSLUG % ("movie", slug))) + username, title, year, + URL_TRAKTSLUG % ("movie", slug))) elif type == "episode": season = page["episode"]["season"] episode_number = page["episode"]["number"] @@ -48,9 +51,9 @@ class Module(object): slug = page["show"]["ids"]["slug"] event["stdout"].write( "%s is now watching %s s%se%s - %s %s" % ( - username, show_title, str(season).zfill(2), - str(episode_number).zfill(2), episode_title, - URL_TRAKTSLUG % ("shows", slug))) + username, show_title, str(season).zfill(2), + str(episode_number).zfill(2), episode_title, + URL_TRAKTSLUG % ("shows", slug))) else: print("ack! unknown trakt media type!") else: |