def save_to_csv()

in python/fb_ads_library_api_operators.py [0:0]

• 30 lines of code
• 11 McCabe index (conditional complexity)

def save_to_csv(generator_ad_archives, args, fields, is_verbose=False):
    """
    Save all retrieved ad_archives to the output file. Each ad_archive will be
    stored as a row in the CSV
    """
    if len(args) != 1:
        raise Exception("save_to_csv action takes 1 argument: output_file")

    delimiter = ","
    total_count = 0
    output = fields + "\n"
    output_file = args[0]

    for ad_archives in generator_ad_archives:
        total_count += len(ad_archives)
        if is_verbose:
            print("Items processed: %d" % total_count)
        for ad_archive in ad_archives:
            for field in list(fields.split(delimiter)):
                if field in ad_archive:
                    value = ad_archive[field]
                    if (type(value) == list and type(value[0]) == dict) or type(
                        value
                    ) == dict:
                        value = json.dumps(value)
                    elif type(value) == list:
                        value = delimiter.join(value)
                    output += (
                        '"' + value.replace("\n", "").replace('"', "") + '"' + delimiter
                    )
                else:
                    output += delimiter
            output = output.rstrip(",") + "\n"

    with open(output_file, "w") as csvfile:
        csvfile.write(output)

    print("Successfully wrote data to file: %s" % output_file)