python將mysql數據導出到csv文本

以下代碼是將mysql數據庫數據導入到csv文件

import pymysql
import csv

def from_mysql_get_all_info():
    conn = pymysql.connect(
        host='xxxxxx',
        port=3306,
        user='xxxx',
        db=xx',
        password='xxxxx',
        charset='utf8mb4'
    cursor = conn.cursor()
    sql = 'select * from py.tencent_news'
    cursor.execute(sql.encode('utf-8'))
    data = cursor.fetchall()
    conn.close()
    return data


def write_csv():
    data = from_mysql_get_all_info()
    filename = 'data/corpus.csv'
    with open(filename,mode='w',encoding='utf-8') as f:
        write = csv.writer(f,dialect='excel')
        for item in data:
            write.writerow(item)

write_csv()