我要上传文件与Python在远程服务器上。 我想先确认是否如果远程路径是真实存在的,如果实在不行,去创造。 在伪代码:
if(remote_path not exist):
create_path(remote_path)
upload_file(local_file, remote_path)
我在想的paramiko中执行一个命令来创建路径(例如mkdir -p remote_path
)。 我想出了这个:
# I didn't test this code
import paramiko, sys
ssh = paramiko.SSHClient()
ssh.connect(myhost, 22, myusername, mypassword)
ssh.exec_command('mkdir -p ' + remote_path)
ssh.close
transport = paramiko.Transport((myhost, 22))
transport.connect(username = myusername, password = mypassword)
sftp = paramiko.SFTPClient.from_transport(transport)
sftp.put(local_path, remote_path)
sftp.close()
transport.close()
但这种方法不健全的对我好,因为我关闭连接,然后再次打开它。 有没有更好的办法做到这一点?
SFTP支持常用的FTP命令(CHDIR,MKDIR,等...),所以使用这些:
sftp = paramiko.SFTPClient.from_transport(transport)
try:
sftp.chdir(remote_path) # Test if remote_path exists
except IOError:
sftp.mkdir(remote_path) # Create remote_path
sftp.chdir(remote_path)
sftp.put(local_path, '.') # At this point, you are in remote_path in either case
sftp.close()
要完全模拟mkdir -p
,就可以通过远程路径工作递归:
import os.path
def mkdir_p(sftp, remote_directory):
"""Change to this directory, recursively making new folders if needed.
Returns True if any folders were created."""
if remote_directory == '/':
# absolute path so change directory to root
sftp.chdir('/')
return
if remote_directory == '':
# top-level relative directory must exist
return
try:
sftp.chdir(remote_directory) # sub-directory exists
except IOError:
dirname, basename = os.path.split(remote_directory.rstrip('/'))
mkdir_p(sftp, dirname) # make parent directories
sftp.mkdir(basename) # sub-directory missing, so created it
sftp.chdir(basename)
return True
sftp = paramiko.SFTPClient.from_transport(transport)
mkdir_p(sftp, remote_path)
sftp.put(local_path, '.') # At this point, you are in remote_path
sftp.close()
当然,如果远程路径还包含一个远程文件名,则需要将其分离出来,正在传递的目录mkdir_p并用来代替文件名“” 在sftp.put。
简单的东西稍微更可读的太
def mkdir_p(sftp, remote, is_dir=False):
"""
emulates mkdir_p if required.
sftp - is a valid sftp object
remote - remote path to create.
"""
dirs_ = []
if is_dir:
dir_ = remote
else:
dir_, basename = os.path.split(remote)
while len(dir_) > 1:
dirs_.append(dir_)
dir_, _ = os.path.split(dir_)
if len(dir_) == 1 and not dir_.startswith("/"):
dirs_.append(dir_) # For a remote path like y/x.txt
while len(dirs_):
dir_ = dirs_.pop()
try:
sftp.stat(dir_)
except:
print "making ... dir", dir_
sftp.mkdir(dir_)
有了这个今天做。 这里是我做到了。
def mkdir_p(sftp, remote_directory):
dir_path = str()
for dir_folder in remote_directory.split("/"):
if dir_folder == "":
continue
dir_path += r"/{0}".format(dir_folder)
try:
sftp.listdir(dir_path)
except IOError:
sftp.mkdir(dir_path)
的paramiko包含的mkdir功能:
http://paramiko-docs.readthedocs.org/en/latest/api/sftp.html#paramiko.sftp_si.SFTPServerInterface.mkdir
假设SFTP操作是昂贵的,我会去的:
def sftp_mkdir_p(sftp, remote_directory):
dirs_exist = remote_directory.split('/')
dirs_make = []
# find level where dir doesn't exist
while len(dirs_exist) > 0:
try:
sftp.listdir('/'.join(dirs_exist))
break
except IOError:
value = dirs_exist.pop()
if value == '':
continue
dirs_make.append(value)
else:
return False
# ...and create dirs starting from that level
for mdir in dirs_make[::-1]:
dirs_exist.append(mdir)
sftp.mkdir('/'.join(dirs_exist))```
文章来源: Upload files using SFTP in Python, but create directories if path doesn't exist