Quick Python script to zip a directory. Useful for AWS Lambda deployments from an npm script, among other things.

zip.py

#!/usr/bin/env python
import sys
import os
import shutil
import tempfile
import uuid

# python zip.py 'folder to zip' 'output file'
# e.g. python zip.py ./ ./build/build.zip
if __name__ == '__main__':
    srcDir = os.path.abspath(sys.argv[1])
    outputFile = os.path.abspath(sys.argv[2])

    tempDir = os.path.join(tempfile.gettempdir(), 'zippy-builds')
    tempFile = os.path.join(tempDir, str(uuid.uuid4()))

    print('creating deployable zip from ' + srcDir + '...')

    try:
        print ('creating zip in \'' + tempDir + '\'...')
        shutil.make_archive(tempFile, 'zip', srcDir)
        
        if os.path.exists(outputFile):
            print('output file exists.  deleting \'' + outputFile + '\'')
            os.remove(outputFile)

        print('copying \'' + tempFile + '.zip\' to \'' + outputFile + '\'...')
        shutil.move(tempFile + '.zip', outputFile)

        print('zip succeeded.')
    except:
        e = sys.exc_info()[0]
        print('error: ' + e)
        print('cleaning up...')

        shutil.rmtree(tempDir)

        print('zip failed.')