1 #----------------------------------------------------------------------
2 # Copyright (c) 2008 Board of Trustees, Princeton University
4 # Permission is hereby granted, free of charge, to any person obtaining
5 # a copy of this software and/or hardware specification (the "Work") to
6 # deal in the Work without restriction, including without limitation the
7 # rights to use, copy, modify, merge, publish, distribute, sublicense,
8 # and/or sell copies of the Work, and to permit persons to whom the Work
9 # is furnished to do so, subject to the following conditions:
11 # The above copyright notice and this permission notice shall be
12 # included in all copies or substantial portions of the Work.
14 # THE WORK IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16 # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
17 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
18 # HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
19 # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 # OUT OF OR IN CONNECTION WITH THE WORK OR THE USE OR OTHER DEALINGS
22 #----------------------------------------------------------------------
23 from __future__ import print_function
27 import dateutil.parser
31 from sfa.util.sfalogging import logger
32 from sfa.util.py23 import StringType
34 SFATIME_FORMAT = "%Y-%m-%dT%H:%M:%SZ"
38 """ Translate a string into a time using dateutil.parser.parse but make sure it's in UTC time and strip
39 the timezone, so that it's compatible with normal datetime.datetime objects.
41 For safety this can also handle inputs that are either timestamps, or datetimes
44 def handle_shorthands(input):
45 """recognize string like +5d or +3w or +2m as
46 2 days, 3 weeks or 2 months from now"""
47 if input.startswith('+'):
48 match = re.match(r"([0-9]+)([dwm])", input[1:])
50 how_many = int(match.group(1))
53 d = datetime.timedelta(days=how_many)
55 d = datetime.timedelta(weeks=how_many)
57 d = datetime.timedelta(weeks=4 * how_many)
58 return datetime.datetime.utcnow() + d
60 # prepare the input for the checks below by
61 # casting strings ('1327098335') to ints
62 if isinstance(input, StringType):
67 new_input = handle_shorthands(input)
68 if new_input is not None:
75 if isinstance(input, datetime.datetime):
76 #logger.info ("argument to utcparse already a datetime - doing nothing")
78 elif isinstance(input, StringType):
79 t = dateutil.parser.parse(input)
80 if t.utcoffset() is not None:
81 t = t.utcoffset() + t.replace(tzinfo=None)
83 elif isinstance(input, (int, float, long)):
84 return datetime.datetime.fromtimestamp(input)
86 logger.error("Unexpected type in utcparse [%s]" % type(input))
89 def datetime_to_string(dt):
90 return datetime.datetime.strftime(dt, SFATIME_FORMAT)
93 def datetime_to_utc(dt):
94 return time.gmtime(datetime_to_epoch(dt))
96 # see https://docs.python.org/2/library/time.html
97 # all timestamps are in UTC so time.mktime() would be *wrong*
100 def datetime_to_epoch(dt):
101 return int(calendar.timegm(dt.timetuple()))
104 def add_datetime(input, days=0, hours=0, minutes=0, seconds=0):
106 Adjust the input date by the specified delta (in seconds).
109 return dt + datetime.timedelta(days=days, hours=hours, minutes=minutes, seconds=seconds)
111 if __name__ == '__main__':
112 # checking consistency
114 print(("Should be close to zero: %s" %
115 (datetime_to_epoch(datetime.datetime.utcnow()) - time.time())))
126 '2014-05-28T15:18:30',
128 print("input=%20s -> parsed %s" %
129 (input, datetime_to_string(utcparse(input))))