Mercurial > repos > guerler > springsuite
comparison planemo/lib/python3.7/site-packages/dateutil/tz/win.py @ 0:d30785e31577 draft
"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
| author | guerler |
|---|---|
| date | Fri, 31 Jul 2020 00:18:57 -0400 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| -1:000000000000 | 0:d30785e31577 |
|---|---|
| 1 # -*- coding: utf-8 -*- | |
| 2 """ | |
| 3 This module provides an interface to the native time zone data on Windows, | |
| 4 including :py:class:`datetime.tzinfo` implementations. | |
| 5 | |
| 6 Attempting to import this module on a non-Windows platform will raise an | |
| 7 :py:obj:`ImportError`. | |
| 8 """ | |
| 9 # This code was originally contributed by Jeffrey Harris. | |
| 10 import datetime | |
| 11 import struct | |
| 12 | |
| 13 from six.moves import winreg | |
| 14 from six import text_type | |
| 15 | |
| 16 try: | |
| 17 import ctypes | |
| 18 from ctypes import wintypes | |
| 19 except ValueError: | |
| 20 # ValueError is raised on non-Windows systems for some horrible reason. | |
| 21 raise ImportError("Running tzwin on non-Windows system") | |
| 22 | |
| 23 from ._common import tzrangebase | |
| 24 | |
| 25 __all__ = ["tzwin", "tzwinlocal", "tzres"] | |
| 26 | |
| 27 ONEWEEK = datetime.timedelta(7) | |
| 28 | |
| 29 TZKEYNAMENT = r"SOFTWARE\Microsoft\Windows NT\CurrentVersion\Time Zones" | |
| 30 TZKEYNAME9X = r"SOFTWARE\Microsoft\Windows\CurrentVersion\Time Zones" | |
| 31 TZLOCALKEYNAME = r"SYSTEM\CurrentControlSet\Control\TimeZoneInformation" | |
| 32 | |
| 33 | |
| 34 def _settzkeyname(): | |
| 35 handle = winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE) | |
| 36 try: | |
| 37 winreg.OpenKey(handle, TZKEYNAMENT).Close() | |
| 38 TZKEYNAME = TZKEYNAMENT | |
| 39 except WindowsError: | |
| 40 TZKEYNAME = TZKEYNAME9X | |
| 41 handle.Close() | |
| 42 return TZKEYNAME | |
| 43 | |
| 44 | |
| 45 TZKEYNAME = _settzkeyname() | |
| 46 | |
| 47 | |
| 48 class tzres(object): | |
| 49 """ | |
| 50 Class for accessing ``tzres.dll``, which contains timezone name related | |
| 51 resources. | |
| 52 | |
| 53 .. versionadded:: 2.5.0 | |
| 54 """ | |
| 55 p_wchar = ctypes.POINTER(wintypes.WCHAR) # Pointer to a wide char | |
| 56 | |
| 57 def __init__(self, tzres_loc='tzres.dll'): | |
| 58 # Load the user32 DLL so we can load strings from tzres | |
| 59 user32 = ctypes.WinDLL('user32') | |
| 60 | |
| 61 # Specify the LoadStringW function | |
| 62 user32.LoadStringW.argtypes = (wintypes.HINSTANCE, | |
| 63 wintypes.UINT, | |
| 64 wintypes.LPWSTR, | |
| 65 ctypes.c_int) | |
| 66 | |
| 67 self.LoadStringW = user32.LoadStringW | |
| 68 self._tzres = ctypes.WinDLL(tzres_loc) | |
| 69 self.tzres_loc = tzres_loc | |
| 70 | |
| 71 def load_name(self, offset): | |
| 72 """ | |
| 73 Load a timezone name from a DLL offset (integer). | |
| 74 | |
| 75 >>> from dateutil.tzwin import tzres | |
| 76 >>> tzr = tzres() | |
| 77 >>> print(tzr.load_name(112)) | |
| 78 'Eastern Standard Time' | |
| 79 | |
| 80 :param offset: | |
| 81 A positive integer value referring to a string from the tzres dll. | |
| 82 | |
| 83 .. note:: | |
| 84 | |
| 85 Offsets found in the registry are generally of the form | |
| 86 ``@tzres.dll,-114``. The offset in this case is 114, not -114. | |
| 87 | |
| 88 """ | |
| 89 resource = self.p_wchar() | |
| 90 lpBuffer = ctypes.cast(ctypes.byref(resource), wintypes.LPWSTR) | |
| 91 nchar = self.LoadStringW(self._tzres._handle, offset, lpBuffer, 0) | |
| 92 return resource[:nchar] | |
| 93 | |
| 94 def name_from_string(self, tzname_str): | |
| 95 """ | |
| 96 Parse strings as returned from the Windows registry into the time zone | |
| 97 name as defined in the registry. | |
| 98 | |
| 99 >>> from dateutil.tzwin import tzres | |
| 100 >>> tzr = tzres() | |
| 101 >>> print(tzr.name_from_string('@tzres.dll,-251')) | |
| 102 'Dateline Daylight Time' | |
| 103 >>> print(tzr.name_from_string('Eastern Standard Time')) | |
| 104 'Eastern Standard Time' | |
| 105 | |
| 106 :param tzname_str: | |
| 107 A timezone name string as returned from a Windows registry key. | |
| 108 | |
| 109 :return: | |
| 110 Returns the localized timezone string from tzres.dll if the string | |
| 111 is of the form `@tzres.dll,-offset`, else returns the input string. | |
| 112 """ | |
| 113 if not tzname_str.startswith('@'): | |
| 114 return tzname_str | |
| 115 | |
| 116 name_splt = tzname_str.split(',-') | |
| 117 try: | |
| 118 offset = int(name_splt[1]) | |
| 119 except: | |
| 120 raise ValueError("Malformed timezone string.") | |
| 121 | |
| 122 return self.load_name(offset) | |
| 123 | |
| 124 | |
| 125 class tzwinbase(tzrangebase): | |
| 126 """tzinfo class based on win32's timezones available in the registry.""" | |
| 127 def __init__(self): | |
| 128 raise NotImplementedError('tzwinbase is an abstract base class') | |
| 129 | |
| 130 def __eq__(self, other): | |
| 131 # Compare on all relevant dimensions, including name. | |
| 132 if not isinstance(other, tzwinbase): | |
| 133 return NotImplemented | |
| 134 | |
| 135 return (self._std_offset == other._std_offset and | |
| 136 self._dst_offset == other._dst_offset and | |
| 137 self._stddayofweek == other._stddayofweek and | |
| 138 self._dstdayofweek == other._dstdayofweek and | |
| 139 self._stdweeknumber == other._stdweeknumber and | |
| 140 self._dstweeknumber == other._dstweeknumber and | |
| 141 self._stdhour == other._stdhour and | |
| 142 self._dsthour == other._dsthour and | |
| 143 self._stdminute == other._stdminute and | |
| 144 self._dstminute == other._dstminute and | |
| 145 self._std_abbr == other._std_abbr and | |
| 146 self._dst_abbr == other._dst_abbr) | |
| 147 | |
| 148 @staticmethod | |
| 149 def list(): | |
| 150 """Return a list of all time zones known to the system.""" | |
| 151 with winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE) as handle: | |
| 152 with winreg.OpenKey(handle, TZKEYNAME) as tzkey: | |
| 153 result = [winreg.EnumKey(tzkey, i) | |
| 154 for i in range(winreg.QueryInfoKey(tzkey)[0])] | |
| 155 return result | |
| 156 | |
| 157 def display(self): | |
| 158 """ | |
| 159 Return the display name of the time zone. | |
| 160 """ | |
| 161 return self._display | |
| 162 | |
| 163 def transitions(self, year): | |
| 164 """ | |
| 165 For a given year, get the DST on and off transition times, expressed | |
| 166 always on the standard time side. For zones with no transitions, this | |
| 167 function returns ``None``. | |
| 168 | |
| 169 :param year: | |
| 170 The year whose transitions you would like to query. | |
| 171 | |
| 172 :return: | |
| 173 Returns a :class:`tuple` of :class:`datetime.datetime` objects, | |
| 174 ``(dston, dstoff)`` for zones with an annual DST transition, or | |
| 175 ``None`` for fixed offset zones. | |
| 176 """ | |
| 177 | |
| 178 if not self.hasdst: | |
| 179 return None | |
| 180 | |
| 181 dston = picknthweekday(year, self._dstmonth, self._dstdayofweek, | |
| 182 self._dsthour, self._dstminute, | |
| 183 self._dstweeknumber) | |
| 184 | |
| 185 dstoff = picknthweekday(year, self._stdmonth, self._stddayofweek, | |
| 186 self._stdhour, self._stdminute, | |
| 187 self._stdweeknumber) | |
| 188 | |
| 189 # Ambiguous dates default to the STD side | |
| 190 dstoff -= self._dst_base_offset | |
| 191 | |
| 192 return dston, dstoff | |
| 193 | |
| 194 def _get_hasdst(self): | |
| 195 return self._dstmonth != 0 | |
| 196 | |
| 197 @property | |
| 198 def _dst_base_offset(self): | |
| 199 return self._dst_base_offset_ | |
| 200 | |
| 201 | |
| 202 class tzwin(tzwinbase): | |
| 203 """ | |
| 204 Time zone object created from the zone info in the Windows registry | |
| 205 | |
| 206 These are similar to :py:class:`dateutil.tz.tzrange` objects in that | |
| 207 the time zone data is provided in the format of a single offset rule | |
| 208 for either 0 or 2 time zone transitions per year. | |
| 209 | |
| 210 :param: name | |
| 211 The name of a Windows time zone key, e.g. "Eastern Standard Time". | |
| 212 The full list of keys can be retrieved with :func:`tzwin.list`. | |
| 213 """ | |
| 214 | |
| 215 def __init__(self, name): | |
| 216 self._name = name | |
| 217 | |
| 218 with winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE) as handle: | |
| 219 tzkeyname = text_type("{kn}\\{name}").format(kn=TZKEYNAME, name=name) | |
| 220 with winreg.OpenKey(handle, tzkeyname) as tzkey: | |
| 221 keydict = valuestodict(tzkey) | |
| 222 | |
| 223 self._std_abbr = keydict["Std"] | |
| 224 self._dst_abbr = keydict["Dlt"] | |
| 225 | |
| 226 self._display = keydict["Display"] | |
| 227 | |
| 228 # See http://ww_winreg.jsiinc.com/SUBA/tip0300/rh0398.htm | |
| 229 tup = struct.unpack("=3l16h", keydict["TZI"]) | |
| 230 stdoffset = -tup[0]-tup[1] # Bias + StandardBias * -1 | |
| 231 dstoffset = stdoffset-tup[2] # + DaylightBias * -1 | |
| 232 self._std_offset = datetime.timedelta(minutes=stdoffset) | |
| 233 self._dst_offset = datetime.timedelta(minutes=dstoffset) | |
| 234 | |
| 235 # for the meaning see the win32 TIME_ZONE_INFORMATION structure docs | |
| 236 # http://msdn.microsoft.com/en-us/library/windows/desktop/ms725481(v=vs.85).aspx | |
| 237 (self._stdmonth, | |
| 238 self._stddayofweek, # Sunday = 0 | |
| 239 self._stdweeknumber, # Last = 5 | |
| 240 self._stdhour, | |
| 241 self._stdminute) = tup[4:9] | |
| 242 | |
| 243 (self._dstmonth, | |
| 244 self._dstdayofweek, # Sunday = 0 | |
| 245 self._dstweeknumber, # Last = 5 | |
| 246 self._dsthour, | |
| 247 self._dstminute) = tup[12:17] | |
| 248 | |
| 249 self._dst_base_offset_ = self._dst_offset - self._std_offset | |
| 250 self.hasdst = self._get_hasdst() | |
| 251 | |
| 252 def __repr__(self): | |
| 253 return "tzwin(%s)" % repr(self._name) | |
| 254 | |
| 255 def __reduce__(self): | |
| 256 return (self.__class__, (self._name,)) | |
| 257 | |
| 258 | |
| 259 class tzwinlocal(tzwinbase): | |
| 260 """ | |
| 261 Class representing the local time zone information in the Windows registry | |
| 262 | |
| 263 While :class:`dateutil.tz.tzlocal` makes system calls (via the :mod:`time` | |
| 264 module) to retrieve time zone information, ``tzwinlocal`` retrieves the | |
| 265 rules directly from the Windows registry and creates an object like | |
| 266 :class:`dateutil.tz.tzwin`. | |
| 267 | |
| 268 Because Windows does not have an equivalent of :func:`time.tzset`, on | |
| 269 Windows, :class:`dateutil.tz.tzlocal` instances will always reflect the | |
| 270 time zone settings *at the time that the process was started*, meaning | |
| 271 changes to the machine's time zone settings during the run of a program | |
| 272 on Windows will **not** be reflected by :class:`dateutil.tz.tzlocal`. | |
| 273 Because ``tzwinlocal`` reads the registry directly, it is unaffected by | |
| 274 this issue. | |
| 275 """ | |
| 276 def __init__(self): | |
| 277 with winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE) as handle: | |
| 278 with winreg.OpenKey(handle, TZLOCALKEYNAME) as tzlocalkey: | |
| 279 keydict = valuestodict(tzlocalkey) | |
| 280 | |
| 281 self._std_abbr = keydict["StandardName"] | |
| 282 self._dst_abbr = keydict["DaylightName"] | |
| 283 | |
| 284 try: | |
| 285 tzkeyname = text_type('{kn}\\{sn}').format(kn=TZKEYNAME, | |
| 286 sn=self._std_abbr) | |
| 287 with winreg.OpenKey(handle, tzkeyname) as tzkey: | |
| 288 _keydict = valuestodict(tzkey) | |
| 289 self._display = _keydict["Display"] | |
| 290 except OSError: | |
| 291 self._display = None | |
| 292 | |
| 293 stdoffset = -keydict["Bias"]-keydict["StandardBias"] | |
| 294 dstoffset = stdoffset-keydict["DaylightBias"] | |
| 295 | |
| 296 self._std_offset = datetime.timedelta(minutes=stdoffset) | |
| 297 self._dst_offset = datetime.timedelta(minutes=dstoffset) | |
| 298 | |
| 299 # For reasons unclear, in this particular key, the day of week has been | |
| 300 # moved to the END of the SYSTEMTIME structure. | |
| 301 tup = struct.unpack("=8h", keydict["StandardStart"]) | |
| 302 | |
| 303 (self._stdmonth, | |
| 304 self._stdweeknumber, # Last = 5 | |
| 305 self._stdhour, | |
| 306 self._stdminute) = tup[1:5] | |
| 307 | |
| 308 self._stddayofweek = tup[7] | |
| 309 | |
| 310 tup = struct.unpack("=8h", keydict["DaylightStart"]) | |
| 311 | |
| 312 (self._dstmonth, | |
| 313 self._dstweeknumber, # Last = 5 | |
| 314 self._dsthour, | |
| 315 self._dstminute) = tup[1:5] | |
| 316 | |
| 317 self._dstdayofweek = tup[7] | |
| 318 | |
| 319 self._dst_base_offset_ = self._dst_offset - self._std_offset | |
| 320 self.hasdst = self._get_hasdst() | |
| 321 | |
| 322 def __repr__(self): | |
| 323 return "tzwinlocal()" | |
| 324 | |
| 325 def __str__(self): | |
| 326 # str will return the standard name, not the daylight name. | |
| 327 return "tzwinlocal(%s)" % repr(self._std_abbr) | |
| 328 | |
| 329 def __reduce__(self): | |
| 330 return (self.__class__, ()) | |
| 331 | |
| 332 | |
| 333 def picknthweekday(year, month, dayofweek, hour, minute, whichweek): | |
| 334 """ dayofweek == 0 means Sunday, whichweek 5 means last instance """ | |
| 335 first = datetime.datetime(year, month, 1, hour, minute) | |
| 336 | |
| 337 # This will work if dayofweek is ISO weekday (1-7) or Microsoft-style (0-6), | |
| 338 # Because 7 % 7 = 0 | |
| 339 weekdayone = first.replace(day=((dayofweek - first.isoweekday()) % 7) + 1) | |
| 340 wd = weekdayone + ((whichweek - 1) * ONEWEEK) | |
| 341 if (wd.month != month): | |
| 342 wd -= ONEWEEK | |
| 343 | |
| 344 return wd | |
| 345 | |
| 346 | |
| 347 def valuestodict(key): | |
| 348 """Convert a registry key's values to a dictionary.""" | |
| 349 dout = {} | |
| 350 size = winreg.QueryInfoKey(key)[1] | |
| 351 tz_res = None | |
| 352 | |
| 353 for i in range(size): | |
| 354 key_name, value, dtype = winreg.EnumValue(key, i) | |
| 355 if dtype == winreg.REG_DWORD or dtype == winreg.REG_DWORD_LITTLE_ENDIAN: | |
| 356 # If it's a DWORD (32-bit integer), it's stored as unsigned - convert | |
| 357 # that to a proper signed integer | |
| 358 if value & (1 << 31): | |
| 359 value = value - (1 << 32) | |
| 360 elif dtype == winreg.REG_SZ: | |
| 361 # If it's a reference to the tzres DLL, load the actual string | |
| 362 if value.startswith('@tzres'): | |
| 363 tz_res = tz_res or tzres() | |
| 364 value = tz_res.name_from_string(value) | |
| 365 | |
| 366 value = value.rstrip('\x00') # Remove trailing nulls | |
| 367 | |
| 368 dout[key_name] = value | |
| 369 | |
| 370 return dout |
