在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())

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

Answer 2

如果您正在处理数据，我强烈建议您研究 Pandas。此外，对于这样的论坛，最好省略__main__ 支票。这是一个未经测试的代码，它可以完成您在 Pandas 中尝试做的事情。

import pandas

connection_string = "'mysql://root:1232580@localhost/db'"
df = pandas.read_sql_query('EXEC AccessionsBycauseOfAffliction_year', connection_string)
df.to_csv('AccessionsBycauseOfAffliction_year.csv')

Answer 3

如果你能找到SP，调用你可以使用下面的代码 请提供更多您想要代码仅用于将结果存储在 csv 中的格式？

from mysql.connector import MySQLConnection, Error
from python_mysql_dbconfig import read_db_config


    def call_find_by_isbn():
        try:
            db_config = read_db_config()
            conn = MySQLConnection(**db_config)
            cursor = conn.cursor()

            args = ['1236400967773', 0]
            result_args = cursor.callproc('AccessionsBycauseOfAffliction_year', args)

            print(result_args[1])

        except Error as e:
            print(e)

        finally:
            cursor.close()
            conn.close()

    if __name__ == '__main__':
        call_find_by_isbn()