Repo created

tools/python/data/borders/__init__.py

@@ -0,0 +1,44 @@
+import logging
+import os
+import tarfile
+
+from six import BytesIO
+
+from data_files import find_data_files
+
+try:
+    import lzma
+except ImportError:
+    from backports import lzma
+
+logger = logging.getLogger(__name__)
+
+
+def init(borders_path=None):
+    data_path = find_data_files("omim-data")
+
+    if data_path is None:
+        logger.error("omim-data was not found.")
+        return False
+
+    if borders_path is None:
+        borders_path = os.path.join(data_path, "borders")
+
+    if not os.path.exists(borders_path):
+        tar_lzma_path = os.path.join(data_path, "borders.tar.xz")
+        lzma_stream = BytesIO()
+        with open(tar_lzma_path, mode="rb") as f:
+            decompressed = lzma.decompress(f.read())
+            lzma_stream.write(decompressed)
+
+        lzma_stream.seek(0)
+        try:
+            with tarfile.open(fileobj=lzma_stream, mode="r") as tar:
+                tar.extractall(borders_path)
+        except PermissionError as e:
+            logger.error(str(e))
+            return False
+
+        logger.info("{} was created.".format(borders_path))
+
+    return True

tools/python/data/borders/setup.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python3
+import os
+import sys
+import tarfile
+from distutils import log
+from distutils.command.build import build
+from distutils.command.clean import clean
+
+from six import BytesIO
+
+try:
+    import lzma
+except ImportError:
+    from backports import lzma
+
+
+module_dir = os.path.abspath(os.path.dirname(__file__))
+sys.path.insert(0, os.path.join(module_dir, "..", ".."))
+
+from data.base import DATA_PATH
+from data.base import chdir
+from data.base import get_version
+from data.base import setup
+
+
+TAR_LZMA_PATH = os.path.join(DATA_PATH, "borders.tar.xz")
+
+
+class BuildCmd(build, object):
+    def run(self):
+        log.info("Creating {}".format(TAR_LZMA_PATH))
+        tar_stream = BytesIO()
+        borders_path = os.path.join(DATA_PATH, "borders")
+        with chdir(borders_path):
+            with tarfile.open(fileobj=tar_stream, mode="w") as tar:
+                for f in os.listdir(borders_path):
+                    tar.add(f)
+
+        tar_stream.seek(0)
+        with lzma.open(TAR_LZMA_PATH, mode="w") as f:
+            f.write(tar_stream.read())
+
+        super(BuildCmd, self).run()
+
+
+class CleanCmd(clean, object):
+    def run(self):
+        if os.path.exists(TAR_LZMA_PATH):
+            log.info("Removing {}".format(TAR_LZMA_PATH))
+            os.remove(TAR_LZMA_PATH)
+
+        super(CleanCmd, self).run()
+
+
+setup(
+    __file__,
+    "borders",
+    ["borders.tar.xz", "packed_polygons.bin"],
+    package_dir={"borders": ""},
+    packages=["borders"],
+    cmdclass={"build": BuildCmd, "clean": CleanCmd},
+    install_requires=["omim-data-files=={}".format(get_version())]
+)