# -*- coding: UTF-8 -*- """ win32timezone: Module for handling datetime.tzinfo time zones using the windows registry for time zone information. The time zone names are dependent on the registry entries defined by the operating system. This module may be tested using the doctest module. Written by Jason R. Coombs (jaraco@jaraco.com). Copyright © 2003-2012. All Rights Reserved. This module is licenced for use in Mark Hammond's pywin32 library under the same terms as the pywin32 library. To use this time zone module with the datetime module, simply pass the TimeZoneInfo object to the datetime constructor. For example, >>> import win32timezone, datetime >>> assert 'Mountain Standard Time' in win32timezone.TimeZoneInfo.get_sorted_time_zone_names() >>> MST = win32timezone.TimeZoneInfo('Mountain Standard Time') >>> now = datetime.datetime.now(MST) The now object is now a time-zone aware object, and daylight savings- aware methods may be called on it. >>> now.utcoffset() in (datetime.timedelta(-1, 61200), datetime.timedelta(-1, 64800)) True (note that the result of utcoffset call will be different based on when now was generated, unless standard time is always used) >>> now = datetime.datetime.now(TimeZoneInfo('Mountain Standard Time', True)) >>> now.utcoffset() datetime.timedelta(-1, 61200) >>> aug2 = datetime.datetime(2003, 8, 2, tzinfo = MST) >>> tuple(aug2.utctimetuple()) (2003, 8, 2, 6, 0, 0, 5, 214, 0) >>> nov2 = datetime.datetime(2003, 11, 25, tzinfo = MST) >>> tuple(nov2.utctimetuple()) (2003, 11, 25, 7, 0, 0, 1, 329, 0) To convert from one timezone to another, just use the astimezone method. >>> aug2.isoformat() '2003-08-02T00:00:00-06:00' >>> aug2est = aug2.astimezone(win32timezone.TimeZoneInfo('Eastern Standard Time')) >>> aug2est.isoformat() '2003-08-02T02:00:00-04:00' calling the displayName member will return the display name as set in the registry. >>> est = win32timezone.TimeZoneInfo('Eastern Standard Time') >>> str(est.displayName) '(UTC-05:00) Eastern Time (US & Canada)' >>> gmt = win32timezone.TimeZoneInfo('GMT Standard Time', True) >>> str(gmt.displayName) '(UTC+00:00) Dublin, Edinburgh, Lisbon, London' To get the complete list of available time zone keys, >>> zones = win32timezone.TimeZoneInfo.get_all_time_zones() If you want to get them in an order that's sorted longitudinally >>> zones = win32timezone.TimeZoneInfo.get_sorted_time_zones() TimeZoneInfo now supports being pickled and comparison >>> import pickle >>> tz = win32timezone.TimeZoneInfo('China Standard Time') >>> tz == pickle.loads(pickle.dumps(tz)) True It's possible to construct a TimeZoneInfo from a TimeZoneDescription including the currently-defined zone. >>> tz = win32timezone.TimeZoneInfo(TimeZoneDefinition.current()) >>> tz == pickle.loads(pickle.dumps(tz)) True >>> aest = win32timezone.TimeZoneInfo('AUS Eastern Standard Time') >>> est = win32timezone.TimeZoneInfo('E. Australia Standard Time') >>> dt = datetime.datetime(2006, 11, 11, 1, 0, 0, tzinfo = aest) >>> estdt = dt.astimezone(est) >>> estdt.strftime('%Y-%m-%d %H:%M:%S') '2006-11-11 00:00:00' >>> dt = datetime.datetime(2007, 1, 12, 1, 0, 0, tzinfo = aest) >>> estdt = dt.astimezone(est) >>> estdt.strftime('%Y-%m-%d %H:%M:%S') '2007-01-12 00:00:00' >>> dt = datetime.datetime(2007, 6, 13, 1, 0, 0, tzinfo = aest) >>> estdt = dt.astimezone(est) >>> estdt.strftime('%Y-%m-%d %H:%M:%S') '2007-06-13 01:00:00' Microsoft now has a patch for handling time zones in 2007 (see http://support.microsoft.com/gp/cp_dst) As a result, patched systems will give an incorrect result for dates prior to the designated year except for Vista and its successors, which have dynamic time zone support. >>> nov2_pre_change = datetime.datetime(2003, 11, 2, tzinfo = MST) >>> old_response = (2003, 11, 2, 7, 0, 0, 6, 306, 0) >>> incorrect_patch_response = (2003, 11, 2, 6, 0, 0, 6, 306, 0) >>> pre_response = nov2_pre_change.utctimetuple() >>> pre_response in (old_response, incorrect_patch_response) True Furthermore, unpatched systems pre-Vista will give an incorrect result for dates after 2007. >>> nov2_post_change = datetime.datetime(2007, 11, 2, tzinfo = MST) >>> incorrect_unpatched_response = (2007, 11, 2, 7, 0, 0, 4, 306, 0) >>> new_response = (2007, 11, 2, 6, 0, 0, 4, 306, 0) >>> post_response = nov2_post_change.utctimetuple() >>> post_response in (new_response, incorrect_unpatched_response) True There is a function you can call to get some capabilities of the time zone data. >>> caps = GetTZCapabilities() >>> isinstance(caps, dict) True >>> 'MissingTZPatch' in caps True >>> 'DynamicTZSupport' in caps True >>> both_dates_correct = (pre_response == old_response and post_response == new_response) >>> old_dates_wrong = (pre_response == incorrect_patch_response) >>> new_dates_wrong = (post_response == incorrect_unpatched_response) >>> caps['DynamicTZSupport'] == both_dates_correct True >>> (not caps['DynamicTZSupport'] and caps['MissingTZPatch']) == new_dates_wrong True >>> (not caps['DynamicTZSupport'] and not caps['MissingTZPatch']) == old_dates_wrong True This test helps ensure language support for unicode characters >>> x = TIME_ZONE_INFORMATION(0, u'français') Test conversion from one time zone to another at a DST boundary =============================================================== >>> tz_hi = TimeZoneInfo('Hawaiian Standard Time') >>> tz_pac = TimeZoneInfo('Pacific Standard Time') >>> time_before = datetime.datetime(2011, 11, 5, 15, 59, 59, tzinfo=tz_hi) >>> tz_hi.utcoffset(time_before) datetime.timedelta(-1, 50400) >>> tz_hi.dst(time_before) datetime.timedelta(0) Hawaii doesn't need dynamic TZ info >>> getattr(tz_hi, 'dynamicInfo', None) Here's a time that gave some trouble as reported in #3523104 because one minute later, the equivalent UTC time changes from DST in the U.S. >>> dt_hi = datetime.datetime(2011, 11, 5, 15, 59, 59, 0, tzinfo=tz_hi) >>> dt_hi.timetuple() time.struct_time(tm_year=2011, tm_mon=11, tm_mday=5, tm_hour=15, tm_min=59, tm_sec=59, tm_wday=5, tm_yday=309, tm_isdst=0) >>> dt_hi.utctimetuple() time.struct_time(tm_year=2011, tm_mon=11, tm_mday=6, tm_hour=1, tm_min=59, tm_sec=59, tm_wday=6, tm_yday=310, tm_isdst=0) Convert the time to pacific time. >>> dt_pac = dt_hi.astimezone(tz_pac) >>> dt_pac.timetuple() time.struct_time(tm_year=2011, tm_mon=11, tm_mday=5, tm_hour=18, tm_min=59, tm_sec=59, tm_wday=5, tm_yday=309, tm_isdst=1) Notice that the UTC time is almost 2am. >>> dt_pac.utctimetuple() time.struct_time(tm_year=2011, tm_mon=11, tm_mday=6, tm_hour=1, tm_min=59, tm_sec=59, tm_wday=6, tm_yday=310, tm_isdst=0) Now do the same tests one minute later in Hawaii. >>> time_after = datetime.datetime(2011, 11, 5, 16, 0, 0, 0, tzinfo=tz_hi) >>> tz_hi.utcoffset(time_after) datetime.timedelta(-1, 50400) >>> tz_hi.dst(time_before) datetime.timedelta(0) >>> dt_hi = datetime.datetime(2011, 11, 5, 16, 0, 0, 0, tzinfo=tz_hi) >>> print dt_hi.timetuple() time.struct_time(tm_year=2011, tm_mon=11, tm_mday=5, tm_hour=16, tm_min=0, tm_sec=0, tm_wday=5, tm_yday=309, tm_isdst=0) >>> print dt_hi.utctimetuple() time.struct_time(tm_year=2011, tm_mon=11, tm_mday=6, tm_hour=2, tm_min=0, tm_sec=0, tm_wday=6, tm_yday=310, tm_isdst=0) According to the docs, this is what astimezone does. >>> utc = (dt_hi - dt_hi.utcoffset()).replace(tzinfo=tz_pac) >>> utc datetime.datetime(2011, 11, 6, 2, 0, tzinfo=TimeZoneInfo('Pacific Standard Time')) >>> tz_pac.fromutc(utc) == dt_hi.astimezone(tz_pac) True >>> tz_pac.fromutc(utc) datetime.datetime(2011, 11, 5, 19, 0, tzinfo=TimeZoneInfo('Pacific Standard Time')) Make sure the converted time is correct. >>> dt_pac = dt_hi.astimezone(tz_pac) >>> dt_pac.timetuple() time.struct_time(tm_year=2011, tm_mon=11, tm_mday=5, tm_hour=19, tm_min=0, tm_sec=0, tm_wday=5, tm_yday=309, tm_isdst=1) >>> dt_pac.utctimetuple() time.struct_time(tm_year=2011, tm_mon=11, tm_mday=6, tm_hour=2, tm_min=0, tm_sec=0, tm_wday=6, tm_yday=310, tm_isdst=0) Check some internal methods >>> tz_pac._getStandardBias(datetime.datetime(2011, 1, 1)) datetime.timedelta(0, 28800) >>> tz_pac._getDaylightBias(datetime.datetime(2011, 1, 1)) datetime.timedelta(0, 25200) Test the offsets >>> offset = tz_pac.utcoffset(datetime.datetime(2011, 11, 6, 2, 0)) >>> offset == datetime.timedelta(hours=-8) True >>> dst_offset = tz_pac.dst(datetime.datetime(2011, 11, 6, 2, 0) + offset) >>> dst_offset == datetime.timedelta(hours=1) True >>> (offset + dst_offset) == datetime.timedelta(hours=-7) True Test offsets that occur right at the DST changeover >>> datetime.datetime.utcfromtimestamp(1320570000).replace( ... tzinfo=TimeZoneInfo.utc()).astimezone(tz_pac) datetime.datetime(2011, 11, 6, 1, 0, tzinfo=TimeZoneInfo('Pacific Standard Time')) """ from __future__ import generators __author__ = 'Jason R. Coombs ' import winreg import struct import datetime import win32api import re import operator from itertools import count import logging log = logging.getLogger(__file__) # A couple of objects for working with objects as if they were native C-type # structures. class _SimpleStruct(object): _fields_ = None # must be overridden by subclasses def __init__(self, *args, **kw): for i, (name, typ) in enumerate(self._fields_): def_arg = None if i < len(args): def_arg = args[i] if name in kw: def_arg = kw[name] if def_arg is not None: if not isinstance(def_arg, tuple): def_arg = (def_arg,) else: def_arg = () if len(def_arg) == 1 and isinstance(def_arg[0], typ): # already an object of this type. # XXX - should copy.copy??? def_val = def_arg[0] else: def_val = typ(*def_arg) setattr(self, name, def_val) def field_names(self): return [f[0] for f in self._fields_] def __eq__(self, other): if not hasattr(other, "_fields_"): return False if self._fields_ != other._fields_: return False for name, _ in self._fields_: if getattr(self, name) != getattr(other, name): return False return True def __ne__(self, other): return not self.__eq__(other) class SYSTEMTIME(_SimpleStruct): _fields_ = [ ('year', int), ('month', int), ('day_of_week', int), ('day', int), ('hour', int), ('minute', int), ('second', int), ('millisecond', int), ] class TIME_ZONE_INFORMATION(_SimpleStruct): _fields_ = [ ('bias', int), ('standard_name', str), ('standard_start', SYSTEMTIME), ('standard_bias', int), ('daylight_name', str), ('daylight_start', SYSTEMTIME), ('daylight_bias', int), ] class DYNAMIC_TIME_ZONE_INFORMATION(_SimpleStruct): _fields_ = TIME_ZONE_INFORMATION._fields_ + [ ('key_name', str), ('dynamic_daylight_time_disabled', bool), ] class TimeZoneDefinition(DYNAMIC_TIME_ZONE_INFORMATION): """ A time zone definition class based on the win32 DYNAMIC_TIME_ZONE_INFORMATION structure. Describes a bias against UTC (bias), and two dates at which a separate additional bias applies (standard_bias and daylight_bias). """ def __init__(self, *args, **kwargs): """ Try to construct a TimeZoneDefinition from a) [DYNAMIC_]TIME_ZONE_INFORMATION args b) another TimeZoneDefinition c) a byte structure (using _from_bytes) """ try: super(TimeZoneDefinition, self).__init__(*args, **kwargs) return except (TypeError, ValueError): pass try: self.__init_from_other(*args, **kwargs) return except TypeError: pass try: self.__init_from_bytes(*args, **kwargs) return except TypeError: pass raise TypeError("Invalid arguments for %s" % self.__class__) def __init_from_bytes( self, bytes, standard_name='', daylight_name='', key_name='', daylight_disabled=False): format = '3l8h8h' components = struct.unpack(format, bytes) bias, standard_bias, daylight_bias = components[:3] standard_start = SYSTEMTIME(*components[3:11]) daylight_start = SYSTEMTIME(*components[11:19]) super(TimeZoneDefinition, self).__init__( bias, standard_name, standard_start, standard_bias, daylight_name, daylight_start, daylight_bias, key_name, daylight_disabled,) def __init_from_other(self, other): if not isinstance(other, TIME_ZONE_INFORMATION): raise TypeError("Not a TIME_ZONE_INFORMATION") for name in other.field_names(): # explicitly get the value from the underlying structure value = super(TimeZoneDefinition, other).__getattribute__(other, name) setattr(self, name, value) # consider instead of the loop above just copying the memory directly # size = max(ctypes.sizeof(DYNAMIC_TIME_ZONE_INFO), ctypes.sizeof(other)) # ctypes.memmove(ctypes.addressof(self), other, size) def __getattribute__(self, attr): value = super(TimeZoneDefinition, self).__getattribute__(attr) if 'bias' in attr: value = datetime.timedelta(minutes=value) return value @classmethod def current(class_): "Windows Platform SDK GetTimeZoneInformation" code, tzi = win32api.GetTimeZoneInformation(True) return code, class_(*tzi) def set(self): tzi = tuple(getattr(self, n) for n, t in self._fields_) win32api.SetTimeZoneInformation(tzi) def copy(self): # XXX - this is no longer a copy! return self.__class__(self) def locate_daylight_start(self, year): return self._locate_day(year, self.daylight_start) def locate_standard_start(self, year): return self._locate_day(year, self.standard_start) @staticmethod def _locate_day(year, cutoff): """ Takes a SYSTEMTIME object, such as retrieved from a TIME_ZONE_INFORMATION structure or call to GetTimeZoneInformation and interprets it based on the given year to identify the actual day. This method is necessary because the SYSTEMTIME structure refers to a day by its day of the week and week of the month (e.g. 4th saturday in March). >>> SATURDAY = 6 >>> MARCH = 3 >>> st = SYSTEMTIME(2000, MARCH, SATURDAY, 4, 0, 0, 0, 0) # according to my calendar, the 4th Saturday in March in 2009 was the 28th >>> expected_date = datetime.datetime(2009, 3, 28) >>> TimeZoneDefinition._locate_day(2009, st) == expected_date True """ # MS stores Sunday as 0, Python datetime stores Monday as zero target_weekday = (cutoff.day_of_week + 6) % 7 # For SYSTEMTIMEs relating to time zone inforamtion, cutoff.day # is the week of the month week_of_month = cutoff.day # so the following is the first day of that week day = (week_of_month - 1) * 7 + 1 result = datetime.datetime( year, cutoff.month, day, cutoff.hour, cutoff.minute, cutoff.second, cutoff.millisecond) # now the result is the correct week, but not necessarily the correct day of the week days_to_go = (target_weekday - result.weekday()) % 7 result += datetime.timedelta(days_to_go) # if we selected a day in the month following the target month, # move back a week or two. # This is necessary because Microsoft defines the fifth week in a month # to be the last week in a month and adding the time delta might have # pushed the result into the next month. while result.month == cutoff.month + 1: result -= datetime.timedelta(weeks = 1) return result class TimeZoneInfo(datetime.tzinfo): """ Main class for handling Windows time zones. Usage: TimeZoneInfo(