File: //home/arjun/projects/unlimited-leads/Unlimited-Leads-Be/utils/csv_splitter.py
import os
import csv
import zipfile
def to_csv(input_data, header, output_file, filename):
csv_filepath = os.path.join(output_file, f"{filename}.csv")
with open(csv_filepath, mode="w", encoding="utf-8") as outfile:
writer = csv.writer(outfile)
writer.writerow(header)
writer.writerows(input_data)
return csv_filepath
def zip_csv_files(output_dir, zip_file):
with zipfile.ZipFile(
zip_file, "w", zipfile.ZIP_DEFLATED, allowZip64=True, compresslevel=6
) as zipf:
for root, _, files in os.walk(output_dir):
for file in files:
zipf.write(
os.path.join(root, file),
os.path.relpath(os.path.join(root, file), output_dir),
)