Open a file that may be compressed in python.

open_with_zip.py

import gzip
import zipfile

def open_with_zip(filepath):
  """Open a file with .gzip, .zip, or no compression.

  Returns:
    [*str] read filepointer of uncompressed file stream
  """
  ext = filepath.lower().rpartition('.')[-1]
  if ext == "gzip":
    return gzip.open(filepath, "rb")
  elif ext == "zip":
    z = zipfile.ZipFile(filepath, "r")
    # Return first file in zip.
    return z.open(z.namelist()[0], "r")
  else:
    return open(filepath, "rb")