Limnoria/plugins/Time/plugin.py
Valentin Lorentz 8b26b675ba Use stdlib instead of pytz on Python >= 3.9
Python 3.9 introduced the zoneinfo module, which provides the only
feature we used pytz for (getting a datetime.tzinfo object from
an IANA timezone id); so let's use it instead of a third-party
dependency.
2021-11-08 21:46:48 +01:00

236 lines
9.1 KiB
Python

###
# Copyright (c) 2004, Jeremiah Fincher
# Copyright (c) 2010-2021, Valentin Lorentz
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
import sys
import time
TIME = time # For later use.
from datetime import datetime
import supybot.conf as conf
import supybot.log as log
import supybot.utils as utils
from supybot.commands import *
import supybot.callbacks as callbacks
from supybot.i18n import PluginInternationalization, internationalizeDocstring
_ = PluginInternationalization('Time')
try:
from ddate.base import DDate as _ddate
except ImportError:
log.debug("Time: the ddate module is not available; disabling that command.")
_ddate = None
try:
from dateutil import parser
def parse(s):
todo = []
s = s.replace('noon', '12:00')
s = s.replace('midnight', '00:00')
if 'tomorrow' in s:
todo.append(lambda i: i + 86400)
s = s.replace('tomorrow', '')
if 'next week' in s:
todo.append(lambda i: i + 86400*7)
s = s.replace('next week', '')
i = int(time.mktime(parser.parse(s, fuzzy=True).timetuple()))
for f in todo:
i = f(i)
return i
except ImportError:
parse = None
try:
from dateutil.tz import tzlocal
except ImportError:
tzlocal = None
class Time(callbacks.Plugin):
"""This plugin allows you to use different time-related functions."""
@internationalizeDocstring
def seconds(self, irc, msg, args):
"""[<years>y] [<weeks>w] [<days>d] [<hours>h] [<minutes>m] [<seconds>s]
Returns the number of seconds in the number of <years>, <weeks>,
<days>, <hours>, <minutes>, and <seconds> given. An example usage is
"seconds 2h 30m", which would return 9000, which is '3600*2 + 30*60'.
Useful for scheduling events at a given number of seconds in the
future.
"""
if not args:
raise callbacks.ArgumentError
seconds = 0
for arg in args:
if not arg or arg[-1] not in 'ywdhms':
raise callbacks.ArgumentError
(s, kind) = arg[:-1], arg[-1]
try:
i = int(s)
except ValueError:
irc.errorInvalid('argument', arg, Raise=True)
if kind == 'y':
seconds += i*31536000
elif kind == 'w':
seconds += i*604800
elif kind == 'd':
seconds += i*86400
elif kind == 'h':
seconds += i*3600
elif kind == 'm':
seconds += i*60
elif kind == 's':
seconds += i
irc.reply(str(seconds))
@internationalizeDocstring
def at(self, irc, msg, args, s=None):
"""[<time string>]
Returns the number of seconds since epoch <time string> is.
<time string> can be any number of natural formats; just try something
and see if it will work.
If the <time string> is not given, defaults to now.
"""
if not s or s == 'now':
irc.reply(str(int(time.time())))
return
if not parse:
irc.error(_('This command is not available on this bot, ask the '
'owner to install the python-dateutil library.'), Raise=True)
now = int(time.time())
new = parse(s)
if new != now:
irc.reply(str(new))
else:
irc.error(_('That\'s right now!'))
at = wrap(at, [optional('text')])
@internationalizeDocstring
def until(self, irc, msg, args, s):
"""<time string>
Returns the number of seconds until <time string>.
"""
if not parse:
irc.error(_('This command is not available on this bot, ask the '
'owner to install the python-dateutil library.'), Raise=True)
now = int(time.time())
new = parse(s)
if new != now:
if new - now < 0:
new += 86400
irc.reply(str(new-now))
else:
irc.error(_('That\'s right now!'))
until = wrap(until, ['text'])
@internationalizeDocstring
def ctime(self, irc, msg, args, seconds):
"""[<seconds since epoch>]
Returns the ctime for <seconds since epoch>, or the current ctime if
no <seconds since epoch> is given.
"""
irc.reply(time.ctime(seconds))
ctime = wrap(ctime,[additional(('int', _('number of seconds since epoch')),
TIME.time)])
@internationalizeDocstring
def time(self, irc, msg, args, channel, format, seconds):
"""[<channel>] [<format>] [<seconds since epoch>]
Returns the current time in <format> format, or, if <format> is not
given, uses the configurable format for the current channel. If no
<seconds since epoch> time is given, the current time is used. If
<channel> is given without <format>, uses the format for <channel>.
"""
if not format:
format = self.registryValue('format', channel or msg.channel,
irc.network)
if tzlocal:
irc.reply(datetime.fromtimestamp(seconds, tzlocal()).strftime(format))
else:
# NOTE: This has erroneous behavior on some older Python versions,
# including at least up to 2.7.5 and 3.2.3. Install dateutil if you
# can't upgrade Python.
irc.reply(time.strftime(format, time.localtime(seconds)))
time = wrap(time, [optional('channel'), optional('nonInt'),
additional('float', TIME.time)])
@internationalizeDocstring
def elapsed(self, irc, msg, args, seconds):
"""<seconds>
Returns a pretty string that is the amount of time represented by
<seconds>.
"""
irc.reply(utils.timeElapsed(seconds))
elapsed = wrap(elapsed, ['int'])
@internationalizeDocstring
def tztime(self, irc, msg, args, timezone):
"""<region>/<city> (or <region>/<state>/<city>)
Takes a city and its region, and returns its local time. This
command uses the IANA Time Zone Database."""
try:
timezone = utils.time.iana_timezone(timezone)
except utils.time.UnknownTimeZone:
irc.error(_('Unknown timezone'))
except utils.time.MissingTimezoneLibrary:
irc.error(_('Python-tz is required by the command, but is not '
'installed on this computer.'))
except utils.time.UnknownTimeZone as e:
irc.error(e.args[0])
else:
format = self.registryValue("format", msg.channel, irc.network)
irc.reply(datetime.now(timezone).strftime(format))
tztime = wrap(tztime, ['text'])
def ddate(self, irc, msg, args, year=None, month=None, day=None):
"""[<year> <month> <day>]
Returns a the Discordian date today, or an optional different date."""
if _ddate is not None:
if year is not None and month is not None and day is not None:
try:
irc.reply(_ddate(datetime(year=year, month=month, day=day)))
except ValueError as e:
irc.error("%s", e)
else:
irc.reply(_ddate())
else:
irc.error(format(_("The 'ddate' module is not installed. Use "
"'%s -m pip install --user ddate' or see "
"%u for more information."), sys.executable,
"https://pypi.python.org/pypi/ddate/"))
ddate = wrap(ddate, [optional('positiveint'), optional('positiveint'), optional('positiveint')])
Class = Time
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: