Files
truepolyglot/tests/test_zip.py
2018-09-18 10:52:38 +02:00

22 lines
525 B
Python
Executable File

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import sys
sys.path.append("../")
import tempfile
from ZipFileTransformer import Zip
input_file = "./samples/test1.zip"
output_file = tempfile.mktemp()
print("Output: " + output_file)
z = Zip(input_file)
a = bytearray(b'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA')
b = bytearray(b'BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB')
z.add_data_to_file(a, b, False)
g = open(output_file, "wb")
g.write(a + z.get_local_file_data() + b + z.get_data_after_central_directory())
g.close()