times.py

"""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.
"""
fromtimeimportlocaltime
fromdatetimeimportdate, datetime, time, timedelta
fromMySQLdb._mysqlimportstring_literal

Date=date
Time=time
TimeDelta=timedelta
Timestamp=datetime

DateTimeDeltaType=timedelta
DateTimeType=datetime


defDateFromTicks(ticks):
"""Convert UNIX ticks into a date instance."""
returndate(*localtime(ticks)[:3])


defTimeFromTicks(ticks):
"""Convert UNIX ticks into a time instance."""
returntime(*localtime(ticks)[3:6])


defTimestampFromTicks(ticks):
"""Convert UNIX ticks into a datetime instance."""
returndatetime(*localtime(ticks)[:6])


format_TIME=format_DATE=str


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


defformat_TIMESTAMP(d):
"""
 :type d: datetime.datetime
 """
ifd.microsecond:
fmt=" ".join(
 [
"{0.year:04}-{0.month:02}-{0.day:02}",
"{0.hour:02}:{0.minute:02}:{0.second:02}.{0.microsecond:06}",
 ]
 )
else:
fmt=" ".join(
 [
"{0.year:04}-{0.month:02}-{0.day:02}",
"{0.hour:02}:{0.minute:02}:{0.second:02}",
 ]
 )
returnfmt.format(d)


defDateTime_or_None(s):
try:
iflen(s) <11:
returnDate_or_None(s)

micros=s[20:]

iflen(micros) ==0:
# 12:00:00
micros=0
eliflen(micros) <7:
# 12:00:00.123456
micros=int(micros) *10** (6-len(micros))
else:
returnNone

returndatetime(
int(s[:4]), # year
int(s[5:7]), # month
int(s[8:10]), # day
int(s[11:13] or0), # hour
int(s[14:16] or0), # minute
int(s[17:19] or0), # second
micros, # microsecond
 )
exceptValueError:
returnNone


defTimeDelta_or_None(s):
try:
h, m, s=s.split(":")
if"."ins:
s, ms=s.split(".")
ms=ms.ljust(6, "0")
else:
ms=0
ifh[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)
ifnegative:
return-td
else:
returntd
exceptValueError:
# unpacking or int/float conversion failed
returnNone


defTime_or_None(s):
try:
h, m, s=s.split(":")
if"."ins:
s, ms=s.split(".")
ms=ms.ljust(6, "0")
else:
ms=0
h, m, s, ms=int(h), int(m), int(s), int(ms)
returntime(hour=h, minute=m, second=s, microsecond=ms)
exceptValueError:
returnNone


defDate_or_None(s):
try:
returndate(
int(s[:4]),
int(s[5:7]),
int(s[8:10]),
 ) # year # month # day
exceptValueError:
returnNone


defDateTime2literal(d, c):
"""Format a DateTime object as an ISO timestamp."""
returnstring_literal(format_TIMESTAMP(d))


defDateTimeDelta2literal(d, c):
"""Format a DateTimeDelta object as a time."""
returnstring_literal(format_TIMEDELTA(d))