Merge pull request #1 from galaxie500/TimeSeriesScaleMeanMaxVariance

Add compile.py

Author: tonylee2016

.gitignore

@@ -16,7 +16,7 @@ __pycache__/
 env/
 build/
 develop-eggs/
-dist/
+#dist/
 downloads/
 eggs/
 .eggs/

dist/tslearn-0.0.0-cp38-cp38-linux_x86_64.whl

@@ -0,0 +1,59 @@
+# coding: utf-8
+import os
+import fnmatch
+import sysconfig
+from setuptools import setup, find_packages
+from setuptools.command.build_py import build_py as _build_py
+from Cython.Build import cythonize
+
+EXCLUDE_FILES = [
+    'tslearn/main.py'
+]
+
+
+def get_ext_paths(root_dir, exclude_files):
+    """get filepaths for compilation"""
+    paths = []
+
+    for root, dirs, files in os.walk(root_dir):
+        for filename in files:
+            if os.path.splitext(filename)[1] != '.py':
+                continue
+
+            file_path = os.path.join(root, filename)
+            if file_path in exclude_files:
+                continue
+
+            paths.append(file_path)
+    return paths
+
+
+class build_py(_build_py):
+
+    def find_package_modules(self, package, package_dir):
+        ext_suffix = sysconfig.get_config_var('EXT_SUFFIX')
+        modules = super().find_package_modules(package, package_dir)
+        filtered_modules = []
+        for (pkg, mod, filepath) in modules:
+            if os.path.exists(filepath.replace('.py', ext_suffix)):
+                continue
+            filtered_modules.append((pkg, mod, filepath, ))
+        return filtered_modules
+
+
+setup(
+    name='tslearn',
+    version='0.0.0',
+    packages=find_packages(),
+    ext_modules=cythonize(
+        get_ext_paths('tslearn', EXCLUDE_FILES),
+        compiler_directives={'language_level': 3}
+    ),
+    cmdclass={
+        'build_py':build_py
+    }
+)
+
+# to compile working directory to wheel
+#$ python setup.py bdist_wheel
+#$ unzip tslearn-0.0.0-cp38-cp38-linux_x86_64.whl