env vars

main.py

@@ -1,7 +1,8 @@
 # External deps
-import paramiko
+import paramiko # for ssh
-import logging
+import logging # for possible debugging
-from time import sleep
+from time import sleep # for sleep()
+import os # for CLI arguments
 
 # Local deps
 import listServers, relativeUsage
@@ -10,10 +11,16 @@ import listServers, relativeUsage
 #logging.basicConfig()
 #logging.getLogger("paramiko").setLevel(logging.DEBUG)
 
+"""
+ENV VARIABLES AVAILABLE:
+ - hostname
+ - sshkeylocation (optional)
+ - sshuser (optional)
+"""
 # Creds
-hostname = '172.23.224.40' 
+hostname = str(os.environ['hostname'])
-sshuser   = 'Administrator'
+sshuser   = str(os.getenv('sshuser', 'Administrator'))
-sshkeylocation   = '/home/jc/.ssh/id_rsa'
+sshkeylocation   = str(os.path.abspath( os.getenv('sshkeypath', '~/.ssh/id_rsa') ))
 
 paramiko.transport.Transport._preferred_keys += ('ssh-dss',) # Allow the insecurities
 ssh = paramiko.SSHClient() # Alias