1
|
|
# -*- coding: utf-8 -*-
|
2
|
20
|
"""
|
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
|
20
|
import datetime
|
11
|
20
|
import struct
|
12
|
|
|
13
|
20
|
from six.moves import winreg
|
14
|
9
|
from six import text_type
|
15
|
|
|
16
|
9
|
try:
|
17
|
9
|
import ctypes
|
18
|
9
|
from ctypes import wintypes
|
19
|
0
|
except ValueError:
|
20
|
|
# ValueError is raised on non-Windows systems for some horrible reason.
|
21
|
0
|
raise ImportError("Running tzwin on non-Windows system")
|
22
|
|
|
23
|
9
|
from ._common import tzrangebase
|
24
|
|
|
25
|
9
|
__all__ = ["tzwin", "tzwinlocal", "tzres"]
|
26
|
|
|
27
|
9
|
ONEWEEK = datetime.timedelta(7)
|
28
|
|
|
29
|
9
|
TZKEYNAMENT = r"SOFTWARE\Microsoft\Windows NT\CurrentVersion\Time Zones"
|
30
|
9
|
TZKEYNAME9X = r"SOFTWARE\Microsoft\Windows\CurrentVersion\Time Zones"
|
31
|
9
|
TZLOCALKEYNAME = r"SYSTEM\CurrentControlSet\Control\TimeZoneInformation"
|
32
|
|
|
33
|
|
|
34
|
9
|
def _settzkeyname():
|
35
|
9
|
handle = winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE)
|
36
|
9
|
try:
|
37
|
9
|
winreg.OpenKey(handle, TZKEYNAMENT).Close()
|
38
|
9
|
TZKEYNAME = TZKEYNAMENT
|
39
|
0
|
except WindowsError:
|
40
|
0
|
TZKEYNAME = TZKEYNAME9X
|
41
|
9
|
handle.Close()
|
42
|
9
|
return TZKEYNAME
|
43
|
|
|
44
|
|
|
45
|
9
|
TZKEYNAME = _settzkeyname()
|
46
|
|
|
47
|
|
|
48
|
9
|
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
|
9
|
p_wchar = ctypes.POINTER(wintypes.WCHAR) # Pointer to a wide char
|
56
|
|
|
57
|
9
|
def __init__(self, tzres_loc='tzres.dll'):
|
58
|
|
# Load the user32 DLL so we can load strings from tzres
|
59
|
9
|
user32 = ctypes.WinDLL('user32')
|
60
|
|
|
61
|
|
# Specify the LoadStringW function
|
62
|
9
|
user32.LoadStringW.argtypes = (wintypes.HINSTANCE,
|
63
|
|
wintypes.UINT,
|
64
|
|
wintypes.LPWSTR,
|
65
|
|
ctypes.c_int)
|
66
|
|
|
67
|
9
|
self.LoadStringW = user32.LoadStringW
|
68
|
9
|
self._tzres = ctypes.WinDLL(tzres_loc)
|
69
|
9
|
self.tzres_loc = tzres_loc
|
70
|
|
|
71
|
9
|
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
|
9
|
resource = self.p_wchar()
|
90
|
9
|
lpBuffer = ctypes.cast(ctypes.byref(resource), wintypes.LPWSTR)
|
91
|
9
|
nchar = self.LoadStringW(self._tzres._handle, offset, lpBuffer, 0)
|
92
|
9
|
return resource[:nchar]
|
93
|
|
|
94
|
9
|
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
|
9
|
if not tzname_str.startswith('@'):
|
114
|
9
|
return tzname_str
|
115
|
|
|
116
|
9
|
name_splt = tzname_str.split(',-')
|
117
|
9
|
try:
|
118
|
9
|
offset = int(name_splt[1])
|
119
|
9
|
except:
|
120
|
9
|
raise ValueError("Malformed timezone string.")
|
121
|
|
|
122
|
9
|
return self.load_name(offset)
|
123
|
|
|
124
|
|
|
125
|
9
|
class tzwinbase(tzrangebase):
|
126
|
|
"""tzinfo class based on win32's timezones available in the registry."""
|
127
|
9
|
def __init__(self):
|
128
|
0
|
raise NotImplementedError('tzwinbase is an abstract base class')
|
129
|
|
|
130
|
9
|
def __eq__(self, other):
|
131
|
|
# Compare on all relevant dimensions, including name.
|
132
|
9
|
if not isinstance(other, tzwinbase):
|
133
|
9
|
return NotImplemented
|
134
|
|
|
135
|
9
|
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
|
9
|
@staticmethod
|
149
|
0
|
def list():
|
150
|
|
"""Return a list of all time zones known to the system."""
|
151
|
0
|
with winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE) as handle:
|
152
|
0
|
with winreg.OpenKey(handle, TZKEYNAME) as tzkey:
|
153
|
0
|
result = [winreg.EnumKey(tzkey, i)
|
154
|
|
for i in range(winreg.QueryInfoKey(tzkey)[0])]
|
155
|
0
|
return result
|
156
|
|
|
157
|
9
|
def display(self):
|
158
|
|
"""
|
159
|
|
Return the display name of the time zone.
|
160
|
|
"""
|
161
|
0
|
return self._display
|
162
|
|
|
163
|
9
|
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
|
9
|
if not self.hasdst:
|
179
|
9
|
return None
|
180
|
|
|
181
|
9
|
dston = picknthweekday(year, self._dstmonth, self._dstdayofweek,
|
182
|
|
self._dsthour, self._dstminute,
|
183
|
|
self._dstweeknumber)
|
184
|
|
|
185
|
9
|
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
|
9
|
dstoff -= self._dst_base_offset
|
191
|
|
|
192
|
9
|
return dston, dstoff
|
193
|
|
|
194
|
9
|
def _get_hasdst(self):
|
195
|
9
|
return self._dstmonth != 0
|
196
|
|
|
197
|
9
|
@property
|
198
|
0
|
def _dst_base_offset(self):
|
199
|
9
|
return self._dst_base_offset_
|
200
|
|
|
201
|
|
|
202
|
9
|
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
|
9
|
def __init__(self, name):
|
216
|
9
|
self._name = name
|
217
|
|
|
218
|
9
|
with winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE) as handle:
|
219
|
9
|
tzkeyname = text_type("{kn}\\{name}").format(kn=TZKEYNAME, name=name)
|
220
|
9
|
with winreg.OpenKey(handle, tzkeyname) as tzkey:
|
221
|
9
|
keydict = valuestodict(tzkey)
|
222
|
|
|
223
|
9
|
self._std_abbr = keydict["Std"]
|
224
|
9
|
self._dst_abbr = keydict["Dlt"]
|
225
|
|
|
226
|
9
|
self._display = keydict["Display"]
|
227
|
|
|
228
|
|
# See http://ww_winreg.jsiinc.com/SUBA/tip0300/rh0398.htm
|
229
|
9
|
tup = struct.unpack("=3l16h", keydict["TZI"])
|
230
|
9
|
stdoffset = -tup[0]-tup[1] # Bias + StandardBias * -1
|
231
|
9
|
dstoffset = stdoffset-tup[2] # + DaylightBias * -1
|
232
|
9
|
self._std_offset = datetime.timedelta(minutes=stdoffset)
|
233
|
9
|
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
|
9
|
(self._stdmonth,
|
238
|
|
self._stddayofweek, # Sunday = 0
|
239
|
|
self._stdweeknumber, # Last = 5
|
240
|
|
self._stdhour,
|
241
|
|
self._stdminute) = tup[4:9]
|
242
|
|
|
243
|
9
|
(self._dstmonth,
|
244
|
|
self._dstdayofweek, # Sunday = 0
|
245
|
|
self._dstweeknumber, # Last = 5
|
246
|
|
self._dsthour,
|
247
|
|
self._dstminute) = tup[12:17]
|
248
|
|
|
249
|
9
|
self._dst_base_offset_ = self._dst_offset - self._std_offset
|
250
|
9
|
self.hasdst = self._get_hasdst()
|
251
|
|
|
252
|
9
|
def __repr__(self):
|
253
|
9
|
return "tzwin(%s)" % repr(self._name)
|
254
|
|
|
255
|
9
|
def __reduce__(self):
|
256
|
0
|
return (self.__class__, (self._name,))
|
257
|
|
|
258
|
|
|
259
|
9
|
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
|
9
|
def __init__(self):
|
277
|
9
|
with winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE) as handle:
|
278
|
9
|
with winreg.OpenKey(handle, TZLOCALKEYNAME) as tzlocalkey:
|
279
|
9
|
keydict = valuestodict(tzlocalkey)
|
280
|
|
|
281
|
9
|
self._std_abbr = keydict["StandardName"]
|
282
|
9
|
self._dst_abbr = keydict["DaylightName"]
|
283
|
|
|
284
|
9
|
try:
|
285
|
9
|
tzkeyname = text_type('{kn}\\{sn}').format(kn=TZKEYNAME,
|
286
|
|
sn=self._std_abbr)
|
287
|
9
|
with winreg.OpenKey(handle, tzkeyname) as tzkey:
|
288
|
7
|
_keydict = valuestodict(tzkey)
|
289
|
7
|
self._display = _keydict["Display"]
|
290
|
9
|
except OSError:
|
291
|
9
|
self._display = None
|
292
|
|
|
293
|
9
|
stdoffset = -keydict["Bias"]-keydict["StandardBias"]
|
294
|
9
|
dstoffset = stdoffset-keydict["DaylightBias"]
|
295
|
|
|
296
|
9
|
self._std_offset = datetime.timedelta(minutes=stdoffset)
|
297
|
9
|
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
|
9
|
tup = struct.unpack("=8h", keydict["StandardStart"])
|
302
|
|
|
303
|
9
|
(self._stdmonth,
|
304
|
|
self._stdweeknumber, # Last = 5
|
305
|
|
self._stdhour,
|
306
|
|
self._stdminute) = tup[1:5]
|
307
|
|
|
308
|
9
|
self._stddayofweek = tup[7]
|
309
|
|
|
310
|
9
|
tup = struct.unpack("=8h", keydict["DaylightStart"])
|
311
|
|
|
312
|
9
|
(self._dstmonth,
|
313
|
|
self._dstweeknumber, # Last = 5
|
314
|
|
self._dsthour,
|
315
|
|
self._dstminute) = tup[1:5]
|
316
|
|
|
317
|
9
|
self._dstdayofweek = tup[7]
|
318
|
|
|
319
|
9
|
self._dst_base_offset_ = self._dst_offset - self._std_offset
|
320
|
9
|
self.hasdst = self._get_hasdst()
|
321
|
|
|
322
|
9
|
def __repr__(self):
|
323
|
9
|
return "tzwinlocal()"
|
324
|
|
|
325
|
9
|
def __str__(self):
|
326
|
|
# str will return the standard name, not the daylight name.
|
327
|
7
|
return "tzwinlocal(%s)" % repr(self._std_abbr)
|
328
|
|
|
329
|
9
|
def __reduce__(self):
|
330
|
0
|
return (self.__class__, ())
|
331
|
|
|
332
|
|
|
333
|
9
|
def picknthweekday(year, month, dayofweek, hour, minute, whichweek):
|
334
|
|
""" dayofweek == 0 means Sunday, whichweek 5 means last instance """
|
335
|
9
|
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
|
9
|
weekdayone = first.replace(day=((dayofweek - first.isoweekday()) % 7) + 1)
|
340
|
9
|
wd = weekdayone + ((whichweek - 1) * ONEWEEK)
|
341
|
9
|
if (wd.month != month):
|
342
|
0
|
wd -= ONEWEEK
|
343
|
|
|
344
|
9
|
return wd
|
345
|
|
|
346
|
|
|
347
|
9
|
def valuestodict(key):
|
348
|
|
"""Convert a registry key's values to a dictionary."""
|
349
|
9
|
dout = {}
|
350
|
9
|
size = winreg.QueryInfoKey(key)[1]
|
351
|
9
|
tz_res = None
|
352
|
|
|
353
|
9
|
for i in range(size):
|
354
|
9
|
key_name, value, dtype = winreg.EnumValue(key, i)
|
355
|
9
|
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
|
9
|
if value & (1 << 31):
|
359
|
7
|
value = value - (1 << 32)
|
360
|
9
|
elif dtype == winreg.REG_SZ:
|
361
|
|
# If it's a reference to the tzres DLL, load the actual string
|
362
|
9
|
if value.startswith('@tzres'):
|
363
|
9
|
tz_res = tz_res or tzres()
|
364
|
9
|
value = tz_res.name_from_string(value)
|
365
|
|
|
366
|
9
|
value = value.rstrip('\x00') # Remove trailing nulls
|
367
|
|
|
368
|
9
|
dout[key_name] = value
|
369
|
|
|
370
|
9
|
return dout
|