define entry point for the plasma store

apache · Aug 18, 2017 · 4c05140 · 4c05140
1 parent 4ef7c89
commit 4c05140
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/python/pyarrow/__init__.py b/python/pyarrow/__init__.py
@@ -105,6 +105,16 @@
 
 localfs = LocalFileSystem.get_instance()
 
+# Entry point for starting the plasma store
+
+def start_plasma_store():
+    import os
+    import pyarrow
+    import subprocess
+    import sys
+    plasma_store_executable = os.path.join(pyarrow.__path__[0], "plasma_store")
+    process = subprocess.Popen([plasma_store_executable] + sys.argv[1:])
+    process.wait()
 
 # ----------------------------------------------------------------------
 # 0.4.0 deprecations

diff --git a/python/setup.py b/python/setup.py
@@ -382,6 +382,11 @@ def has_ext_modules(foo):
         'clean': clean,
         'build_ext': build_ext
     },
+    entry_points = {
+        'console_scripts': [
+            'plasma_store = pyarrow:start_plasma_store'
+        ]
+    },
     use_scm_version={"root": "..", "relative_to": __file__},
     setup_requires=['setuptools_scm', 'cython >= 0.23'],
     install_requires=['numpy >= 1.10', 'six >= 1.0.0'],