sample code for generating a zip file in memory using Python. Useful for AWS Lambda Python

zipfilestring.py

import StringIO
import zipfile
import os
import os.path

def append_requests_module_code(zf):
    requests_path = os.path.dirname(os.path.realpath(requests.__file__))
    for root, dirs, files in os.walk(requests_path):
        ziproot = os.path.join('requests', root[len(requests_path)+1:])
        for fn in files:
            if not fn.endswith('.pyc'):
                zf.write(os.path.join(root, fn), os.path.join(ziproot, fn))


def append_delta_function_code(zf):
    zf.write('path/to/pythonfile.py', 'pythonfile.py')


def get_zip_file():
    zipfile_content = StringIO.StringIO()
    zf = zipfile.ZipFile(zipfile_content, mode='w',
        compression=zipfile.ZIP_DEFLATED)

    append_requests_module_code(zf)
    append_delta_function_code(zf)

    zf.close()
    return zipfile_content.getvalue()