import gzip

def gz_wrapper(fname, mode):
    if fname.endswith(".gz"):
        return gzip.open(fname, mode)
    else:
        return open(fname, mode)