diff --git a/connect.py b/connect.py index 84a02048a..79b266e29 100755 --- a/connect.py +++ b/connect.py @@ -33,7 +33,10 @@ from jperm.ansible_api import MyRunner from jlog.models import ExecLog, FileLog login_user = get_object(User, username=getpass.getuser()) -remote_ip = os.environ.get('SSH_CLIENT').split()[0] +try: + remote_ip = os.environ.get('SSH_CLIENT').split()[0] +except (IndexError, AttributeError): + remote_ip = os.popen("who -m | awk '{ print $NF }'").read().strip('()\n') try: import termios