59 lines
2.2 KiB
Python
59 lines
2.2 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):
|
|
iso8601 = page.data["datetime"]
|
|
iso8601_dt, sep, timezone = iso8601.partition("+")
|
|
if sep:
|
|
iso8601 = "%s+%s" % (
|
|
iso8601_dt, timezone.replace(":", "", 1))
|
|
|
|
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)
|