migrations.py 文件源码

def create_csv(csv_filename, header_list, sql_query_string):
    """Transfer records from database query to a new .csv file.

    Keyword arguments:
    csv_filename -- String with a '.csv' suffix
    header_list -- List of strings (as CSV header names)
    sql_query_string -- Passed to match header_list format
    """

    with open(csv_filename, 'w', newline='\n') as csvfile:
        csv_writer = csv.writer(csvfile, quotechar="'", quoting=csv.QUOTE_MINIMAL)
        csv_writer.writerow(header_list)

        CUR.execute(sql_query_string)
        records = CUR.fetchall()
        CONN.commit()
        for entry in records:
            csv_writer.writerow(entry)