gen_tests.py

#!/usr/bin/env python
# -*- coding: ascii -*-
'''
$Id: gen_tests.py,v 1.15 2005/01/07 04:51:30 zenzen Exp $
'''

import os
import os.path
import subprocess
import sys
from gen_tzinfo import allzones
import gen_tzinfo

zdump = os.path.abspath(os.path.join(
    os.path.dirname(__file__), 'build', 'bin', 'zdump'
))


def main():
    dest_dir = os.path.abspath(os.path.join(os.path.dirname(__file__)))

    datf = open(os.path.join(dest_dir, 'zdump.out'), 'w')

    for zone in allzones():
        print('Collecting zdump(1) output for %s in zdump.out' % (zone,))
        # We don't yet support v2 format tzfile(5) files, so limit
        # the daterange we test against - zdump understands v2 format
        # files and will output historical records we can't cope with
        # otherwise.
        command = [zdump, '-V', '-c', '1902,2038', zone]
        zd_out = subprocess.check_output(command, encoding="utf8")
        lines = [line.strip() for line in zd_out.splitlines()]

        for line in lines:
            print(line, file=datf)
    datf.flush()
    datf.close()

if __name__ == '__main__':
    try:
        gen_tzinfo.target = sys.argv[1:]
    except IndexError:
        gen_tzinfo.target = None
    main()

# vim: set filetype=python ts=4 sw=4 et