From: Zac Medico <zmedico@gentoo.org>
Date: Mon, 25 Jun 2007 03:22:42 +0000 (-0000)
Subject: For bug #175891, add sftp protocol support via the paramiko library.
X-Git-Tag: v2.1.3~145
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=62fbee1b50d3f63756f7832fff275722b2466364;p=portage.git

For bug #175891, add sftp protocol support via the paramiko library.

svn path=/main/branches/2.1.2/; revision=7022
---

diff --git a/pym/getbinpkg.py b/pym/getbinpkg.py
index c3bd0b5ee..8bd43f29c 100644
--- a/pym/getbinpkg.py
+++ b/pym/getbinpkg.py
@@ -141,6 +141,15 @@ def create_conn(baseurl,conn=None):
 				conn.login(username)
 			conn.set_pasv(passive)
 			conn.set_debuglevel(0)
+		elif protocol == "sftp":
+			try:
+				import paramiko
+			except ImportError:
+				raise NotImplementedError(
+					"paramiko must be installed for sftp support")
+			t = paramiko.Transport(host)
+			t.connect(username=username, password=password)
+			conn = paramiko.SFTPClient.from_transport(t)
 		else:
 			raise NotImplementedError, "%s is not a supported protocol." % protocol
 
@@ -307,6 +316,8 @@ def dir_get_list(baseurl,conn=None):
 			del olddir
 		else:
 			listing = conn.nlst(address)
+	elif protocol == "sftp":
+		listing = conn.listdir(address)
 	else:
 		raise TypeError, "Unknown protocol. '%s'" % protocol
 
@@ -332,6 +343,13 @@ def file_get_metadata(baseurl,conn=None, chunk_size=3000):
 		data,rc,msg = make_http_request(conn, address, params, headers)
 	elif protocol in ["ftp"]:
 		data,rc,msg = make_ftp_request(conn, address, -chunk_size)
+	elif protocol == "sftp":
+		f = conn.open(address)
+		try:
+			f.seek(-chunk_size, 2)
+			data = f.read()
+		finally:
+			f.close()
 	else:
 		raise TypeError, "Unknown protocol. '%s'" % protocol
 	
@@ -400,6 +418,8 @@ def file_get_lib(baseurl,dest,conn=None):
 		data,rc,msg = make_http_request(conn, address, params, headers, dest=dest)
 	elif protocol in ["ftp"]:
 		data,rc,msg = make_ftp_request(conn, address, dest=dest)
+	elif protocol == "sftp":
+		conn.get(address, dest)
 	else:
 		raise TypeError, "Unknown protocol. '%s'" % protocol