aboutsummaryrefslogtreecommitdiff
path: root/updatezinfo.py
blob: ade4d05f4f21aa7e2a9cd428abc5dec92002ba7e (plain)
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
#!/usr/bin/env python
import os
import hashlib
import json
import io

from six.moves.urllib import request

from dateutil.zoneinfo import rebuild

METADATA_FILE = "zonefile_metadata.json"

def main():
    with io.open(METADATA_FILE,'r') as f:
        metadata = json.load(f)

    if not os.path.isfile(metadata['tzdata_file']):
        print("Downloading tz file from iana")
        request.urlretrieve(os.path.join(metadata['releases_url'],metadata['tzdata_file']), metadata['tzdata_file'])
    with open(metadata['tzdata_file'],'rb') as tzfile:
        sha_hasher = hashlib.sha512()
        sha_hasher.update(tzfile.read())
        sha_512_file = sha_hasher.hexdigest()
        assert metadata['tzdata_file_sha512'] == sha_512_file, "SHA failed for downloaded tz file"
    print("Updating timezone information...")
    rebuild(metadata['tzdata_file'], zonegroups=metadata['zonegroups'])
    print("Done.")

if __name__ == "__main__":
    main()