47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
#--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)
|