如何重写此函数以在数据库中存储数据？

import mysql.connector

def store_data_in_database(data):
    try:
        # 连接数据库
        conn = mysql.connector.connect(
            host='localhost',
            user='username',
            password='password',
            database='database_name'
        )
        
        # 创建游标
        cursor = conn.cursor()
        
        # 创建数据表（如果需要）
        create_table_query = "CREATE TABLE IF NOT EXISTS table_name (id INT AUTO_INCREMENT PRIMARY KEY, data VARCHAR(255))"
        cursor.execute(create_table_query)
        
        # 插入数据
        insert_data_query = "INSERT INTO table_name (data) VALUES (%s)"
        cursor.execute(insert_data_query, (data,))
        
        # 提交事务
        conn.commit()
        
        # 关闭游标和连接
        cursor.close()
        conn.close()
        
        return "Data stored successfully in the database."
    except mysql.connector.Error as error:
        return "Error storing data in the database: {}".format(error)