fix name error
[sfa.git] / sfa / util / sfatime.py
1 from types import StringTypes
2 import dateutil.parser
3 import datetime
4 import time
5
6 from sfa.util.sfalogging import logger
7
8 DATEFORMAT = "%Y-%m-%dT%H:%M:%SZ"
9
10 def utcparse(input):
11     """ Translate a string into a time using dateutil.parser.parse but make sure it's in UTC time and strip
12 the timezone, so that it's compatible with normal datetime.datetime objects.
13
14 For safety this can also handle inputs that are either timestamps, or datetimes
15 """
16     
17     if isinstance (input, datetime.datetime):
18         logger.warn ("argument to utcparse already a datetime - doing nothing")
19         return input
20     elif isinstance (input, StringTypes):
21         t = dateutil.parser.parse(input)
22         if t.utcoffset() is not None:
23             t = t.utcoffset() + t.replace(tzinfo=None)
24         return t
25     elif isinstance (input, (int,float,long)):
26         return datetime.datetime.fromtimestamp(input)
27     else:
28         logger.error("Unexpected type in utcparse [%s]"%type(input))
29
30 def datetime_to_string(input):
31     return datetime.datetime.strftime(input, DATEFORMAT)
32
33 def datetime_to_utc(input):
34     return time.gmtime(datetime_to_epoch(input))    
35
36 def datetime_to_epoch(input):
37     return time.mktime(input.timetuple())
38
39
40