Tomi Pieviläinen | 8c6026b | 2013-11-01 10:37:15 +0200 | [diff] [blame] | 1 | #!/usr/bin/env python |
niemeyer | 672e2eb | 2005-02-23 00:04:12 +0000 | [diff] [blame] | 2 | import os |
Yaron de Leeuw | d722aad | 2014-11-20 22:36:52 +0200 | [diff] [blame] | 3 | import hashlib |
Yaron de Leeuw | 6111892 | 2014-11-24 20:33:36 +0200 | [diff] [blame] | 4 | import json |
| 5 | import io |
Yaron de Leeuw | d722aad | 2014-11-20 22:36:52 +0200 | [diff] [blame] | 6 | |
| 7 | from six.moves.urllib import request |
Paul Ganssle | 9271514 | 2016-02-20 10:22:46 -0500 | [diff] [blame] | 8 | from six.moves.urllib import error as urllib_error |
niemeyer | 672e2eb | 2005-02-23 00:04:12 +0000 | [diff] [blame] | 9 | |
Tomi Pieviläinen | e551507 | 2012-03-28 18:04:02 +0300 | [diff] [blame] | 10 | from dateutil.zoneinfo import rebuild |
| 11 | |
Yaron de Leeuw | 6111892 | 2014-11-24 20:33:36 +0200 | [diff] [blame] | 12 | METADATA_FILE = "zonefile_metadata.json" |
niemeyer | 672e2eb | 2005-02-23 00:04:12 +0000 | [diff] [blame] | 13 | |
Yaron de Leeuw | 683c66e | 2014-11-29 23:17:59 +0200 | [diff] [blame] | 14 | |
Paul Ganssle | 6f9422a | 2018-07-02 10:45:01 -0400 | [diff] [blame] | 15 | def main(metadata_file): |
| 16 | with io.open(metadata_file, 'r') as f: |
Yaron de Leeuw | 6111892 | 2014-11-24 20:33:36 +0200 | [diff] [blame] | 17 | metadata = json.load(f) |
| 18 | |
Paul Ganssle | 9271514 | 2016-02-20 10:22:46 -0500 | [diff] [blame] | 19 | releases_urls = metadata['releases_url'] |
| 20 | if metadata['metadata_version'] < 2.0: |
| 21 | # In later versions the releases URL is a mirror URL |
| 22 | releases_urls = [releases_urls] |
| 23 | |
Yaron de Leeuw | 6111892 | 2014-11-24 20:33:36 +0200 | [diff] [blame] | 24 | if not os.path.isfile(metadata['tzdata_file']): |
Paul Ganssle | 9271514 | 2016-02-20 10:22:46 -0500 | [diff] [blame] | 25 | |
| 26 | for ii, releases_url in enumerate(releases_urls): |
| 27 | print("Downloading tz file from mirror {ii}".format(ii=ii)) |
| 28 | try: |
| 29 | request.urlretrieve(os.path.join(releases_url, |
| 30 | metadata['tzdata_file']), |
| 31 | metadata['tzdata_file']) |
| 32 | except urllib_error.URLError as e: |
| 33 | print("Download failed, trying next mirror.") |
| 34 | last_error = e |
| 35 | continue |
| 36 | |
| 37 | last_error = None |
| 38 | break |
| 39 | |
| 40 | if last_error is not None: |
| 41 | raise last_error |
| 42 | |
Yaron de Leeuw | 683c66e | 2014-11-29 23:17:59 +0200 | [diff] [blame] | 43 | with open(metadata['tzdata_file'], 'rb') as tzfile: |
Yaron de Leeuw | d722aad | 2014-11-20 22:36:52 +0200 | [diff] [blame] | 44 | sha_hasher = hashlib.sha512() |
| 45 | sha_hasher.update(tzfile.read()) |
| 46 | sha_512_file = sha_hasher.hexdigest() |
Yaron de Leeuw | 683c66e | 2014-11-29 23:17:59 +0200 | [diff] [blame] | 47 | assert metadata['tzdata_file_sha512'] == sha_512_file, "SHA failed for" |
Gustavo Niemeyer | ac37a80 | 2011-03-24 14:16:39 -0300 | [diff] [blame] | 48 | print("Updating timezone information...") |
Alex Verdyan | 186cff2 | 2015-07-13 14:30:09 +0300 | [diff] [blame] | 49 | rebuild.rebuild(metadata['tzdata_file'], zonegroups=metadata['zonegroups'], |
Yaron de Leeuw | a0e8518 | 2015-05-25 21:32:10 +0300 | [diff] [blame] | 50 | metadata=metadata) |
Gustavo Niemeyer | ac37a80 | 2011-03-24 14:16:39 -0300 | [diff] [blame] | 51 | print("Done.") |
niemeyer | 672e2eb | 2005-02-23 00:04:12 +0000 | [diff] [blame] | 52 | |
Jon Dufresne | 0e73f3d | 2017-04-17 11:30:13 -0700 | [diff] [blame] | 53 | |
niemeyer | 672e2eb | 2005-02-23 00:04:12 +0000 | [diff] [blame] | 54 | if __name__ == "__main__": |
Paul Ganssle | 6f9422a | 2018-07-02 10:45:01 -0400 | [diff] [blame] | 55 | import argparse |
| 56 | parser = argparse.ArgumentParser() |
| 57 | parser.add_argument('metadata', metavar='METADATA_FILE', |
| 58 | default=METADATA_FILE, |
| 59 | nargs='?') |
| 60 | |
| 61 | args = parser.parse_args() |
| 62 | main(args.metadata) |