Télécharger des fichiers en utilisant SFTP en Python, mais créer des répertoires si le chemin n'existe pas
je veux télécharger un fichier sur un serveur distant avec Python. J'aimerais vérifier à l'avance si le chemin distant existe vraiment, et si ce n'est pas le cas, le créer. En pseudo-code:
if(remote_path not exist):
create_path(remote_path)
upload_file(local_file, remote_path)
je pensais exécuter une commande dans Paramiko pour créer le chemin (e.g. ). J'ai trouvé ceci:
# 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()
mais cette solution ne me semble pas bonne, parce que je ferme la connexion et je la rouvre à nouveau. Est-il un meilleur moyen de le faire?
4 réponses
SFTP supporte les commandes FTP habituelles (chdir, mkdir, etc...), afin d'utiliser ces:
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()
Pour imiter mkdir -p, vous pouvez travailler à travers remote_path de manière récursive:
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()
bien sûr, si remote_path contient aussi un nom de fichier distant, alors il doit être séparé, le répertoire étant passé à mkdir_p et le nom de fichier utilisé à la place de '."en sftp.mettre.
quelque chose de plus simple et légèrement plus lisible aussi
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_)
Dû le faire aujourd'hui. Voici comment je l'ai fait.
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)