Function build_archive

tests/helpers.py:31–53

(path, name, archive_format, files)

def build_archive(path, name, archive_format, files):
    filepath = path / f"{name}.{archive_format}"

    if archive_format == "tar.gz":
        with tarfile.open(filepath, "x:gz") as archive:
            for mname, content in files.items():
                if isinstance(content, tarfile.TarInfo):
                    content.name = mname
                    archive.addfile(content)
                else:
                    data = textwrap.dedent(content).encode("utf8")
                    member = tarfile.TarInfo(mname)
                    member.size = len(data)
                    archive.addfile(member, io.BytesIO(data))
        return filepath

    if archive_format == "zip":
        with zipfile.ZipFile(filepath, mode="w") as archive:
            for mname, content in files.items():
                archive.writestr(mname, textwrap.dedent(content))
        return filepath

    raise ValueError(format)