def scp_put_file(self, source_file=None, dest_file=None):
"""Put file using SCP."""
try:
scp = SCPConn(ssh_conn=self)
scp.scp_put_file(source_file=source_file, dest_file=dest_file)
except Exception:
raise
finally:
scp.close()
def sendFileToMachine(host: machine.Machine, source_file, dest_file):
net_device = {
'device_type': 'linux',
'ip': host.ip,
'username': host.username,
'password': host.password,
}
try:
ssh_conn = ConnectHandler(**net_device)
scp_conn = SCPConn(ssh_conn)
putFileState = scp_conn.scp_put_file(source_file, dest_file)
scp_conn.close()
ssh_conn.disconnect()
host.status = 2 # Passed
return True # succeeded sending the file to machine
except:
print(f"Failed connect to {host.ip}")
host.status = 0 # Failed
return False # Filed sending the file to machine
def scp_file(self, source_file, dest_file):
try:
scp_transfer = SCPConn(self.device)
scp_transfer.scp_put_file(source_file, dest_file)
except:
raise ConnectionException("SCP transfer to remote device failed")
