1 2 3 4 5 6 7
import gzip def gz_wrapper(fname, mode): if fname.endswith(".gz"): return gzip.open(fname, mode) else: return open(fname, mode)