import zipfile
from io import BytesIO


def zip_file(byte_file):
    buf = BytesIO()
    with zipfile.ZipFile(buf, mode="w", compression=zipfile.ZIP_DEFLATED) as zf:
        zf.writestr('result', byte_file)
    file = buf.getvalue()
    buf.close()
    return file


def unzip_file(byte_file):
    buf = BytesIO()
    buf.write(byte_file)
    zipfile_obj = zipfile.ZipFile(buf, compression=zipfile.ZIP_DEFLATED)
    file = zipfile_obj.read(zipfile_obj.namelist()[0])
    buf.close()
    zipfile_obj.close()
    return file