bitbot-3.11-fork/modules/user_time.py
2019-07-27 21:09:45 +01:00

53 lines
1.9 KiB
Python

#--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)