add PyInstaller hooks

.gitignore

@@ -6,6 +6,7 @@ venv/
 flintrock-logo/
 .hypothesis/
 *.prf
+.DS_Store
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

.travis.yml

@@ -4,6 +4,13 @@ python:
   - "3.5"
 install:
   - "pip install -r requirements/developer.pip"
+  - "curl -L -o pandoc.deb https://github.com/jgm/pandoc/releases/download/1.16.0.2/pandoc-1.16.0.2-1-amd64.deb && sudo dpkg -i pandoc.deb"
 script:
   - "py.test ./tests/test_static.py"
   - "py.test ./tests/test_flintrock.py"
+  - "py.test ./tests/test_pyinstaller_packaging.py"
+addons:
+  artifacts:
+    paths:
+      - "dist/*.zip"
+  s3_region: "us-east-1"

flintrock/core.py

@@ -5,13 +5,20 @@
 import os
 import posixpath
 import shlex
+import sys
 import time
 
 # Flintrock modules
 from .exceptions import SSHError, NodeError
 from .ssh import get_ssh_client, ssh_check_output, ssh
 
-THIS_DIR = os.path.dirname(os.path.realpath(__file__))
+FROZEN = getattr(sys, 'frozen', False)
+
+if FROZEN:
+    THIS_DIR = sys._MEIPASS
+else:
+    THIS_DIR = os.path.dirname(os.path.realpath(__file__))
+
 SCRIPTS_DIR = os.path.join(THIS_DIR, 'scripts')
 
 

flintrock/flintrock.py

@@ -21,7 +21,12 @@
 from flintrock import __version__
 from .services import HDFS, Spark  # TODO: Remove this dependency.
 
-THIS_DIR = os.path.dirname(os.path.realpath(__file__))
+FROZEN = getattr(sys, 'frozen', False)
+
+if FROZEN:
+    THIS_DIR = sys._MEIPASS
+else:
+    THIS_DIR = os.path.dirname(os.path.realpath(__file__))
 
 
 def format_message(*, message: str, indent: int=4, wrap: int=70):

flintrock/services.py

@@ -12,7 +12,13 @@
 from .core import FlintrockCluster
 from .ssh import ssh_check_output
 
-THIS_DIR = os.path.dirname(os.path.realpath(__file__))
+FROZEN = getattr(sys, 'frozen', False)
+
+if FROZEN:
+    THIS_DIR = sys._MEIPASS
+else:
+    THIS_DIR = os.path.dirname(os.path.realpath(__file__))
+
 SCRIPTS_DIR = os.path.join(THIS_DIR, 'scripts')
 
 

generate-standalone-package.py

@@ -0,0 +1,30 @@
+import os
+import platform
+import shutil
+import subprocess
+
+from flintrock import __version__ as flintrock_version
+
+THIS_DIR = os.path.dirname(os.path.realpath(__file__))
+
+if __name__ == '__main__':
+    operating_system = platform.system()
+    machine_type = platform.machine()
+
+    subprocess.run([
+            'pyinstaller',
+            '--noconfirm',
+            '--name', 'flintrock',
+            '--additional-hooks-dir', '.',
+            'standalone.py'],
+        check=True)
+
+    shutil.make_archive(
+        base_name=os.path.join(
+            THIS_DIR, 'dist',
+            'flintrock-{v}-{os}-{m}'.format(
+                v=flintrock_version,
+                os=operating_system,
+                m=machine_type)),
+        format='zip',
+        root_dir=os.path.join(THIS_DIR, 'dist', 'flintrock'))

hook-flintrock.py

@@ -0,0 +1,5 @@
+datas=[
+    ('flintrock/scripts', './scripts'),
+    ('flintrock/templates', './templates'),
+    ('flintrock/config.yaml.template', './'),
+]

requirements/maintainer.pip

@@ -2,3 +2,4 @@
 wheel >= 0.26.0
 twine >= 1.6.4
 pypandoc >= 1.1.2
+PyInstaller >= 3.1

requirements/user.pip

@@ -1,7 +1,10 @@
 # NOTE: Run pip from Flintrock's root directory, not from the directory containing
 #       this file.
+
+# Due to: https://github.com/pyinstaller/pyinstaller/issues/1781
+setuptools >= 19.0, <= 19.2
+
 # NOTE: The `-e .` syntax lets us reuse the requirements already specified under
 #       `install_requires` in setup.py.
 #       See: https://caremad.io/2013/07/setup-vs-requirement/
-setuptools >= 18.8.1
 -e .

standalone.py

@@ -0,0 +1,11 @@
+"""
+A standalone script for use by PyInstaller.
+
+Users should not be running this script.
+"""
+
+import sys
+from flintrock.flintrock import main
+
+if __name__ == '__main__':
+    sys.exit(main())

tests/test_pyinstaller_packaging.py

@@ -0,0 +1,20 @@
+import glob
+import subprocess
+import sys
+
+# External modules
+import pytest
+
+
+@pytest.mark.skipif(sys.version_info < (3, 5), reason="Python 3.5+ is required")
+def test_pyinstaller_packaging():
+    subprocess.run(
+        ['pip', 'install', '-r', 'requirements/maintainer.pip'],
+        check=True)
+    subprocess.run(
+        ['python', 'generate-standalone-package.py'],
+        check=True)
+    subprocess.run(
+        ['./dist/flintrock/flintrock'],
+        check=True)
+    assert glob.glob('./dist/*.zip')