mysql数据库聊天室

import mysql.connector
from mysql.connector import Error

def create_connection():
    try:
        connection = mysql.connector.connect(host='localhost',
                                             database='chatroom',
                                             user='user',
                                             password='password')
        if connection.is_connected():
            print("Connected to MySQL database")
        return connection
    except Error as e:
        print(f"Error while connecting to MySQL: {e}")
        return None

def insert_message(connection, sender, message):
    cursor = connection.cursor()
    query = "INSERT INTO messages (sender, message) VALUES (%s, %s)"
    cursor.execute(query, (sender, message))
    connection.commit()
    print("Message inserted successfully")

def get_messages(connection):
    cursor = connection.cursor()
    query = "SELECT sender, message FROM messages ORDER BY id DESC LIMIT 10"
    cursor.execute(query)
    records = cursor.fetchall()
    for record in records:
        print(f"{record[0]}: {record[1]}")

if __name__ == "__main__":
    connection = create_connection()
    if connection:
        insert_message(connection, "Alice", "Hello, everyone!")
        get_messages(connection)
        connection.close()