open file depending on file extension: supports .bz2, .zip, .tar, .gz

fancy_open.py

def fancy_open(filename, mode='r'):
    
    if args.file.endswith('.bz2'):
        from bz2 import BZ2File
        input_file = BZ2File(filename, mode)
    elif args.file.endswith('.zip'):
        from zipfile import ZipFile
        input_file = ZipFile(filename, mode)
    elif args.file.endswith('.tar'):
        from tarfile import TarFile
        input_file = TarFile(filename, mode)
    elif args.file.endswith('.gz'):
        from gzip import GzipFile
        input_file = GzipFile(filename, mode)
    else:
        input_file = open(filename, mode)
    
    return input_file