From a2f05726379d1440f8709eb9b81913e1ca72f771 Mon Sep 17 00:00:00 2001 From: Samo Penic <samo.penic@gmail.com> Date: Sun, 06 May 2018 16:33:53 +0000 Subject: [PATCH] Added some extra field into wrappers --- trisurf/Remote.py | 44 +++++++++++++++++++++++++++----------------- 1 files changed, 27 insertions(+), 17 deletions(-) diff --git a/trisurf/Remote.py b/trisurf/Remote.py index 2bf80f1..977b96a 100644 --- a/trisurf/Remote.py +++ b/trisurf/Remote.py @@ -20,20 +20,23 @@ def connect(self, Timeout=5): if(not self.connected): -# try: - print("Trying to connect to: "+self.username+"@"+self.hostname+":"+str(self.port)+".") - self.ssh.connect(self.hostname, username=self.username, password=self.password, port=self.port, timeout=Timeout) - self.connected=True -# except: -# print("Error establishing connection with "+self.username+"@"+self.hostname+":"+str(self.port)+".") -# exit(1) + try: + print("Trying to connect to: "+self.username+"@"+self.hostname+":"+str(self.port)+".") + self.ssh.connect(self.hostname, username=self.username, password=self.password, port=self.port, timeout=Timeout) + self.connected=True + except: + print("Error establishing connection with "+self.username+"@"+self.hostname+":"+str(self.port)+".") + exit(1) else: print("Already connected!") return def disconnect(self): if(self.connected): - self.ssh.close() + try: + self.ssh.close() + except: + print("Cannot disconect. Unknown error.") else: print("Cannot disconect. Already disconnected.") self.connected=False @@ -52,16 +55,14 @@ print("Cannot execute remote commands. Connect first.") def send_file(self, local, remote): - pass -# sftp=self.ssh.open_sftp() -# sftp.put(local,remote) -# sftp.close() + sftp=self.ssh.open_sftp() + sftp.put(local,remote) + sftp.close() def receive_file(self,remote,local): - pass -# sftp=self.ssh.open_sftp() -# sftp.get(remote,local) -# sftp.close() + sftp=self.ssh.open_sftp() + sftp.get(remote,local) + sftp.close() def mkdir_remote(self,directory): sftp=self.ssh.open_sftp() @@ -88,4 +89,13 @@ return True def send_multiple_files_in_directory(self,local_files,directory): - pass + sftp=self.ssh.open_sftp() +# try: +# sftp.chdir(directory) # Test if remote_path exists +# except (IOError,FileNotFoundError): +# sftp.mkdir(directory) # Create remote_path +# sftp.chdir(directory) + self.mkdir_p(sftp, directory) + for f in set(local_files): + sftp.put(f, f) + sftp.close() -- Gitblit v1.9.3