aboutsummaryrefslogtreecommitdiff
path: root/src/utils/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/utils/__init__.py')
-rw-r--r--src/utils/__init__.py127
1 files changed, 2 insertions, 125 deletions
diff --git a/src/utils/__init__.py b/src/utils/__init__.py
index b7b581e2..294b9083 100644
--- a/src/utils/__init__.py
+++ b/src/utils/__init__.py
@@ -1,6 +1,6 @@
-import contextlib, datetime, decimal, enum, io, ipaddress, multiprocessing
+import contextlib, decimal, enum, io, ipaddress, multiprocessing
import queue, re, signal, threading, typing
-from . import cli, consts, decorators, irc, http, parse, security
+from . import cli, consts, datetime, decorators, irc, http, parse, security
from .decorators import export, hook, kwarg
from .settings import (BoolSetting, FunctionSetting, IntRangeSetting,
@@ -10,129 +10,6 @@ class Direction(enum.Enum):
Send = 0
Recv = 1
-ISO8601_PARSE = "%Y-%m-%dT%H:%M:%S%z"
-ISO8601_PARSE_MICROSECONDS = "%Y-%m-%dT%H:%M:%S.%f%z"
-
-ISO8601_FORMAT_DT = "%Y-%m-%dT%H:%M:%S"
-ISO8601_FORMAT_TZ = "%z"
-
-
-DATETIME_HUMAN = "%Y/%m/%d %H:%M:%S"
-DATE_HUMAN = "%Y-%m-%d"
-
-def datetime_utcnow() -> datetime.datetime:
- return datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc)
-def datetime_timestamp(seconds: float) -> datetime.datetime:
- return datetime.datetime.fromtimestamp(seconds).replace(
- tzinfo=datetime.timezone.utc)
-
-def iso8601_format(dt: datetime.datetime, milliseconds: bool=False) -> str:
- dt_format = dt.strftime(ISO8601_FORMAT_DT)
- tz_format = dt.strftime(ISO8601_FORMAT_TZ)
-
- ms_format = ""
- if milliseconds:
- ms_format = ".%s" % str(int(dt.microsecond/1000)).zfill(3)
-
- return "%s%s%s" % (dt_format, ms_format, tz_format)
-def iso8601_format_now(milliseconds: bool=False) -> str:
- return iso8601_format(datetime_utcnow(), milliseconds=milliseconds)
-def iso8601_parse(s: str, microseconds: bool=False) -> datetime.datetime:
- fmt = ISO8601_PARSE_MICROSECONDS if microseconds else ISO8601_PARSE
- return datetime.datetime.strptime(s, fmt)
-
-def datetime_human(dt: datetime.datetime):
- return datetime.datetime.strftime(dt, DATETIME_HUMAN)
-def date_human(dt: datetime.datetime):
- return datetime.datetime.strftime(dt, DATE_HUMAN)
-
-TIME_SECOND = 1
-TIME_MINUTE = TIME_SECOND*60
-TIME_HOUR = TIME_MINUTE*60
-TIME_DAY = TIME_HOUR*24
-TIME_WEEK = TIME_DAY*7
-
-def time_unit(seconds: int) -> typing.Tuple[int, str]:
- since = None
- unit = None
- if seconds >= TIME_WEEK:
- since = seconds/TIME_WEEK
- unit = "week"
- elif seconds >= TIME_DAY:
- since = seconds/TIME_DAY
- unit = "day"
- elif seconds >= TIME_HOUR:
- since = seconds/TIME_HOUR
- unit = "hour"
- elif seconds >= TIME_MINUTE:
- since = seconds/TIME_MINUTE
- unit = "minute"
- else:
- since = seconds
- unit = "second"
- since = int(since)
- if since > 1:
- unit = "%ss" % unit # pluralise the unit
- return (since, unit)
-
-REGEX_PRETTYTIME = re.compile(
- r"(?:(\d+)w)?(?:(\d+)d)?(?:(\d+)h)?(?:(\d+)m)?(?:(\d+)s)?", re.I)
-
-SECONDS_MINUTES = 60
-SECONDS_HOURS = SECONDS_MINUTES*60
-SECONDS_DAYS = SECONDS_HOURS*24
-SECONDS_WEEKS = SECONDS_DAYS*7
-
-def from_pretty_time(pretty_time: str) -> typing.Optional[int]:
- seconds = 0
-
- match = re.match(REGEX_PRETTYTIME, pretty_time)
- if match:
- seconds += int(match.group(1) or 0)*SECONDS_WEEKS
- seconds += int(match.group(2) or 0)*SECONDS_DAYS
- seconds += int(match.group(3) or 0)*SECONDS_HOURS
- seconds += int(match.group(4) or 0)*SECONDS_MINUTES
- seconds += int(match.group(5) or 0)
-
- if seconds > 0:
- return seconds
- return None
-
-UNIT_MINIMUM = 6
-UNIT_SECOND = 5
-UNIT_MINUTE = 4
-UNIT_HOUR = 3
-UNIT_DAY = 2
-UNIT_WEEK = 1
-def to_pretty_time(total_seconds: int, minimum_unit: int=UNIT_SECOND,
- max_units: int=UNIT_MINIMUM) -> str:
- if total_seconds == 0:
- return "0s"
-
- minutes, seconds = divmod(total_seconds, 60)
- hours, minutes = divmod(minutes, 60)
- days, hours = divmod(hours, 24)
- weeks, days = divmod(days, 7)
- out = []
-
- units = 0
- if weeks and minimum_unit >= UNIT_WEEK and units < max_units:
- out.append("%dw" % weeks)
- units += 1
- if days and minimum_unit >= UNIT_DAY and units < max_units:
- out.append("%dd" % days)
- units += 1
- if hours and minimum_unit >= UNIT_HOUR and units < max_units:
- out.append("%dh" % hours)
- units += 1
- if minutes and minimum_unit >= UNIT_MINUTE and units < max_units:
- out.append("%dm" % minutes)
- units += 1
- if seconds and minimum_unit >= UNIT_SECOND and units < max_units:
- out.append("%ds" % seconds)
- units += 1
- return " ".join(out)
-
def prevent_highlight(nickname: str) -> str:
return nickname[0]+"\u200c"+nickname[1:]