setup.py: fixes building

Author: kaelzhang

Makefile

@@ -34,7 +34,7 @@ build-pkg:
 
 build-ext:
 	@echo "\033[1m>> Building extension... <<\033[0m"
-	@STOCK_PANDAS_BUILDING=1 python setup.py build_ext --inplace
+	@STOCK_PANDAS_BUILDING_EXT=1 python setup.py build_ext --inplace
 
 build-doc:
 	sphinx-build -b html docs build_docs

setup.py

@@ -9,7 +9,7 @@
 
 import numpy as np
 
-BUILDING = os.environ.get('STOCK_PANDAS_BUILDING')
+BUILDING_EXT = os.environ.get('STOCK_PANDAS_BUILDING_EXT')
 
 ext_kwargs = dict(
     # Ignore warning caused by cpython for using deprecated apis
@@ -19,7 +19,7 @@
 
 # Distribution ref
 # https://cython.readthedocs.io/en/latest/src/userguide/source_files_and_compilation.html#distributing-cython-modules
-if BUILDING:
+if BUILDING_EXT:
     from Cython.Build import cythonize
     extensions = cythonize(
         Extension(
@@ -33,6 +33,8 @@
             'language_level': 3
         }
     )
+
+# When packaging the package, use the pre-compiled c++ file
 else:
     extensions = [
         Extension(
@@ -42,13 +44,15 @@
         )
     ]
 
-setup(
-    ext_modules=extensions,
-    packages=[
-        'stock_pandas',
-        'stock_pandas.commands',
-        'stock_pandas.directive',
-        'stock_pandas.math',
-        'stock_pandas.meta'
-    ]
-)
+# Only run setup when building the package
+if __name__ == '__main__':
+    setup(
+        ext_modules=extensions,
+        packages=[
+            'stock_pandas',
+            'stock_pandas.commands',
+            'stock_pandas.directive',
+            'stock_pandas.math',
+            'stock_pandas.meta'
+        ]
+    )