satcfdi.zip

Complexity

Total Complexity

9

Size/Duplication

Total Lines	43
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	zip_create()	0	19	5
A	zip_file()	0	14	4

import io

from collections import namedtuple

from zipfile import ZipFile, ZipInfo


ZipData = namedtuple("ZipFile", "filename data")



def zip_create(target: io.BytesIO, files: list[ZipData]):

    p = target.tell()


    with ZipFile(target, "w") as myzip:

        for f in files:

            zinfo = ZipInfo(

                filename=f.filename

            )

            zinfo.filename = f.filename

            zinfo.compress_type = 8

            zinfo.create_system = 0


            with myzip.open(zinfo, 'w') as stream:

                zinfo.flag_bits = 2056

                zinfo.external_attr = 0

                f.data(stream)


    with target.getbuffer() as view:  # change zip flag bytes

        view[p + 6:p + 8] = b"\x08\x08"



def zip_file(zipfile, files: list[ZipData]):

    # Create a ZipFile object in write mode

    with ZipFile(zipfile, 'w') as zipf:

        # Add the input file to the zip archive with its base name

        for f in files:

            zinfo = ZipInfo(

                filename=f.filename,

                # date_time=datetime_to_tuple(datetime.now())

            )

            zinfo.compress_type = 8

            zinfo.create_system = 0


            with zipf.open(zinfo, 'w') as stream:

                stream.write(f.data)