python实现ftp上传目录中所有文件且结构不变

2022-11-15编程源码 浏览

    from ftplib import FTP
    import os, re

再设定所需的参数IP，PORT。

    self.ftp = FTP()
    self.IP = IP
    self.PORT = PORT
    self.buf_size = 1024

我们就可以开始解决问题，首先要解决的是目录结构的保存，这里可以采用os.walk()来获取

    def dir_path(self, path):
            cont = []
            for content_list in os.walk(path):
                    cont.append(content_list)
            return cont

返回出来的列表中的数据为(当前文件夹路径,文件夹，文件)，就这样目录结构的数据已经得到了，接下来开始实现FTP的连接

    def start_ftp(self, user=None, pwd=None):
            self.ftp.connect(self.IP, self.PORT)
            if user is None:
                    user = ""
                    pwd = ""
            self.ftp.login(user, pwd)
            print("登录成功")

连接后就可以开始实现文件的上传问题（并没有写太多判断是否存在的代码，毕竟没写多长时间）

    def upload_file(self, local_path, net_path):
            with open(local_path, 'rb') as file:
                    self.ftp.storbinary(f'STOR {net_path}', file, self.buf_size)
            print(local_path)

这是一个最基本的上传代码，每次只能单个文件，我们现在要做的就是对代码进行改进。
已经知道了我们得到的cont里面文件结构的形式，于是我们可以利用os.path.join()这个函数来组合文件路径，那么又产生了个难点——ftp端的目录结构怎么建立。
苦思冥想之下，我想到了万能的正则表达式，使用正则表达所去掉路径中不需要的部分，再利用ftp.mkd()创建文件夹。到这里问题基本就解决了，剩下的全放在简易代码中。

    from ftplib import FTP
    import os, re

    class Ftp_th:
            def __init__(self, IP, PORT):
                    self.ftp = FTP()
                    self.IP = IP
                    self.PORT = PORT
                    self.buf_size = 1024

            def dir_path(self, path):
                    cont = []
                    for content_list in os.walk(path):
                            cont.append(content_list)
                    return cont

            def start_ftp(self, user=None, pwd=None):
                    self.ftp.connect(self.IP, self.PORT)
                    if user is None:
                            user = ""
                            pwd = ""
                    self.ftp.login(user, pwd)
                    print("登录成功")

            def upload_file(self, local_path, net_path):
                    with open(local_path, 'rb') as file:
                            self.ftp.storbinary(f'STOR {net_path}', file, self.buf_size)
                    print(local_path)

            def th_upload_file(self, local_path, net_path, cont):
                    for items in cont:
                            if items[2] == None: pass
                            else:
                                    for item_1 in items[2]:
                                            path_1 = re.findall(r'(?<={}/).*'.format(local_path), os.path.join(items[0], item_1))[0]
                                            print(path_1)
                                            with open(os.path.join(items[0], item_1), 'rb') as file:
                                                    self.ftp.storbinary(f'STOR {path_1}', file, self.buf_size)
                            if items[1] == None: pass
                            else:
                                    for item_2 in items[1]:
                                            print(item_2)
                                            path_2 = re.findall(r'(?<={}).*'.format(local_path), os.path.join(local_path, item_2))[0]
                                            print(path_2)
                                            self.ftp.mkd(path_2)