Update conc_gzip_files.py

This commit is contained in:
Sebastian Raschka 2014-04-01 15:59:49 -04:00
parent 98980fc77b
commit 8c3461624f

View File

@ -14,15 +14,15 @@ def conc_gzip_files(in_dir, out_file, append=False):
else creates a new output file.
"""
write_mode = 'w'
write_mode = 'bw'
if append:
write_mode = 'a'
write_mode = 'ba'
gzips = [os.path.join(in_dir, i) for i in os.listdir(in_dir) if i.endswith('.gz')]
with open(out_file, write_mode) as ofile:
for f in gzips:
with gzip.open(f, 'rb') as gzipf:
for line in gzipf:
ofile.write(line.decode())
ofile.write(line)
if __name__ == '__main__':
conc_gzip_files('/home/usr/my_dir', '/home/usr/test.txt')