Update how the python version is imported.

A pure Python version can be installed when only writing input files is reguired.

1 files changed, 13 insertions, 1 deletions

diff --git a/Python/setup.py b/Python/setup.py
index 71f82dbc7..8d4c5cd12 100644
--- a/Python/setup.py
+++ b/Python/setup.py
@@ -4,18 +4,30 @@
 setup.py file for WarpX
 """
 
+import sys
+import argparse
+
 from distutils.core import setup
 
+argparser = argparse.ArgumentParser(add_help=False)
+argparser.add_argument('--with-libwarpx', action='store_true', help='Install libwarpx. This option is only used by the makefile.')
+args, unknown = argparser.parse_known_args()
+sys.argv = [sys.argv[0]] + unknown
+
 try:
     from distutils.command.build_py import build_py_2to3 as build_py
 except ImportError:
     from distutils.command.build_py import build_py
 
+if args.with_libwarpx:
+    package_data = {'pywarpx' : ['libwarpx.so']}
+else:
+    package_data = {}
 
 setup (name = 'pywarpx',
        packages = ['pywarpx'],
        package_dir = {'pywarpx':'pywarpx'},
        description = """Wrapper of WarpX""",
-       package_data = {'pywarpx' : ['libwarpx.so']},
+       package_data = package_data,
        cmdclass={'build_py': build_py}
        )