Function make_zip_archive

tools/create_zipfile.py:146–171

Create a zip archive out of a input folder Parameters ---------- archive_path Path to the zip file that will be created input_dir input dir to compress compression_level compression level of the resulting zip file.

(
    archive_path: Path,
    input_dir: Path,
    compression_level: int = 6,
)

def make_zip_archive(
    archive_path: Path,
    input_dir: Path,
    compression_level: int = 6,
) -> None:
    """Create a zip archive out of a input folder

    Parameters
    ----------
    archive_path
       Path to the zip file that will be created
    input_dir
       input dir to compress
    compression_level
       compression level of the resulting zip file.
    """
    if compression_level > 0:
        compression = zipfile.ZIP_DEFLATED
    else:
        compression = zipfile.ZIP_STORED

    with zipfile.ZipFile(
        archive_path, "w", compression=compression, compresslevel=compression_level
    ) as zf:
        for file in input_dir.rglob("*"):
            zf.write(file, file.relative_to(input_dir))


if __name__ == "__main__":