feat: add exceptions handling

pull/5/head
Jean-Baptiste VESLIN 5 years ago committed by GitHub
parent e1c2bf0c8d
commit d5d6f7b3cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -60,8 +60,14 @@ def connect(callback=None):
ssh.connect(INPUT_HOST, port=INPUT_PORT, username=INPUT_USER, ssh.connect(INPUT_HOST, port=INPUT_PORT, username=INPUT_USER,
pkey=p_key, password=INPUT_PASS, pkey=p_key, password=INPUT_PASS,
timeout=convert_to_seconds(INPUT_CONNECT_TIMEOUT)) timeout=convert_to_seconds(INPUT_CONNECT_TIMEOUT))
except:
print("Connect error")
raise
else:
if callback: if callback:
callback(ssh) callback(ssh)
finally: finally:
os.unlink(tmp.name) os.unlink(tmp.name)
tmp.close() tmp.close()
@ -98,10 +104,8 @@ def ssh_process(ssh, input_ssh):
err = "".join(stderr.readlines()) err = "".join(stderr.readlines())
err = err.strip() if err is not None else None err = err.strip() if err is not None else None
if err: if err:
if out is None:
raise Exception(err)
else:
print(f"Error: \n{err}") print(f"Error: \n{err}")
raise Exception(err)
pass pass
@ -128,10 +132,19 @@ def scp_process(ssh, input_scp):
with scp.SCPClient(ssh.get_transport(), progress=progress, sanitize=lambda x: x) as conn: with scp.SCPClient(ssh.get_transport(), progress=progress, sanitize=lambda x: x) as conn:
for l2r in copy_list: for l2r in copy_list:
remote = l2r.get('r') remote = l2r.get('r')
ssh.exec_command(f"mkdir -p {remote} || true") try:
ssh.exec_command(f"mkdir -p {remote}")
except:
print(f"Remote mkdir error. Can't create {remote}")
raise
for f in [f for f in glob(l2r.get('l'))]: for f in [f for f in glob(l2r.get('l'))]:
try:
conn.put(f, remote_path=remote, recursive=True) conn.put(f, remote_path=remote, recursive=True)
print(f"{f} -> {remote}") print(f"{f} -> {remote}")
except:
print(f"scp error. Can't copy {f} on {remote}")
raise
pass pass

Loading…
Cancel
Save