Fix input
This commit is contained in:
parent
0b7d0cf0a2
commit
7cdbbf4bdc
|
@ -59,13 +59,13 @@ def agent_auth(transport, username):
|
||||||
|
|
||||||
def manual_auth(username, hostname):
|
def manual_auth(username, hostname):
|
||||||
default_auth = 'p'
|
default_auth = 'p'
|
||||||
auth = raw_input('Auth by (p)assword, (r)sa key, or (d)ss key? [%s] ' % default_auth)
|
auth = input('Auth by (p)assword, (r)sa key, or (d)ss key? [%s] ' % default_auth)
|
||||||
if len(auth) == 0:
|
if len(auth) == 0:
|
||||||
auth = default_auth
|
auth = default_auth
|
||||||
|
|
||||||
if auth == 'r':
|
if auth == 'r':
|
||||||
default_path = os.path.join(os.environ['HOME'], '.ssh', 'id_rsa')
|
default_path = os.path.join(os.environ['HOME'], '.ssh', 'id_rsa')
|
||||||
path = raw_input('RSA key [%s]: ' % default_path)
|
path = input('RSA key [%s]: ' % default_path)
|
||||||
if len(path) == 0:
|
if len(path) == 0:
|
||||||
path = default_path
|
path = default_path
|
||||||
try:
|
try:
|
||||||
|
@ -76,7 +76,7 @@ def manual_auth(username, hostname):
|
||||||
t.auth_publickey(username, key)
|
t.auth_publickey(username, key)
|
||||||
elif auth == 'd':
|
elif auth == 'd':
|
||||||
default_path = os.path.join(os.environ['HOME'], '.ssh', 'id_dsa')
|
default_path = os.path.join(os.environ['HOME'], '.ssh', 'id_dsa')
|
||||||
path = raw_input('DSS key [%s]: ' % default_path)
|
path = input('DSS key [%s]: ' % default_path)
|
||||||
if len(path) == 0:
|
if len(path) == 0:
|
||||||
path = default_path
|
path = default_path
|
||||||
try:
|
try:
|
||||||
|
@ -99,7 +99,7 @@ if len(sys.argv) > 1:
|
||||||
if hostname.find('@') >= 0:
|
if hostname.find('@') >= 0:
|
||||||
username, hostname = hostname.split('@')
|
username, hostname = hostname.split('@')
|
||||||
else:
|
else:
|
||||||
hostname = raw_input('Hostname: ')
|
hostname = input('Hostname: ')
|
||||||
if len(hostname) == 0:
|
if len(hostname) == 0:
|
||||||
print('*** Hostname required.')
|
print('*** Hostname required.')
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
@ -112,8 +112,8 @@ if hostname.find(':') >= 0:
|
||||||
try:
|
try:
|
||||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
sock.connect((hostname, port))
|
sock.connect((hostname, port))
|
||||||
except Exception, e:
|
except Exception:
|
||||||
print '*** Connect failed: ' + str(e)
|
e = sys.exc_info()[1]
|
||||||
print('*** Connect failed: ' + str(e))
|
print('*** Connect failed: ' + str(e))
|
||||||
traceback.print_exc()
|
traceback.print_exc()
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
@ -150,7 +150,7 @@ try:
|
||||||
# get username
|
# get username
|
||||||
if username == '':
|
if username == '':
|
||||||
default_username = getpass.getuser()
|
default_username = getpass.getuser()
|
||||||
username = raw_input('Username [%s]: ' % default_username)
|
username = input('Username [%s]: ' % default_username)
|
||||||
if len(username) == 0:
|
if len(username) == 0:
|
||||||
username = default_username
|
username = default_username
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue