bitbot-3.11-fork/modules/user_time.py

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

import datetime
import pytz
from src import ModuleManager, utils

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:
            dt = datetime.datetime.now(tz=pytz.timezone(location["timezone"]))
            human = utils.datetime_human(dt)
            event["stdout"].write("Time for %s: %s" % (target_user.nickname,
                human))
        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)