在Python中压缩文件

Question

我的程序运行顺畅，但我希望从ftp下载的文件在我的本地驱动器中被压缩

问题在于仅有一个文件在调用主函数后被压缩

以下是我的代码：

import os
import upload
import download
import zipfile
import ConfigParser
import ftputil

def main():
    
    #create a folder Temp on d drive for later use
    path = r'D:\Temp'
    os.mkdir(path)
    
    #parse all the  values at config.ini file
    config = ConfigParser.ConfigParser()
    config.readfp(open('config.ini'))
    server = config.get('main', 'Server')
    username = config.get('main', 'Username')
    password = config.get('main', 'Password')
    uploads = config.get('main', 'Upload folder')
    downloads = config.get('main', 'Download folder')

    #connect to ftp
    ftp = ftputil.FTPHost(server, username, password)

    dirlist = ftp.listdir(downloads)
    
    for list in dirlist:
        ftp.chdir(downloads)
        target = os.path.join(path, list)
        ftp.download(list, target)
        
    
    #########################################################
    #   THis section is where algo fails but the program run#
    ########################################################
    
    #zipping files
    absolute_path = r'D:\Temp'
    dirlist = os.listdir(absolute_path)
    filepath = r'D:\Temp\project2.zip'
    for list in dirlist:
        get_file = os.path.join(absolute_path, list)
        zip_name = zipfile.ZipFile(filepath, 'w')
        zip_name.write(get_file, 'Project2b\\' + list)
        
                
        

if __name__ == '__main__':
    print "cannot be"

- neo

---

你的归档文件中是否包含最后一个文件 target？ - Dmitry Zagorulkin

你缺少压缩方法，如果没有给出zipfile.ZIP_DEFLATED，则文件将不会被压缩。应该使用zipfile.ZipFile（filepath，'w'，zipfile.ZIP_DEFLATED）来包括压缩。 - Aravind Krishnakumar

Answer 1

当你这样做时：

for list in dirlist:
        get_file = os.path.join(absolute_path, list)
        zip_name = zipfile.ZipFile(filepath, 'w')
        zip_name.write(get_file, 'Project2b\\' + list)

你需要为每个要压缩的文件重新创建一个ZipFile对象，“w”模式意味着你需要从头开始创建。

尝试这样做（在循环之前创建zip文件）：

zip_name = zipfile.ZipFile(filepath, 'w')
for list in dirlist:
        get_file = os.path.join(absolute_path, list)
        zip_name.write(get_file, 'Project2b\\' + list)

或者使用这个，它将以追加模式打开zip文件：

for list in dirlist:
        get_file = os.path.join(absolute_path, list)
        zip_name = zipfile.ZipFile(filepath, 'a')
        zip_name.write(get_file, 'Project2b\\' + list)

- Cédric Julien

---

耶！这个可行！！抱歉我是新手！我还不能点击最佳答案按钮，等待3分钟。 :D 再次感谢！ - neo

Answer 2

请查看shutil模块。有一个使用shutil.make_archive()的示例： http://docs.python.org/library/shutil.html

- geertjanvdk

---

我们需要使用zipfile模块来完成这个任务。 :D - neo

make_archive() 使用 zipfile 模块。在我看来，这只是更方便而已。 - geertjanvdk

Answer 3

如果您有很多文件，可以并行压缩它们：

import zipfile
from pathlib import Path, WindowsPath
from typing import List, Text
import logging
from time import time
from concurrent.futures import ThreadPoolExecutor

logging.basicConfig(
    format="%(asctime)s - %(message)s", datefmt="%H:%M:%S", level=logging.DEBUG
)

PATH = (r"\\some_directory\subdirectory\zipped")


def file_names() -> List[WindowsPath]:
    p = Path(PATH)
    file_names = list(p.glob("./*.csv"))
    logging.info("There are %d files", len(file_names))
    return file_names


def zip_file(file: WindowsPath) -> None:
    zip_file_name = Path(PATH, f"{file.stem}.zip")
    with zipfile.ZipFile(zip_file_name, "w") as zip:
        zip.write(file, arcname=file.name, compress_type=zipfile.ZIP_DEFLATED)


def main(files: List[Text]) -> None:
    t0 = time()
    number_of_files = len(files)
    with ThreadPoolExecutor() as executor:
        for counter, _ in enumerate(executor.map(zip_file, files), start=1):
            # update progress every 100 files
            if counter % 100 == 0:
                logging.info(
                    "Processed %d/%d. TT: %d:%d",
                    counter,
                    number_of_files,
                    *divmod(int(time() - t0), 60),
                )

    logging.info(
        "Finished zipping %d files. Total time: %d:%d",
        len(files),
        *divmod(int(time() - t0), 60),
    )


if __name__ == "__main__":
    files = file_names()
    main(files)

- Vlad Bezden

Answer 4

最好的方法是在您的for循环中放置调试语句，有两种可能性;

一种是第一个for循环仅从ftp文件夹下载一个文件

第二个是第一个循环下载所有文件，但第二个循环仅压缩其中一个文件

使用打印语句查看循环中下载/压缩的文件，祝你好运

- sed

---

谢谢建议，我会尝试的。实际上第一个循环没问题，它可以下载我FTP中的所有文件，但第二个循环有问题。 - neo