HEX
Server: Apache/2.4.52 (Ubuntu)
System: Linux spn-python 5.15.0-89-generic #99-Ubuntu SMP Mon Oct 30 20:42:41 UTC 2023 x86_64
User: arjun (1000)
PHP: 8.1.2-1ubuntu2.20
Disabled: NONE
Upload Files
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),
                )