1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
#!/usr/bin/env python
'''
$Id: gen_tzinfo.py,v 1.21 2005/02/15 20:21:38 zenzen Exp $
'''
import sys, os, os.path, shutil
from glob import glob
from datetime import datetime,timedelta,tzinfo
from pprint import pprint
from bisect import bisect_right
import re
sys.path.insert(0, 'src')
import pytz
zoneinfo = os.path.abspath(os.path.join(
os.path.dirname(__file__), 'build','etc','zoneinfo'
))
def allzones():
''' Return all available tzfile(5) files in the zoneinfo database '''
zones = []
for dirpath, dirnames, filenames in os.walk(zoneinfo):
zones.extend([
os.path.join(dirpath,f) for f in filenames
if not f.endswith('.tab')
])
stripnum = len(os.path.commonprefix(zones))
zones = [z[stripnum:] for z in zones]
if target:
wanted = target + ['US/Eastern', 'UTC']
zones = [z for z in zones if z in wanted]
# Does not cope with Riyadh87-89 - it appears this region went
# on solar time during this period and their DST offset changed
# minute to minute (the Olson database could only capture a precision
# of 5 seconds because of way too many zone changes, so the data isn't
# 100% accurate anyway).
# 'Factory' and 'localtime' appear to be Olson reference code specific
# and are skipped
zones = [z for z in zones if 'Riyadh8' not in z and z not in [
'Factory', 'localtime', 'posixrules']]
zones.sort()
return zones
def links():
'''Mapping of alias -> canonical name'''
l = {}
olson_src_files = glob('elsie.nci.nih.gov/src/*')
assert olson_src_files, 'No src files'
for filename in olson_src_files:
# Filenames containing a '.' are not data files.
if '.' in os.path.basename(filename):
continue
for line in open(filename):
if line.strip().startswith('#') or not line.strip():
continue
match = re.search(r'^\s*Link\s+([\w/\-]+)\s+([\w/\-]+)', line)
if match is not None:
new_name = match.group(1)
old_name = match.group(2)
l[old_name] = new_name
else:
assert not line.startswith('Link'), line
assert 'US/Pacific-New' in l, 'US/Pacific-New should be in links()'
return l
def dupe_src(destdir):
''' Copy ./src to our dest directory '''
if not os.path.isdir(destdir):
os.makedirs(destdir)
for f in glob(os.path.join('src','*')):
if not os.path.isdir(f):
shutil.copy(f, destdir)
destdir = os.path.join(destdir, 'pytz')
if not os.path.isdir(destdir):
os.makedirs(destdir)
for f in glob(os.path.join('src','pytz', '*')):
if not os.path.isdir(f):
shutil.copy(f, destdir)
destdir = os.path.join(destdir, 'tests')
if not os.path.isdir(destdir):
os.makedirs(destdir)
for f in glob(os.path.join('src', 'pytz', 'tests', '*')):
if not os.path.isdir(f):
shutil.copy(f, destdir)
def add_allzones(filename):
''' Append a list of all know timezones to the end of the file '''
outf = open(filename, 'a')
obsolete_zones = links().keys()
# Calculate 'common' timezones as best we can. We start with all
# timezones, strip out the legacy noise, and any name linked to
# a more canonical name (eg. Asia/Singapore is preferred to just
# Singapore)
cz = [z for z in allzones()
if z not in obsolete_zones
and '/' in z
and not z.startswith('SystemV/')
and not z.startswith('Etc/')]
# And extend our list manually with stuff we think deserves to be
# labelled 'common'.
cz.extend([
'UTC', 'GMT', 'US/Eastern', 'US/Pacific', 'US/Mountain',
'US/Central', 'US/Arizona', 'US/Hawaii', 'US/Alaska',
# Canadian timezones per Bug #506341
'Canada/Newfoundland', 'Canada/Atlantic', 'Canada/Eastern',
'Canada/Central', 'Canada/Mountain', 'Canada/Pacific'])
# And extend out list with all preferred country timezones.
zone_tab = open(os.path.join(zoneinfo, 'zone.tab'), 'r')
for line in zone_tab:
if line.startswith('#'):
continue
code, coordinates, zone = line.split(None, 4)[:3]
if zone not in cz:
cz.append(zone)
cz.sort()
print >> outf, 'all_timezones = \\'
pprint(sorted(allzones()), outf)
print >> outf, '''all_timezones = LazyList(
tz for tz in all_timezones if resource_exists(tz))
'''
print >> outf, 'all_timezones_set = LazySet(all_timezones)'
print >> outf, 'common_timezones = \\'
pprint(cz, outf)
print >> outf, '''common_timezones = LazyList(
tz for tz in common_timezones if tz in all_timezones)
'''
print >> outf, 'common_timezones_set = LazySet(common_timezones)'
outf.close()
def main(destdir):
_destdir = os.path.join(os.path.abspath(destdir), 'dist')
dupe_src(_destdir)
add_allzones(os.path.join(_destdir, 'pytz', '__init__.py'))
target = None
if __name__ == '__main__':
try:
target = sys.argv[1:]
except IndexError:
target = None
main('build')
|
