diff --git a/modules/user_time.py b/modules/user_time.py index 0e6fa6ce..47ba1708 100644 --- a/modules/user_time.py +++ b/modules/user_time.py @@ -5,24 +5,26 @@ import datetime from src import ModuleManager, utils API = "http://worldtimeapi.org/api/timezone/%s" +NOLOCATION = "%s doesn't have a location set" class Module(ModuleManager.BaseModule): _name = "Time" - @utils.hook("received.command.time") - def time(self, event): - """ - :help: Get the time for you or someone else - :usage: [nickname] - :require_setting: location - :require_setting_unless: 1 - """ + def _find_setting(self, event): target_user = event["user"] + if event["args"]: target_user = event["server"].get_user(event["args_split"][0]) - location = target_user.get_setting("location", None) + return target_user, target_user.get_setting("location", None) + @utils.hook("received.command.time") + @utils.kwarg("help", "Get the time for you or someone else") + @utils.kwarg("usage", "[nickname]") + @utils.kwarg("require_setting", "location") + @utils.kwarg("require_setting_unless", "1") + def time(self, event): + target_user, location = self._find_setting(event) if not location == None: page = utils.http.request(API % location["timezone"], json=True) @@ -35,5 +37,17 @@ class Module(ModuleManager.BaseModule): else: raise utils.EventsResultsError() else: - event["stderr"].write("%s doesn't have a location set" % - target_user.nickname) + event["stderr"].write(NOLOCATION % target_user.nickname) + + @utils.hook("received.command.timezone") + @utils.kwarg("help", "Get the timezone for you or someone else") + @utils.kwarg("usage", "[nickname]") + @utils.kwarg("require_setting", "location") + @utils.kwarg("require_setting_unless", "1") + def timezone(self, event): + target_user, location = self._find_setting(event) + if not location == None: + event["stdout"].write("%s is in %s" % (target_user.nickname, + location["timezone"])) + else: + event["stderr"].write(NOLOCATION % target_user.nickname)