(pycompress.py) A file is compressed by "tar.bz2", "tar.gz" or "zip".

pycompress.py

#!/usr/bin/env python
# Author: Iyori Komiyama
# Contact: [email protected]
# site: http://hazimarino.blogspot.com/
"""\
A file is compressed by "tar.bz2", "tar.gz" or "zip". 
"""
import os
import tarfile
import zipfile
from fnmatch import fnmatch

exts = ['*.lzh', '*.zip', '*.rar', '*.gz', '*.bz2', '*.xz', '*.?[gbx]z']

def pycompress(cmpfile, types="bz2", min_size=16):
    if os.stat(cmpfile).st_size > min_size:
        if not any(fnmatch(cmpfile.lower(), p) for p in exts):
            if types == "zip":
                zp = zipfile.ZipFile("{0}.zip".format(cmpfile),
                                     'w', zipfile.ZIP_DEFLATED)
                zp.write(cmpfile)
                zp.close()
            else:
                tar = tarfile.open('{0}.tar.{1}'.format(cmpfile, types),
                                   'w:{0}'.format(types))
                tar.add(cmpfile)
                tar.close()
        else:
            print("It is already a data compressed file.")
    else:
        print("The file is too small. ")

if __name__ == '__main__':
    import sys
    try:
        if sys.argv[2] == "bz2":
            pycompress(sys.argv[1], 'bz2')
        elif sys.argv[2] == "gz":
            pycompress(sys.argv[1], 'gz')
        elif sys.argv[2] == "zip":
            pycompress(sys.argv[1], 'zip')
        else:
            raise
    except (IndexError, TypeError):
        print("Please input appropriate type (bz2, gz, zip). ")