improve packaging of data files

This commit stops setuptools from presenting a depracetion warning.
This commit ensures .mo files will not be included in source distributions.

1 files changed, 33 insertions, 4 deletions

diff --git a/setup.py b/setup.py
index 345febc..9bc0a62 100755
--- a/setup.py
+++ b/setup.py
@@ -8,13 +8,42 @@
 import setuptools
 
 from setuptools.command.build_py import build_py
+from setuptools.command.sdist import sdist
+
+from pathlib import Path
+
+here = Path(__file__).resolve().parent
 
 class CustomBuildCommand(build_py):
-    '''
-    The build command but runs babel before build.
-    '''
+    """The build command but runs babel before build."""
     def run(self, *args, **kwargs):
+        """Wrapper around build_py's original run() method."""
         self.run_command('compile_catalog')
+
+        super().run(*args, **kwargs)
+
+class CustomSdistCommand(sdist):
+    """
+    The sdist command but prevents compiled message catalogs from being included
+    in the archive.
+    """
+    def run(self, *args, **kwargs):
+        """Wrapper around sdist's original run() method."""
+        locales_dir = here / 'src/hydrilla/builder/locales'
+        locale_files = {}
+
+        for path in locales_dir.rglob('*.mo'):
+            locale_files[path] = path.read_bytes()
+
+        for path in locale_files:
+            path.unlink()
+
         super().run(*args, **kwargs)
 
-setuptools.setup(cmdclass={'build_py': CustomBuildCommand})
+        for path, contents in locale_files.items():
+            path.write_bytes(contents)
+
+setuptools.setup(cmdclass = {
+    'build_py': CustomBuildCommand,
+    'sdist': CustomSdistCommand
+})