bitbot-3.11-fork/modules/user_time.py

#--depends-on commands
#--depends-on location

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"

    def _find_setting(self, event):
        target_user = event["user"]

        if event["args"]:
            target_user = event["server"].get_user(event["args_split"][0])

        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)

            if page and page.data and not page.data.get("error", None):
                dt = utils.iso8601_parse(page.data["datetime"],
                    microseconds=True)
                human = utils.datetime_human(dt)
                event["stdout"].write("Time for %s: %s" % (target_user.nickname,
                    human))
            else:
                raise utils.EventsResultsError()
        else:
            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)