import csv
import os

import mysql.connector


def main(objects_to_copy):

    connection = mysql.connector.connect(
        host=os.environ.get('host'),
        user=os.environ.get('user'),
        password=os.environ.get('password'),
        database=os.environ.get('database')
    )

    for to_copy in objects_to_copy:
        cursor = connection.cursor()
        query = (to_copy['sql'])
        cursor.execute(query)
        data = cursor.fetchall()
        with open(to_copy['filename'], 'w') as f:
            csv_file = csv.writer(f)
            csv_file.writerow(cursor.column_names)
            csv_file.writerows(data)
        cursor.close()
    connection.close()

if __name__ == '__main__':
    objects_to_copy = [
        {
            'sql': 'select song_id, song_name, date_created, date_modified, extension, acrcloud, acrcloud_rights, audiblemagic, upc, tuid, isrc, path, label, track_name, release_name, artist from songs',
            'filename': 'songs.csv'
        },
        {
            'sql': 'select id, song_id, acrid, label, title, artist, isrc, upc, album, release_date from acrcloud_matches',
            'filename': 'acrcloud_matches.csv'
        },
    ]
    main(objects_to_copy)