From b5abb8b1e1b053c5d3ddd391469edc3ed02aa22c Mon Sep 17 00:00:00 2001
From: rmspeers <ryan@riverloopsecurity.com>
Date: Wed, 28 Nov 2018 21:11:03 -0500
Subject: Updates for Python3 headless install.

---
 scripts/install_api.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'scripts/install_api.py')

diff --git a/scripts/install_api.py b/scripts/install_api.py
index 2eca2292..1237e1e5 100755
--- a/scripts/install_api.py
+++ b/scripts/install_api.py
@@ -7,6 +7,11 @@ import sys
 import os
 from site import getsitepackages, getusersitepackages, check_enableusersite
 
+# Hacky command line parsing to accept a silent-install -s flag like linux-setup.sh:
+INTERACTIVE = True
+if '-s' in sys.argv[1:]:
+    INTERACTIVE = False
+
 try:
     import binaryninja
     print("Binary Ninja API already Installed")
@@ -51,9 +56,12 @@ def validate_path(path):
 
 
 while not validate_path(api_path):
-    print("\nBinary Ninja not found. Please provide the path to Binary " + \
-          "Ninja's install directory: \n [{}] : ".format(api_path))
+    print("\nBinary Ninja not found.")
+    if not INTERACTIVE:
+        print("Non-interactive mode selected, failing.")
+        sys.exit(-1)
 
+    print("Please provide the path to Binary Ninja's install directory: \n [{}] : ".format(api_path))
     new_path = sys.stdin.readline().strip()
     if len(new_path) == 0:
         print("\nInvalid path")
-- 
cgit v1.3.1