def for_writing(file_name, mode="w", gzip_mode="wb"):
assert mode in ["w", "wb", "a", "ab"]
assert gzip_mode in ["w", "wb", "a", "ab"]
file_name = os.path.expanduser(file_name)
if (file_name.endswith(".gz")):
return gzip_open(file_name=file_name, mode=gzip_mode)
try:
return open(file_name, mode)
except IOError, e:
raise IOError(
"Cannot open file for writing: %s\n" % show_string(file_name)
+ " "+str(e))
def for_reading(file_name, mode="r", gzip_mode="rb"):
assert mode in ["r", "rb"]
assert gzip_mode in ["r", "rb"]
file_name = os.path.expanduser(file_name)
if (file_name.endswith(".gz")):
return gzip_open(file_name=file_name, mode=gzip_mode)
if (file_name.endswith(".Z")):
return StringIO(
easy_run.fully_buffered(
command='gunzip -c "%s"' % escape_sh_double_quoted(file_name),
stdout_splitlines=False).raise_if_errors().stdout_buffer)
if file_name.endswith('.bz2'):
return bz2_open(file_name=file_name, mode=mode)
try:
return open(file_name, mode)
except IOError, e:
raise IOError("Cannot open file for reading: %s\n" %
show_string(file_name) + " " + str(e))
def for_reading(file_name, mode="r", gzip_mode="rb"):
assert mode in ["r", "rb"]
assert gzip_mode in ["r", "rb"]
file_name = os.path.expanduser(file_name)
if (file_name.endswith(".gz")):
return gzip_open(file_name=file_name, mode=gzip_mode)
if (file_name.endswith(".Z")):
return StringIO(easy_run.fully_buffered(
command='gunzip -c "%s"' % escape_sh_double_quoted(file_name),
stdout_splitlines=False).raise_if_errors().stdout_buffer)
if file_name.endswith('.bz2'):
return bz2_open(file_name=file_name, mode=mode)
try:
return open(file_name, mode)
except IOError, e:
raise IOError(
"Cannot open file for reading: %s\n" % show_string(file_name)
+ " "+str(e))
