Extract gzipped files

extract_gz.py

import gzip
import os

root_folder = "mock-data"

for root, dirs, files in os.walk(root_folder):
    # if file is a .csv.gzip, extract it
    for file in files:
        if file.endswith(".gz"):
            file_path = os.path.join(root, file)
            with gzip.open(file_path, "rb") as f_in:
                with open(file_path[:-3], "wb") as f_out:
                    f_out.write(f_in.read())
            os.remove(file_path)
            print(f"Extracted {file_path}")