zip entire directory in python

zip_directory.py

from zipfile import ZipFile
import os
from os.path import basename

def zip_dir(dirname, zipfilename):
    with ZipFile(zipfilename, 'w') as zipObj:
        # Iterate over all the files in directory
        for folderName, subfolders, filenames in os.walk(dirname):
            for filename in filenames:
                #create complete filepath of file in directory
                filePath = os.path.join(folderName, filename)
                # Add file to zip
                zipObj.write(filePath)