zip a folder

zip.py

# zips a folder

import os
import sys
import argparse
import zipfile

def zip(src, dst):
    fname = "%s.zip" % (dst)
    if os.path.exists(fname):
        os.remove(fname)
        print("deleted file " + fname)
    zf = zipfile.ZipFile("%s.zip" % (dst), "w", zipfile.ZIP_DEFLATED)
    abs_src = os.path.abspath(src)
    for dirname, subdirs, files in os.walk(src):
        for filename in files:
            absname = os.path.abspath(os.path.join(dirname, filename))
            arcname = absname[len(abs_src) + 1:]
            print("zipping " + os.path.join(dirname, filename) + " as " + arcname)
            zf.write(absname, arcname)
    zf.close()
    return True

def main():

    parser = argparse.ArgumentParser()
    parser.add_argument("src", help="Name of the root file")
    parser.add_argument("dst", help="Name of the zip file")
    args = parser.parse_args()

    if not zip(args.src, args.dst):
        sys.exit(1)

if __name__ == "__main__":
    main()