1 files changed, 164 insertions, 0 deletions
diff --git a/pytzdata/commands/make.py b/pytzdata/commands/make.py
new file mode 100644
index 0000000..e0a7b4b
--- /dev/null
+++ b/pytzdata/commands/make.py
@@ -0,0 +1,164 @@
+# -*- coding: utf-8 -*-
+
+import os
+import errno
+import shutil
+import tarfile
+import subprocess
+
+from urllib.request import urlopen
+from contextlib import closing
+from distutils.dir_util import copy_tree, remove_tree
+from cleo import Command
+
+
+class MakeCommand(Command):
+    """
+    Make tzdata files.
+
+    make
+        {version? : The version to make.}
+        {--path= : The destination directory.}
+    """
+
+    FILES = ["tzcode-latest.tar.gz", "tzdata-latest.tar.gz"]
+
+    BASE_URL = "http://www.iana.org/time-zones/repository"
+
+    def __init__(self):
+        super(MakeCommand, self).__init__()
+
+        self.path = None
+
+    def handle(self):
+        self.path = self.option("path") or self.get_build_dir()
+
+        self.mkdir(self.path)
+
+        self.download()
+        self.line("")
+        self.uncompress()
+        self.line("")
+        self.build()
+        self.line("")
+        self.copy()
+        self.line("")
+        self.clean()
+        self.line("")
+        self.call("zones:dump")
+
+    def download(self):
+        if not self.argument("version"):
+            files = self.FILES
+            base_url = self.BASE_URL
+        else:
+            files = [
+                "tzcode{}.tar.gz".format(self.argument("version")),
+                "tzdata{}.tar.gz".format(self.argument("version")),
+            ]
+            base_url = "https://data.iana.org/time-zones/releases"
+
+        self.line("[<comment>Downloading archives</>]")
+        for filename in files:
+            url = os.path.join(base_url, filename)
+            self.write("<comment>Downloading</> <fg=cyan>{}</>".format(filename))
+            dest = os.path.join(self.path, filename)
+            with closing(urlopen(url)) as r:
+                with open(dest, "wb") as f:
+                    shutil.copyfileobj(r, f)
+
+            self.overwrite("<info>Downloaded</> <fg=cyan>{}</> ".format(filename))
+            self.line("")
+
+    def uncompress(self):
+        self.line("[<comment>Uncompressing archives</>]")
+        dest_path = os.path.join(self.path, "tz")
+        self.mkdir(dest_path)
+
+        if not self.argument("version"):
+            files = self.FILES
+        else:
+            files = [
+                "tzcode{}.tar.gz".format(self.argument("version")),
+                "tzdata{}.tar.gz".format(self.argument("version")),
+            ]
+
+        for filename in files:
+            filepath = os.path.join(self.path, filename)
+            self.write("<comment>Uncompressing</> <fg=cyan>{}</>".format(filename))
+            with closing(tarfile.open(filepath)) as f:
+                f.extractall(dest_path)
+
+            self.overwrite("<info>Uncompressed</> <fg=cyan>{}</> ".format(filename))
+            self.line("")
+
+    def build(self):
+        self.line("[<comment>Building tzdata</>]")
+        dest_path = os.path.join(self.path, "tz")
+
+        # Getting VERSION
+        with open(os.path.join(dest_path, "version")) as f:
+            version = f.read().strip()
+
+        self.write("<comment>Building</> version <fg=cyan>{}</>".format(version))
+        os.chdir(dest_path)
+
+        with open(os.devnull, "w") as temp:
+            subprocess.call(
+                ["make", "TOPDIR={}".format(dest_path), "install"],
+                stdout=temp,
+                stderr=temp,
+            )
+
+        self.overwrite("<info>Built</> version <fg=cyan>{}</>".format(version))
+        self.line("")
+
+    def copy(self):
+        self.line("[<comment>Copying tzdata</>]")
+        tzdata_dir = os.path.realpath(
+            os.path.join(self.path, "tz", "usr", "share", "zoneinfo")
+        )
+        local_dir = os.path.realpath(
+            os.path.join(os.path.dirname(__file__), "..", "zoneinfo")
+        )
+        self.line(
+            "<comment>Copying <fg=cyan>{}</> to <fg=cyan>{}</></>".format(
+                tzdata_dir, local_dir
+            )
+        )
+        remove_tree(local_dir)
+        copy_tree(tzdata_dir, local_dir)
+
+    def clean(self):
+        self.line("[<comment>Cleaning up</>]")
+        if not self.argument("version"):
+            files = self.FILES
+        else:
+            files = [
+                "tzcode{}.tar.gz".format(self.argument("version")),
+                "tzdata{}.tar.gz".format(self.argument("version")),
+            ]
+
+        for filename in files:
+            filepath = os.path.join(self.path, filename)
+            self.write("<comment>Removing</> <fg=cyan>{}</>".format(filename))
+            os.remove(filepath)
+
+            self.overwrite("<info>Removed</> <fg=cyan>{}</> ".format(filename))
+            self.line("")
+
+        self.write("<comment>Removing</> <fg=cyan>tz/*</>")
+        shutil.rmtree(os.path.join(self.path, "tz"))
+        self.overwrite("<info>Removed</> <fg=cyan>tz/*</>")
+
+    def get_build_dir(self):
+        return os.path.join(os.path.dirname(__file__), "..", "..", "_build")
+
+    def mkdir(self, path, mode=0o777):
+        try:
+            os.makedirs(path, mode)
+        except OSError as exc:
+            if exc.errno == errno.EEXIST and os.path.isdir(path):
+                pass
+            else:
+                raise