数据库聊天室源码

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, message):
    cursor = connection.cursor()
    query = "INSERT INTO messages (sender, content, timestamp) VALUES (%s, %s, NOW())"
    cursor.execute(query, (message['sender'], message['content']))
    connection.commit()
    print("Message inserted successfully")

def get_messages(connection):
    cursor = connection.cursor()
    query = "SELECT * FROM messages ORDER BY timestamp DESC LIMIT 10"
    cursor.execute(query)
    records = cursor.fetchall()
    return records

if __name__ == "__main__":
    connection = create_connection()
    if connection:
        message = {'sender': 'Alice', 'content': 'Hello, Bob!'}
        insert_message(connection, message)
        messages = get_messages(connection)
        for msg in messages:
            print(f"{msg[1]}: {msg[2]} ({msg[3]})")
        connection.close()