构建简易的消息管理系统

大家好，今天我要跟你们分享的是如何用Python和SQLite数据库来创建一个简单的消息管理系统。这个系统能帮助你更好地管理和查询你的信息。

准备工作

首先，我们需要安装Python和SQLite。如果你已经安装了Python，那你就已经拥有了SQLite。我们还需要安装一个Python的SQLite扩展库，叫做sqlite3。这可以通过pip命令轻松完成：

pip install sqlite3

创建数据库表

接下来，让我们打开一个新的Python文件，命名为message_manager.py。在文件中，我们首先需要连接到SQLite数据库并创建一个表来存储我们的消息。这里是一个简单的示例：

import sqlite3

# 连接到SQLite数据库
conn = sqlite3.connect('messages.db')
cursor = conn.cursor()

# 创建表
cursor.execute('''
CREATE TABLE IF NOT EXISTS messages (
    id INTEGER PRIMARY KEY,
    sender TEXT NOT NULL,
    receiver TEXT NOT NULL,
    message TEXT NOT NULL,
    timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
)
''')

conn.commit()
conn.close()
        

添加新消息

现在我们有了一个可以存储消息的数据库表，我们可以开始向里面添加新的消息了。下面是一个函数，用于向数据库中插入一条新的消息：

def add_message(sender, receiver, message):
    conn = sqlite3.connect('messages.db')
    cursor = conn.cursor()
    cursor.execute("INSERT INTO messages (sender, receiver, message) VALUES (?, ?, ?)", (sender, receiver, message))
    conn.commit()
    conn.close()
        

查询消息

最后，我们要添加一个查询消息的功能。这个函数可以根据发送者或接收者的名字来获取所有相关的信息。

def get_messages(user):
    conn = sqlite3.connect('messages.db')
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM messages WHERE sender=? OR receiver=?", (user, user))
    messages = cursor.fetchall()
    conn.close()
    return messages
        

好了，这就是全部的内容。现在你可以尝试运行这些代码，并根据自己的需求进行调整和扩展。希望这篇文章对你有所帮助！