Comment puis-je créer une archive zip d'une structure de répertoires en Python?

Comme d'autres l'ont souligné, vous devez utiliser un fichier zip . La documentation vous indique quelles fonctions sont disponibles, mais n'explique pas vraiment comment vous pouvez les utiliser pour compresser un répertoire entier. Je pense qu'il est plus facile d'expliquer avec un exemple de code:

#!/usr/bin/env python
import os
import zipfile

def zipdir(path, ziph):
    # ziph is zipfile handle
    for root, dirs, files in os.walk(path):
        for file in files:
            ziph.write(os.path.join(root, file))

if __name__ == '__main__':
    zipf = zipfile.ZipFile('Python.zip', 'w', zipfile.ZIP_DEFLATED)
    zipdir('tmp/', zipf)
    zipf.close()

Adapté de: http://www.devshed.com/c/a/Python/Python-UnZipped/

La façon la plus simple est d'utiliser shutil.make_archive. Il prend en charge les formats zip et tar.

import shutil
shutil.make_archive(output_filename, 'zip', dir_name)

Si vous devez faire quelque chose de plus compliqué que de zipper tout le répertoire (comme sauter certains fichiers), alors vous devrez creuser dans le zipfilemodule comme d'autres l'ont suggéré.

Pour ajouter le contenu de mydirectoryà un nouveau fichier zip, y compris tous les fichiers et sous-répertoires:

import os
import zipfile

zf = zipfile.ZipFile("myzipfile.zip", "w")
for dirname, subdirs, files in os.walk("mydirectory"):
    zf.write(dirname)
    for filename in files:
        zf.write(os.path.join(dirname, filename))
zf.close()

Comment puis-je créer une archive zip d'une structure de répertoires en Python?

Dans un script Python

En Python 2.7+, shutila une make_archivefonction.

from shutil import make_archive
make_archive(
  'zipfile_name', 
  'zip',           # the archive format - or tar, bztar, gztar 
  root_dir=None,   # root for archive - current working dir if None
  base_dir=None)   # start archiving from here - cwd if None too

Ici, l'archive zippée sera nommée zipfile_name.zip. Si base_direst plus bas, root_diril exclura les fichiers qui ne se trouvent pas dans le base_dir, mais archivera toujours les fichiers dans les répertoires parents jusqu'auroot_dir .

J'ai eu un problème en testant cela sur Cygwin avec 2.7 - il veut un argument root_dir, pour cwd:

make_archive('zipfile_name', 'zip', root_dir='.')

Utiliser Python à partir du shell

Vous pouvez le faire avec Python à partir du shell en utilisant également le zipfilemodule:

$ python -m zipfile -c zipname sourcedir

Où zipnameest le nom du fichier de destination que vous souhaitez (ajoutez .zipsi vous le souhaitez, il ne le fera pas automatiquement) et sourcedir est le chemin d'accès au répertoire.

Compressez Python (ou ne voulez tout simplement pas le répertoire parent):

Si vous essayez de compresser un paquet python avec un __init__.pyet __main__.py, et que vous ne voulez pas le répertoire parent, c'est

$ python -m zipfile -c zipname sourcedir/*

Et

$ python zipname

exécuterait le package. (Notez que vous ne pouvez pas exécuter de sous-packages comme point d'entrée à partir d'une archive zippée.)

Zipper une application Python:

Si vous avez python3.5 + et que vous souhaitez spécifiquement compresser un paquet Python, utilisez zipapp :

$ python -m zipapp myapp
$ python myapp.pyz

Cette fonction compressera récursivement une arborescence de répertoires, compressant les fichiers et enregistrant les noms de fichiers relatifs corrects dans l'archive. Les entrées d'archive sont les mêmes que celles générées par zip -r output.zip source_dir.

import os
import zipfile
def make_zipfile(output_filename, source_dir):
    relroot = os.path.abspath(os.path.join(source_dir, os.pardir))
    with zipfile.ZipFile(output_filename, "w", zipfile.ZIP_DEFLATED) as zip:
        for root, dirs, files in os.walk(source_dir):
            # add directory (needed for empty dirs)
            zip.write(root, os.path.relpath(root, relroot))
            for file in files:
                filename = os.path.join(root, file)
                if os.path.isfile(filename): # regular files only
                    arcname = os.path.join(os.path.relpath(root, relroot), file)
                    zip.write(filename, arcname)

Utilisez shutil, qui fait partie de l'ensemble de bibliothèques standard python. L'utilisation de shutil est si simple (voir le code ci-dessous):

• 1er argument: nom du fichier zip / tar résultant,
• 2ème argument: zip / tar,
• 3e argument: dir_name

Code:

import shutil
shutil.make_archive('/home/user/Desktop/Filename','zip','/home/username/Desktop/Directory')

Pour ajouter une compression au fichier zip résultant, consultez ce lien .

Vous devez changer:

zip = zipfile.ZipFile('Python.zip', 'w')

à

zip = zipfile.ZipFile('Python.zip', 'w', zipfile.ZIP_DEFLATED)

J'ai apporté quelques modifications au code donné par Mark Byers . La fonction ci-dessous ajoute également des répertoires vides si vous en avez. Les exemples devraient clarifier le chemin ajouté au zip.

#!/usr/bin/env python
import os
import zipfile

def addDirToZip(zipHandle, path, basePath=""):
    """
    Adding directory given by \a path to opened zip file \a zipHandle

    @param basePath path that will be removed from \a path when adding to archive

    Examples:
        # add whole "dir" to "test.zip" (when you open "test.zip" you will see only "dir")
        zipHandle = zipfile.ZipFile('test.zip', 'w')
        addDirToZip(zipHandle, 'dir')
        zipHandle.close()

        # add contents of "dir" to "test.zip" (when you open "test.zip" you will see only it's contents)
        zipHandle = zipfile.ZipFile('test.zip', 'w')
        addDirToZip(zipHandle, 'dir', 'dir')
        zipHandle.close()

        # add contents of "dir/subdir" to "test.zip" (when you open "test.zip" you will see only contents of "subdir")
        zipHandle = zipfile.ZipFile('test.zip', 'w')
        addDirToZip(zipHandle, 'dir/subdir', 'dir/subdir')
        zipHandle.close()

        # add whole "dir/subdir" to "test.zip" (when you open "test.zip" you will see only "subdir")
        zipHandle = zipfile.ZipFile('test.zip', 'w')
        addDirToZip(zipHandle, 'dir/subdir', 'dir')
        zipHandle.close()

        # add whole "dir/subdir" with full path to "test.zip" (when you open "test.zip" you will see only "dir" and inside it only "subdir")
        zipHandle = zipfile.ZipFile('test.zip', 'w')
        addDirToZip(zipHandle, 'dir/subdir')
        zipHandle.close()

        # add whole "dir" and "otherDir" (with full path) to "test.zip" (when you open "test.zip" you will see only "dir" and "otherDir")
        zipHandle = zipfile.ZipFile('test.zip', 'w')
        addDirToZip(zipHandle, 'dir')
        addDirToZip(zipHandle, 'otherDir')
        zipHandle.close()
    """
    basePath = basePath.rstrip("\\/") + ""
    basePath = basePath.rstrip("\\/")
    for root, dirs, files in os.walk(path):
        # add dir itself (needed for empty dirs
        zipHandle.write(os.path.join(root, "."))
        # add files
        for file in files:
            filePath = os.path.join(root, file)
            inZipPath = filePath.replace(basePath, "", 1).lstrip("\\/")
            #print filePath + " , " + inZipPath
            zipHandle.write(filePath, inZipPath)

Ci-dessus est une fonction simple qui devrait fonctionner pour des cas simples. Vous pouvez trouver une classe plus élégante dans mon Gist: https://gist.github.com/Eccenux/17526123107ca0ac28e6

Python moderne (3.6+) utilisant le pathlibmodule pour une gestion concise de type OOP des chemins, et pathlib.Path.rglob()pour un globbing récursif. Pour autant que je sache, cela équivaut à la réponse de George V. Reilly: zips avec compression, l'élément le plus haut est un répertoire, conserve des répertoires vides, utilise des chemins relatifs.

from pathlib import Path
from zipfile import ZIP_DEFLATED, ZipFile

from os import PathLike
from typing import Union


def zip_dir(zip_name: str, source_dir: Union[str, PathLike]):
    src_path = Path(source_dir).expanduser().resolve(strict=True)
    with ZipFile(zip_name, 'w', ZIP_DEFLATED) as zf:
        for file in src_path.rglob('*'):
            zf.write(file, file.relative_to(src_path.parent))

Remarque: comme les indications de type facultatives l'indiquent, zip_namene peut pas être un objet Path ( serait corrigé dans 3.6.2+ ).

J'ai un autre exemple de code qui peut aider, en utilisant python3, pathlib et zipfile. Cela devrait fonctionner dans n'importe quel système d'exploitation.

from pathlib import Path
import zipfile
from datetime import datetime

DATE_FORMAT = '%y%m%d'


def date_str():
    """returns the today string year, month, day"""
    return '{}'.format(datetime.now().strftime(DATE_FORMAT))


def zip_name(path):
    """returns the zip filename as string"""
    cur_dir = Path(path).resolve()
    parent_dir = cur_dir.parents[0]
    zip_filename = '{}/{}_{}.zip'.format(parent_dir, cur_dir.name, date_str())
    p_zip = Path(zip_filename)
    n = 1
    while p_zip.exists():
        zip_filename = ('{}/{}_{}_{}.zip'.format(parent_dir, cur_dir.name,
                                             date_str(), n))
        p_zip = Path(zip_filename)
        n += 1
    return zip_filename


def all_files(path):
    """iterator returns all files and folders from path as absolute path string
    """
    for child in Path(path).iterdir():
        yield str(child)
        if child.is_dir():
            for grand_child in all_files(str(child)):
                yield str(Path(grand_child))


def zip_dir(path):
    """generate a zip"""
    zip_filename = zip_name(path)
    zip_file = zipfile.ZipFile(zip_filename, 'w')
    print('create:', zip_filename)
    for file in all_files(path):
        print('adding... ', file)
        zip_file.write(file)
    zip_file.close()


if __name__ == '__main__':
    zip_dir('.')
    print('end!')

Vous voudrez probablement regarder le zipfilemodule; il y a de la documentation sur http://docs.python.org/library/zipfile.html .

Vous pouvez également vouloir os.walk()indexer la structure du répertoire.

Voici une variation de la réponse donnée par Nux qui fonctionne pour moi:

def WriteDirectoryToZipFile( zipHandle, srcPath, zipLocalPath = "", zipOperation = zipfile.ZIP_DEFLATED ):
    basePath = os.path.split( srcPath )[ 0 ]
    for root, dirs, files in os.walk( srcPath ):
        p = os.path.join( zipLocalPath, root [ ( len( basePath ) + 1 ) : ] )
        # add dir
        zipHandle.write( root, p, zipOperation )
        # add files
        for f in files:
            filePath = os.path.join( root, f )
            fileInZipPath = os.path.join( p, f )
            zipHandle.write( filePath, fileInZipPath, zipOperation )

Essayez celui ci-dessous. Cela a fonctionné pour moi .

import zipfile, os
zipf = "compress.zip"  
def main():
    directory = r"Filepath"
    toZip(directory)
def toZip(directory):
    zippedHelp = zipfile.ZipFile(zipf, "w", compression=zipfile.ZIP_DEFLATED )

    list = os.listdir(directory)
    for file_list in list:
        file_name = os.path.join(directory,file_list)

        if os.path.isfile(file_name):
            print file_name
            zippedHelp.write(file_name)
        else:
            addFolderToZip(zippedHelp,file_list,directory)
            print "---------------Directory Found-----------------------"
    zippedHelp.close()

def addFolderToZip(zippedHelp,folder,directory):
    path=os.path.join(directory,folder)
    print path
    file_list=os.listdir(path)
    for file_name in file_list:
        file_path=os.path.join(path,file_name)
        if os.path.isfile(file_path):
            zippedHelp.write(file_path)
        elif os.path.isdir(file_name):
            print "------------------sub directory found--------------------"
            addFolderToZip(zippedHelp,file_name,path)


if __name__=="__main__":
    main()

Si vous voulez une fonctionnalité comme le dossier compresser de n'importe quel gestionnaire de fichiers graphique commun, vous pouvez utiliser le code suivant, il utilise le module zipfile . En utilisant ce code, vous aurez le fichier zip avec le chemin comme dossier racine.

import os
import zipfile

def zipdir(path, ziph):
    # Iterate all the directories and files
    for root, dirs, files in os.walk(path):
        # Create a prefix variable with the folder structure inside the path folder. 
        # So if a file is at the path directory will be at the root directory of the zip file
        # so the prefix will be empty. If the file belongs to a containing folder of path folder 
        # then the prefix will be that folder.
        if root.replace(path,'') == '':
                prefix = ''
        else:
                # Keep the folder structure after the path folder, append a '/' at the end 
                # and remome the first character, if it is a '/' in order to have a path like 
                # folder1/folder2/file.txt
                prefix = root.replace(path, '') + '/'
                if (prefix[0] == '/'):
                        prefix = prefix[1:]
        for filename in files:
                actual_file_path = root + '/' + filename
                zipped_file_path = prefix + filename
                zipf.write( actual_file_path, zipped_file_path)


zipf = zipfile.ZipFile('Python.zip', 'w', zipfile.ZIP_DEFLATED)
zipdir('/tmp/justtest/', zipf)
zipf.close()

Pour donner plus de flexibilité, par exemple sélectionnez répertoire / fichier par nom, utilisez:

import os
import zipfile

def zipall(ob, path, rel=""):
    basename = os.path.basename(path)
    if os.path.isdir(path):
        if rel == "":
            rel = basename
        ob.write(path, os.path.join(rel))
        for root, dirs, files in os.walk(path):
            for d in dirs:
                zipall(ob, os.path.join(root, d), os.path.join(rel, d))
            for f in files:
                ob.write(os.path.join(root, f), os.path.join(rel, f))
            break
    elif os.path.isfile(path):
        ob.write(path, os.path.join(rel, basename))
    else:
        pass

Pour une arborescence de fichiers:

.
├── dir
│   ├── dir2
│   │   └── file2.txt
│   ├── dir3
│   │   └── file3.txt
│   └── file.txt
├── dir4
│   ├── dir5
│   └── file4.txt
├── listdir.zip
├── main.py
├── root.txt
└── selective.zip

Vous pouvez par exemple sélectionner uniquement dir4et root.txt:

cwd = os.getcwd()
files = [os.path.join(cwd, f) for f in ['dir4', 'root.txt']]

with zipfile.ZipFile("selective.zip", "w" ) as myzip:
    for f in files:
        zipall(myzip, f)

Ou tout simplement listdirdans le répertoire d'invocation de script et ajoutez tout à partir de là:

with zipfile.ZipFile("listdir.zip", "w" ) as myzip:
    for f in os.listdir():
        if f == "listdir.zip":
            # Creating a listdir.zip in the same directory
            # will include listdir.zip inside itself, beware of this
            continue
        zipall(myzip, f)

Supposons que vous souhaitiez compresser tous les dossiers (sous-répertoires) du répertoire actuel.

for root, dirs, files in os.walk("."):
    for sub_dir in dirs:
        zip_you_want = sub_dir+".zip"
        zip_process = zipfile.ZipFile(zip_you_want, "w", zipfile.ZIP_DEFLATED)
        zip_process.write(file_you_want_to_include)
        zip_process.close()

        print("Successfully zipped directory: {sub_dir}".format(sub_dir=sub_dir))

Pour un moyen concis de conserver la hiérarchie des dossiers sous le répertoire parent à archiver:

import glob
import zipfile

with zipfile.ZipFile(fp_zip, "w", zipfile.ZIP_DEFLATED) as zipf:
    for fp in glob(os.path.join(parent, "**/*")):
        base = os.path.commonpath([parent, fp])
        zipf.write(fp, arcname=fp.replace(base, ""))

Si vous le souhaitez, vous pouvez le modifier pour l'utiliser pathlib pour la globalisation de fichiers .

Tant de réponses ici, et j'espère que je pourrais contribuer avec ma propre version, qui est basée sur la réponse originale (en passant), mais avec une perspective plus graphique, en utilisant également le contexte pour chaque zipfileconfiguration et tri os.walk(), afin d'avoir un sortie ordonnée.

Ayant ces dossiers et ces fichiers (entre autres dossiers), je voulais créer un .zippour chaque cap_dossier:

$ tree -d
.
├── cap_01
|    ├── 0101000001.json
|    ├── 0101000002.json
|    ├── 0101000003.json
|
├── cap_02
|    ├── 0201000001.json
|    ├── 0201000002.json
|    ├── 0201001003.json
|
├── cap_03
|    ├── 0301000001.json
|    ├── 0301000002.json
|    ├── 0301000003.json
| 
├── docs
|    ├── map.txt
|    ├── main_data.xml
|
├── core_files
     ├── core_master
     ├── core_slave

Voici ce que j'ai appliqué, avec des commentaires pour une meilleure compréhension du processus.

$ cat zip_cap_dirs.py 
""" Zip 'cap_*' directories. """           
import os                                                                       
import zipfile as zf                                                            


for root, dirs, files in sorted(os.walk('.')):                                                                                               
    if 'cap_' in root:                                                          
        print(f"Compressing: {root}")                                           
        # Defining .zip name, according to Capítulo.                            
        cap_dir_zip = '{}.zip'.format(root)                                     
        # Opening zipfile context for current root dir.                         
        with zf.ZipFile(cap_dir_zip, 'w', zf.ZIP_DEFLATED) as new_zip:          
            # Iterating over os.walk list of files for the current root dir.    
            for f in files:                                                     
                # Defining relative path to files from current root dir.        
                f_path = os.path.join(root, f)                                  
                # Writing the file on the .zip file of the context              
                new_zip.write(f_path) 

Fondamentalement, pour chaque itération terminée os.walk(path), j'ouvre un contexte pour la zipfileconfiguration et ensuite, itère l'itération sur files, qui est un listdes fichiers du rootrépertoire, formant le chemin relatif pour chaque fichier en fonction du rootrépertoire actuel , en ajoutant à lazipfile contexte qui est en cours d'exécution .

Et la sortie est présentée comme ceci:

$ python3 zip_cap_dirs.py
Compressing: ./cap_01
Compressing: ./cap_02
Compressing: ./cap_03

Pour voir le contenu de chaque .ziprépertoire, vous pouvez utiliser la lesscommande:

$ less cap_01.zip

Archive:  cap_01.zip
 Length   Method    Size  Cmpr    Date    Time   CRC-32   Name
--------  ------  ------- ---- ---------- ----- --------  ----
  22017  Defl:N     2471  89% 2019-09-05 08:05 7a3b5ec6  cap_01/0101000001.json
  21998  Defl:N     2471  89% 2019-09-05 08:05 155bece7  cap_01/0101000002.json
  23236  Defl:N     2573  89% 2019-09-05 08:05 55fced20  cap_01/0101000003.json
--------          ------- ---                           -------
  67251             7515  89%                            3 files

Voici une approche moderne, utilisant pathlib et un gestionnaire de contexte. Place les fichiers directement dans le zip, plutôt que dans un sous-dossier.

def zip_dir(filename: str, dir_to_zip: pathlib.Path):
    with zipfile.ZipFile(filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
        # Use glob instead of iterdir(), to cover all subdirectories.
        for directory in dir_to_zip.glob('**'):
            for file in directory.iterdir():
                if not file.is_file():
                    continue
                # Strip the first component, so we don't create an uneeded subdirectory
                # containing everything.
                zip_path = pathlib.Path(*file.parts[1:])
                # Use a string, since zipfile doesn't support pathlib  directly.
                zipf.write(str(file), str(zip_path))

J'ai préparé une fonction en consolidant la solution de Mark Byers avec les commentaires de Reimund et Morten Zilmer (chemin relatif et y compris les répertoires vides). En tant que meilleure pratique,with est utilisé dans la construction de fichiers ZipFile.

La fonction prépare également un nom de fichier zip par défaut avec le nom du répertoire compressé et l'extension «.zip». Par conséquent, cela fonctionne avec un seul argument: le répertoire source à compresser.

import os
import zipfile

def zip_dir(path_dir, path_file_zip=''):
if not path_file_zip:
    path_file_zip = os.path.join(
        os.path.dirname(path_dir), os.path.basename(path_dir)+'.zip')
with zipfile.ZipFile(path_file_zip, 'wb', zipfile.ZIP_DEFLATED) as zip_file:
    for root, dirs, files in os.walk(path_dir):
        for file_or_dir in files + dirs:
            zip_file.write(
                os.path.join(root, file_or_dir),
                os.path.relpath(os.path.join(root, file_or_dir),
                                os.path.join(path_dir, os.path.pardir)))

# import required python modules
# You have to install zipfile package using pip install

import os,zipfile

# Change the directory where you want your new zip file to be

os.chdir('Type your destination')

# Create a new zipfile ( I called it myfile )

zf = zipfile.ZipFile('myfile.zip','w')

# os.walk gives a directory tree. Access the files using a for loop

for dirnames,folders,files in os.walk('Type your directory'):
    zf.write('Type your Directory')
    for file in files:
        zf.write(os.path.join('Type your directory',file))

Eh bien, après avoir lu les suggestions, j'ai trouvé une méthode très similaire qui fonctionne avec 2.7.x sans créer de noms de répertoires "drôles" (noms absolus), et ne crée que le dossier spécifié à l'intérieur du zip.

Ou juste au cas où vous auriez besoin que votre zip contienne un dossier à l'intérieur avec le contenu du répertoire sélectionné.

def zipDir( path, ziph ) :
 """
 Inserts directory (path) into zipfile instance (ziph)
 """
 for root, dirs, files in os.walk( path ) :
  for file in files :
   ziph.write( os.path.join( root, file ) , os.path.basename( os.path.normpath( path ) ) + "\\" + file )

def makeZip( pathToFolder ) :
 """
 Creates a zip file with the specified folder
 """
 zipf = zipfile.ZipFile( pathToFolder + 'file.zip', 'w', zipfile.ZIP_DEFLATED )
 zipDir( pathToFolder, zipf )
 zipf.close()
 print( "Zip file saved to: " + pathToFolder)

makeZip( "c:\\path\\to\\folder\\to\\insert\\into\\zipfile" )

Fonction pour créer un fichier zip.

def CREATEZIPFILE(zipname, path):
    #function to create a zip file
    #Parameters: zipname - name of the zip file; path - name of folder/file to be put in zip file

    zipf = zipfile.ZipFile(zipname, 'w', zipfile.ZIP_DEFLATED)
    zipf.setpassword(b"password") #if you want to set password to zipfile

    #checks if the path is file or directory
    if os.path.isdir(path):
        for files in os.listdir(path):
            zipf.write(os.path.join(path, files), files)

    elif os.path.isfile(path):
        zipf.write(os.path.join(path), path)
    zipf.close()

Utiliser zipfly

import zipfly

paths = [
    {
        'fs': '/path/to/large/file'
    },
]

zfly = zipfly.ZipFly( paths = paths )

with open("large.zip", "wb") as f:
    for i in zfly.generator():
        f.write(i)