You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
145 lines
3.9 KiB
145 lines
3.9 KiB
"""times module
|
|
|
|
This module provides some Date and Time classes for dealing with MySQL data.
|
|
|
|
Use Python datetime module to handle date and time columns.
|
|
"""
|
|
from time import localtime
|
|
from datetime import date, datetime, time, timedelta
|
|
from _mysql import string_literal
|
|
|
|
Date = date
|
|
Time = time
|
|
TimeDelta = timedelta
|
|
Timestamp = datetime
|
|
|
|
DateTimeDeltaType = timedelta
|
|
DateTimeType = datetime
|
|
|
|
def DateFromTicks(ticks):
|
|
"""Convert UNIX ticks into a date instance."""
|
|
return date(*localtime(ticks)[:3])
|
|
|
|
def TimeFromTicks(ticks):
|
|
"""Convert UNIX ticks into a time instance."""
|
|
return time(*localtime(ticks)[3:6])
|
|
|
|
def TimestampFromTicks(ticks):
|
|
"""Convert UNIX ticks into a datetime instance."""
|
|
return datetime(*localtime(ticks)[:6])
|
|
|
|
format_TIME = format_DATE = str
|
|
|
|
def format_TIMEDELTA(v):
|
|
seconds = int(v.seconds) % 60
|
|
minutes = int(v.seconds // 60) % 60
|
|
hours = int(v.seconds // 3600) % 24
|
|
return '%d %d:%d:%d' % (v.days, hours, minutes, seconds)
|
|
|
|
def format_TIMESTAMP(d):
|
|
"""
|
|
:type d: datetime.datetime
|
|
"""
|
|
if d.microsecond:
|
|
fmt = "{0.year:04}-{0.month:02}-{0.day:02} {0.hour:02}:{0.minute:02}:{0.second:02}.{0.microsecond:06}"
|
|
else:
|
|
fmt = "{0.year:04}-{0.month:02}-{0.day:02} {0.hour:02}:{0.minute:02}:{0.second:02}"
|
|
return fmt.format(d)
|
|
|
|
|
|
def DateTime_or_None(s):
|
|
try:
|
|
if len(s) < 11:
|
|
return Date_or_None(s)
|
|
|
|
micros = s[20:]
|
|
|
|
if len(micros) == 0:
|
|
# 12:00:00
|
|
micros = 0
|
|
elif len(micros) < 7:
|
|
# 12:00:00.123456
|
|
micros = int(micros) * 10 ** (6 - len(micros))
|
|
else:
|
|
return None
|
|
|
|
return datetime(
|
|
int(s[:4]), # year
|
|
int(s[5:7]), # month
|
|
int(s[8:10]), # day
|
|
int(s[11:13] or 0), # hour
|
|
int(s[14:16] or 0), # minute
|
|
int(s[17:19] or 0), # second
|
|
micros, # microsecond
|
|
)
|
|
except ValueError:
|
|
return None
|
|
|
|
def TimeDelta_or_None(s):
|
|
try:
|
|
h, m, s = s.split(':')
|
|
if '.' in s:
|
|
s, ms = s.split('.')
|
|
ms = ms.ljust(6, '0')
|
|
else:
|
|
ms = 0
|
|
if h[0] == '-':
|
|
negative = True
|
|
else:
|
|
negative = False
|
|
h, m, s, ms = abs(int(h)), int(m), int(s), int(ms)
|
|
td = timedelta(hours=h, minutes=m, seconds=s,
|
|
microseconds=ms)
|
|
if negative:
|
|
return -td
|
|
else:
|
|
return td
|
|
except ValueError:
|
|
# unpacking or int/float conversion failed
|
|
return None
|
|
|
|
def Time_or_None(s):
|
|
try:
|
|
h, m, s = s.split(':')
|
|
if '.' in s:
|
|
s, ms = s.split('.')
|
|
ms = ms.ljust(6, '0')
|
|
else:
|
|
ms = 0
|
|
h, m, s, ms = int(h), int(m), int(s), int(ms)
|
|
return time(hour=h, minute=m, second=s,
|
|
microsecond=ms)
|
|
except ValueError:
|
|
return None
|
|
|
|
def Date_or_None(s):
|
|
try:
|
|
return date(
|
|
int(s[:4]), # year
|
|
int(s[5:7]), # month
|
|
int(s[8:10]), # day
|
|
)
|
|
except ValueError:
|
|
return None
|
|
|
|
def DateTime2literal(d, c):
|
|
"""Format a DateTime object as an ISO timestamp."""
|
|
return string_literal(format_TIMESTAMP(d), c)
|
|
|
|
def DateTimeDelta2literal(d, c):
|
|
"""Format a DateTimeDelta object as a time."""
|
|
return string_literal(format_TIMEDELTA(d),c)
|
|
|
|
def mysql_timestamp_converter(s):
|
|
"""Convert a MySQL TIMESTAMP to a Timestamp object."""
|
|
# MySQL>4.1 returns TIMESTAMP in the same format as DATETIME
|
|
if s[4] == '-': return DateTime_or_None(s)
|
|
s = s + "0"*(14-len(s)) # padding
|
|
parts = map(int, filter(None, (s[:4],s[4:6],s[6:8],
|
|
s[8:10],s[10:12],s[12:14])))
|
|
try:
|
|
return Timestamp(*parts)
|
|
except (SystemExit, KeyboardInterrupt):
|
|
raise # pragma: no cover
|
|
except:
|
|
return None
|