src/nepi/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 lock is 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
  29     :param date : user define date matchin the pattern _strf
  30     :type date : date
  31
  32     """
  33     if not date:
  34         return strfnow()
  35     if _reabs.match(date):
  36         return date
  37     m = _rerel.match(date)
  38     if m:
  39         time = float(m.groupdict()['time'])
  40         units = m.groupdict()['units']
  41         if units == 'h':
  42             delta = datetime.timedelta(hours = time)
  43         elif units == 'm':
  44             delta = datetime.timedelta(minutes = time)
  45         elif units == 's':
  46             delta = datetime.timedelta(seconds = time)
  47         elif units == 'ms':
  48             delta = datetime.timedelta(microseconds = (time*1000))
  49         else:
  50             delta = datetime.timedelta(microseconds = time)
  51         now = datetime.datetime.now()
  52         d = now + delta
  53         return d.strftime(_strf)
  54     return None
  55