python - python调用mysql存储过程并将结果写入csv文件

Question

我是 python 的新手，我试图在 python 中定义一个函数，该函数使用 mysql.connector 在 python 中调用存储过程并将结果写入 csv 文件。我已经定义了一个函数来调用存储过程并打印结果。但无法弄清楚如何将我的结果写入 csv 文件而不是打印。尝试了一些在网络上找到的方法，但没有带我到任何地方

到目前为止我的代码是:

def call_accessionsByaffliction():
    try:
        conn = mysql.connector.connect( host="localhost", user="root", 
password="1232580", database="db")
        cur = conn.cursor()
        cur.callproc('AccessionsBycauseOfAffliction_year')
        for result in cur.stored_results():
            print(result.fetchall())
    except Error as e:
        print(e)
    finally:
        cur.close()
        conn.close() 

和

if __name__ == '__main__':
    call_accessionsByaffliction()

有什么帮助吗？非常感谢

Answer 1

import csv

def cs_writer(string_chunk):
    # define desired output file name      
    file_name = "output.csv"
    try:
        #open file in write mode and hold object 
        out_file = open(file_name, "wt")

        # create csv write object
        writer = csv.writer(out_file, delimiter=',', quotechar='"', quoting=csv.QUOTE_ALL)
        for line in string_chunk:
            writer.writerow(line)
    except Exception as e:
        print(e)
    finally:
        # close file
        out_file.close()

def call_accessionsByaffliction():
    try:
        conn = mysql.connector.connect( host="localhost", user="root", password="1232580", database="db")
        cur = conn.cursor()

        # execute stored procedure
        cur.callproc('AccessionsBycauseOfAffliction_year')

        # fetch all results or use fetchone()
        for res in cur.stored_results():
            result = res.fetchall()

        print(result)
        return result
    except Error as e:
        print(e)
    finally:
        cur.close()
        conn.close() 

if __name__ == '__main__':
    cs_writer(call_accessionsByaffliction())

我希望这能回答您的问题。