src/neco/util/timefuncs.py

   1 import datetime
   2 import re
   3
   4 _strf = "%Y%m%d%H%M%S%f"
   5 _reabs = re.compile("^\d{20}$")
   6 _rerel = re.compile("^(?P<time>\d+(.\d+)?)(?P<units>h|m|s|ms|us)$")
   7
   8 # Work around to fix "ImportError: Failed to import _strptime because the import lockis held by another thread."
   9 datetime.datetime.strptime("20120807124732894211", _strf)
  10
  11 def strfnow():
  12     """ Current date """
  13     return datetime.datetime.now().strftime(_strf)
  14
  15 def strfdiff(str1, str2):
  16     # Time difference in seconds without ignoring miliseconds
  17     d1 = datetime.datetime.strptime(str1, _strf)
  18     d2 = datetime.datetime.strptime(str2, _strf)
  19     diff = d1 - d2
  20     ddays = diff.days * 86400
  21     dus = round(diff.microseconds * 1.0e-06, 2)
  22     ret = ddays + diff.seconds + dus
  23     # delay must be > 0
  24     return (ret or 0.001)
  25
  26 def strfvalid(date):
  27     """ User defined date to scheduler date """
  28     if not date:
  29         return strfnow()
  30     if _reabs.match(date):
  31         return date
  32     m = _rerel.match(date)
  33     if m:
  34         time = float(m.groupdict()['time'])
  35         units = m.groupdict()['units']
  36         if units == 'h':
  37             delta = datetime.timedelta(hours = time)
  38         elif units == 'm':
  39             delta = datetime.timedelta(minutes = time)
  40         elif units == 's':
  41             delta = datetime.timedelta(seconds = time)
  42         elif units == 'ms':
  43             delta = datetime.timedelta(microseconds = (time*1000))
  44         else:
  45             delta = datetime.timedelta(microseconds = time)
  46         now = datetime.datetime.now()
  47         d = now + delta
  48         return d.strftime(_strf)
  49     return None
  50