Paramiko SFTP - 避免指定完整的本地文件名？

Question

我有一些使用 Paramiko 从远程服务器获取构建文件的 Python 代码：

def setup_sftp_session(self, host='server.com', port=22, username='puppy'):
    self.transport = paramiko.Transport((host, port))
    privatekeyfile = os.path.expanduser('~/.ssh/id_dsa')
    try:
        ssh_key = paramiko.DSSKey.from_private_key_file(privatekeyfile)
    except IOError, e:
        self.logger.error('Unable to find SSH keyfile: %s' % str(e))
        sys.exit(1)
    try:
        self.transport.connect(username = username, pkey = ssh_key)
    except paramiko.AuthenticationException, e:
        self.logger.error("Unable to logon - are you sure you've added the pubkey to the server?: %s" % str(e))
        sys.exit(1)
    self.sftp = paramiko.SFTPClient.from_transport(self.transport)
    self.sftp.chdir('/some/location/buildfiles')

def get_file(self, remote_filename):
    try:
        self.sftp.get(remote_filename, 'I just want to save it in the local cwd')
    except IOError, e:
        self.logger.error('Unable to find copy remote file %s' % str(e))

def close_sftp_session(self):
    self.sftp.close()
    self.transport.close()

我想检索每个文件，并将其存放在当前的本地工作目录中。

然而，Paramiko 似乎没有这个选项 - 您需要指定完整的本地目的地。您甚至无法指定目录（例如“./”，甚至“/home/victorhooi/files”） - 您需要包括文件名的完整路径。

有什么办法解决这个问题吗？如果我们还必须指定本地文件名，而不是仅仅复制远程文件名，那会很烦人。

另外 - 我在 setup_sftp_session 中使用 exit(1) 处理异常的方式 - 这是一个很好的做法，还是有更好的方法？

干杯， 胜利者

Answer 1

你必须

os.path.join(os.getcwd(), remote_filename)

在函数中插入调用 exit() 不是一个好主意。也许您想重用代码并在出现异常时采取一些操作。如果你继续调用 exit() ，你就会迷路。
我建议修改此函数，以便在成功时返回 True，否则返回 False。然后调用者可以决定要做什么。

另一种方法是不捕获异常。因此调用者必须处理它们，并且调用者获得有关失败情况的完整信息（包括堆栈跟踪）。