From dcb7fa304c6e609d9b7d017590b8287516ad322e Mon Sep 17 00:00:00 2001 From: Wirawan Purwanto Date: Sat, 26 May 2012 08:37:01 -0400 Subject: [PATCH] * Added sftp_client.put() method. * Added sftp_status_callback class. --- net/sftp.py | 59 +++++++++++++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 57 insertions(+), 2 deletions(-) diff --git a/net/sftp.py b/net/sftp.py index af76255..0457576 100644 --- a/net/sftp.py +++ b/net/sftp.py @@ -15,6 +15,9 @@ only works for Linux OS since it uses /proc/$PID approach. import getpass import os +import stat +import sys +import time try: import paramiko @@ -43,7 +46,7 @@ if has_paramiko: #print "here1" conn.load_host_keys(os.path.expanduser("~/.ssh/known_hosts")) #print "here1" - conn.set_missing_host_key_policy(paramiko.RejectPolicy) # default=paranoid + conn.set_missing_host_key_policy(paramiko.RejectPolicy()) # default=paranoid #print "here1" conn.connect(hostname=hostname, username=username, password=password, port=port, @@ -60,11 +63,21 @@ if has_paramiko: def get(self, remotepath, localpath, callback=None): # TODO: progress bar self.sftp.get(remotepath, localpath, callback) - # Save the attributes as much as I can do it + # Preserve the attributes as much as I can do it stats = self.sftp.stat(remotepath) os.utime(localpath, (stats.st_atime, stats.st_mtime)) os.chmod(localpath, stats.st_mode) + def put(self, localpath, remotepath, callback=None): + # TODO: progress bar + self.sftp.put(localpath, remotepath, callback) + # Preserve the attributes as much as I can do it + stats = os.stat(localpath) + self.sftp.utime(remotepath, (stats.st_atime, stats.st_mtime)) + self.sftp.chmod(remotepath, stats.st_mode) + + def close(self): + self.sftp.close() # TODO: recode the sftp driver from @@ -74,3 +87,45 @@ class sftp_driver(object): def __init__(self, hostname, username=None, port=22, password=None): raise NotImplementedError # TODO + + +class sftp_status_callback(object): + """SFTP progress bar for sftp_client object above. + + Example usage: + + statproc = sftp_status_callback(fname=remote_root + "/myfile.txt") + sftp.get(remote_root + "/myfile.txt", "mylocalfile.txt", statproc) + statproc.done() + + """ + def __init__(self, fname, out=sys.stderr): + self.first = True + self.fname = fname + self.out = out + self.marks = [ 0, 10, 20, 30, 40, 50, 60, 70, 80, 90 ] + self.nbars = 25 + self.lastbar = 0 + self.begin_time = time.time() + def printbar(self, bar): + if bar > self.lastbar: + dbar = bar - self.lastbar + self.out.write("*" * dbar) + self.out.flush() + self.lastbar = bar + def __call__(self, nbytes, filesize): + if self.first: + self.first = False + self.filesize = filesize + self.last = 0 + self.out.write("%-40s : %12d " % (self.fname[-40:], self.filesize)) + self.out.flush() + self.cursize = nbytes + curbar = int(float(nbytes) / self.filesize * self.nbars) + self.printbar(curbar) + def done(self): + self.end_time = time.time() + dtime = self.end_time - self.begin_time + self.out.write(" %.2f kibps\n" % (self.filesize / 1024.0 / dtime)) + self.out.flush() +