mirror of
https://github.com/davidkeegan/dklrt
synced 2024-12-25 09:59:19 +01:00
89 lines
2.6 KiB
Python
Executable file
89 lines
2.6 KiB
Python
Executable file
#!/usr/bin/python
|
|
# Time and Date Utilities (dklrt).
|
|
# (c) David Keegan 2011-08-06.
|
|
import sys, re
|
|
from time import *
|
|
import datetime
|
|
import Misc
|
|
|
|
ModuleName = __name__
|
|
ReDateSep = '[-/]'
|
|
ReDate = '\d{4}%s\d{1,2}%s\d{1,2}' % (ReDateSep, ReDateSep)
|
|
RePeriod = '(\d+)([ymwdh])'
|
|
|
|
DateFormat = '%Y-%m-%d'
|
|
|
|
ReDateTimeSep = "[-/: ]";
|
|
DateTimeFormat = '%Y%m%d%H%M%S'
|
|
|
|
SecPerHour = 60
|
|
SecPerDay = 24 * SecPerHour * SecPerHour
|
|
|
|
def _Throw(Msg): Misc.Throw(Msg, ModuleName)
|
|
|
|
def DateTimeParse(DateTimeStr):
|
|
"""Converts a date(/time) string to seconds since the epoch.
|
|
Assumes zeroes for missing time components.
|
|
"""
|
|
Dts = re.sub(ReDateTimeSep, '', DateTimeStr);
|
|
if len(Dts) < 8:
|
|
_Throw('Bad Date/Time string: "%s"!', DateTimeStr)
|
|
while len(Dts) < 14: Dts = Dts + "0";
|
|
return mktime(strptime(Dts, DateTimeFormat))
|
|
|
|
def DateToText(Seconds):
|
|
# Round seconds to integer first as we're truncating the time
|
|
# component.
|
|
return strftime(DateFormat, localtime(round(Seconds)))
|
|
|
|
def DateToday():
|
|
return DateTimeParse(DateToText(time()))
|
|
|
|
def DateAddPeriod(Seconds, Periodstr):
|
|
"""Adds the period to the Seconds (a date)."""
|
|
|
|
Match = re.match(RePeriod, Periodstr)
|
|
if not Match: _Throw("Bad Period String: %s!", Periodstr)
|
|
Count = int(Match.group(1))
|
|
Unit = Match.group(2)
|
|
|
|
Rv = Seconds
|
|
if Unit == 'y': Rv = DateAddYears(Rv, Count)
|
|
elif Unit== 'm': Rv = DateAddMonths(Rv, Count)
|
|
elif Unit == 'w': Rv = Rv + (Count * SecPerDay * 7)
|
|
elif Unit == 'd': Rv = Rv + (Count * SecPerDay)
|
|
elif Unit == 'h': Rv = Rv + (Count * SecPerHour)
|
|
else: _Throw('Bad Period Unit: "%s"!' % Unit)
|
|
return Rv
|
|
|
|
def DateAddYears(Seconds, Count):
|
|
"""Shifts Seconds (a date) forward by Count years.
|
|
If Seconds is Feb 29, shifts to Feb 28, even if shifing to a
|
|
leap year.
|
|
"""
|
|
if not isinstance(Count, (int, long)):
|
|
_Throw("Count argument not an int!")
|
|
|
|
dtd = datetime.date.fromtimestamp(Seconds)
|
|
if not Count == 0:
|
|
if (dtd.month == 2) and (dtd.day == 29):
|
|
dtd = dtd.replace(day=28)
|
|
dtd = dtd.replace(year=(dtd.year + Count))
|
|
return mktime(dtd.timetuple())
|
|
|
|
def DateAddMonths(Seconds, Count):
|
|
"""Shifts Seconds (a date) forward by Count months.
|
|
If the day is >= 29, shifts to 28.
|
|
"""
|
|
if not isinstance(Count, (int, long)):
|
|
_Throw("Count argument not an int!")
|
|
|
|
dtd = datetime.date.fromtimestamp(Seconds)
|
|
if not Count == 0:
|
|
if dtd.day >= 29: dtd = dtd.replace(day=28)
|
|
Month = (dtd.month + Count) - 1
|
|
Years = Month / 12
|
|
dtd = dtd.replace(year=(dtd.year + Years))
|
|
Month = (Month % 12) + 1
|
|
dtd = dtd.replace(month=Month)
|
|
return mktime(dtd.timetuple())
|