feat: add exceptions handling

This commit is contained in:
Jean-Baptiste VESLIN
2021-03-04 16:17:44 +01:00
committed by GitHub
parent e1c2bf0c8d
commit d5d6f7b3cb

27
app.py
View File

@@ -60,8 +60,14 @@ def connect(callback=None):
ssh.connect(INPUT_HOST, port=INPUT_PORT, username=INPUT_USER,
pkey=p_key, password=INPUT_PASS,
timeout=convert_to_seconds(INPUT_CONNECT_TIMEOUT))
except:
print("Connect error")
raise
else:
if callback:
callback(ssh)
finally:
os.unlink(tmp.name)
tmp.close()
@@ -98,10 +104,8 @@ def ssh_process(ssh, input_ssh):
err = "".join(stderr.readlines())
err = err.strip() if err is not None else None
if err:
if out is None:
raise Exception(err)
else:
print(f"Error: \n{err}")
print(f"Error: \n{err}")
raise Exception(err)
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:
for l2r in copy_list:
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'))]:
conn.put(f, remote_path=remote, recursive=True)
print(f"{f} -> {remote}")
try:
conn.put(f, remote_path=remote, recursive=True)
print(f"{f} -> {remote}")
except:
print(f"scp error. Can't copy {f} on {remote}")
raise
pass