From b7cf524e9869828ff85954cb731f7fee98bee8d9 Mon Sep 17 00:00:00 2001 From: jesopo Date: Fri, 31 Aug 2018 12:55:52 +0100 Subject: Give modules event objects with "context"s, to facilitate purging all the event hooks for a module --- modules/nr.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) (limited to 'modules/nr.py') diff --git a/modules/nr.py b/modules/nr.py index 7716b567..59df34c6 100644 --- a/modules/nr.py +++ b/modules/nr.py @@ -20,29 +20,29 @@ class Module(object): PASSENGER_ACTIVITIES = ["U", "P", "R"] COLOURS = [Utils.COLOR_LIGHTBLUE, Utils.COLOR_GREEN, Utils.COLOR_RED, Utils.COLOR_CYAN, Utils.COLOR_LIGHTGREY, Utils.COLOR_ORANGE] - def __init__(self, bot): + def __init__(self, bot, events): self.bot = bot self._client = None - bot.events.on("received").on("command").on("nrtrains" + events.on("received").on("command").on("nrtrains" ).hook(self.trains, min_args=1, help="Get train/bus services for a station (Powered by NRE)", usage="") - bot.events.on("received").on("command").on("nrservice" + events.on("received").on("command").on("nrservice" ).hook(self.service, min_args=1, help="Get train service information for a UID, headcode or RID (Powered by NRE)", usage="") - bot.events.on("received").on("command").on("nrhead" + events.on("received").on("command").on("nrhead" ).hook(self.head, min_args=1, help="Get information for a given headcode/UID/RID (Powered by NRE)", usage="") - bot.events.on("received").on("command").on("nrcode" + events.on("received").on("command").on("nrcode" ).hook(self.service_code, min_args=1, help="Get the text for a given delay/cancellation code (Powered by NRE)", usage="") - bot.events.on("telegram").on("command").on("nrtrains").hook(self.trains) - bot.events.on("telegram").on("command").on("nrcode").hook(self.service_code) - bot.events.on("telegram").on("command").on("nrhead").hook(self.head) - bot.events.on("telegram").on("command").on("nrservice").hook(self.service) + events.on("telegram").on("command").on("nrtrains").hook(self.trains) + events.on("telegram").on("command").on("nrcode").hook(self.service_code) + events.on("telegram").on("command").on("nrhead").hook(self.head) + events.on("telegram").on("command").on("nrservice").hook(self.service) @property def client(self): -- cgit v1.3.1-10-gc9f91