example script usability improvements

Author: SeaBlooms

examples/09_custom_file_transfer_example.py

@@ -20,7 +20,11 @@
 import sys
 
 # Add the parent directory to the path so we can import the jupiterone client
-sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+try:
+    sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+except NameError:
+    # Handle case when __file__ is not available (e.g., when exec'd)
+    sys.path.append('..')
 
 from jupiterone.client import JupiterOneClient
 

examples/J1QLdeferredResponse.py

@@ -13,8 +13,8 @@
 # JupiterOne GraphQL API headers
 j1_graphql_headers = {
         'Content-Type': 'application/json',
-        'Authorization': 'Bearer ' + token,
-        'Jupiterone-Account': acct
+        'Authorization': 'Bearer ' + (token or ''),
+        'Jupiterone-Account': acct or ''
 }
 
 gql_query = """

examples/bulk_upload.py

@@ -5,6 +5,12 @@
 token = os.environ.get("JUPITERONE_TOKEN")
 url = "https://graphql.us.jupiterone.io"
 
+# Check if credentials are available
+if not account or not token:
+    print("Error: JUPITERONE_ACCOUNT and JUPITERONE_TOKEN environment variables must be set")
+    print("This example script requires valid JupiterOne credentials to run")
+    exit(1)
+
 j1 = JupiterOneClient(account=account, token=token, url=url)
 
 instance_id = "<Integration Instance ID>"

examples/examples.py

@@ -8,6 +8,12 @@
 token = os.environ.get("JUPITERONE_TOKEN")
 url = "https://graphql.us.jupiterone.io"
 
+# Check if credentials are available
+if not account or not token:
+    print("Error: JUPITERONE_ACCOUNT and JUPITERONE_TOKEN environment variables must be set")
+    print("This example script requires valid JupiterOne credentials to run")
+    exit(1)
+
 j1 = JupiterOneClient(account=account, token=token, url=url)
 
 # query_v1