Python3访问MySQL数据库的实现步骤

 更新时间:2024年11月17日 11:23:48   作者:HBLOG  
要实现一个简单的IM(即时通讯)系统,支持用户注册、登录和聊天记录存储,你可以使用Python和mysql数据库,以下是一个基本的实现步骤,并通过代码示例讲解的非常详细,需要的朋友可以参考下

前言

要实现一个简单的IM(即时通讯)系统,支持用户注册、登录和聊天记录存储,你可以使用Python和mysql数据库。以下是一个基本的实现示例: 要使用MySQL创建表并通过Python提供一个API服务,你可以使用Flask框架来实现API服务,并使用PyMySQL库来连接MySQL数据库。以下是一个基本的实现步骤:

1. 安装所需库

首先,确保你安装了Flask和PyMySQL库:

pip install flask pymysql

2. MySQL数据库设置

docker run --hostname=a5ddc3708f2e --env=MYSQL_ROOT_PASSWORD=123456 --env=MYSQL_DATABASE=jwordpress --env=TZ=Asia/Shanghai --env=LANG=en_US.UTF-8 --env=PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin --env=GOSU_VERSION=1.7 --env=MYSQL_MAJOR=5.7 --env=MYSQL_VERSION=5.7.26-1debian9 --volume=D:\IdeaProjects\Jwordpress-parent-s02-81f3dea303558f4388ef39435a52ea1cfab22904\docker\mysql\my.cnf:/etc/mysql/my.cnf:rw --volume=D:\IdeaProjects\Jwordpress-parent-s02-81f3dea303558f4388ef39435a52ea1cfab22904\docker\mysql\init-file.sql:/etc/mysql/init-file.sql:rw --volume=D:\IdeaProjects\Jwordpress-parent-s02-81f3dea303558f4388ef39435a52ea1cfab22904\docker\mysql\data:/var/lib/mysql:rw --volume=D:\IdeaProjects\Jwordpress-parent-s02-81f3dea303558f4388ef39435a52ea1cfab22904\docker\mysql\docker-entrypoint-initdb.d:/docker-entrypoint-initdb.d:rw --volume=/var/lib/mysql --network=docker_default -p 3306:3306 --restart=unless-stopped --label='com.docker.compose.config-hash=f33622a4d32e092d39a39c3dc0bd2259df09b24ad897567bcaa7f7fa0630b019' --label='com.docker.compose.container-number=1' --label='com.docker.compose.depends_on=' --label='com.docker.compose.image=sha256:a1aa4f76fab910095dfcf4011f32fbe7acdb84c46bb685a8cf0a75e7d0da8f6b' --label='com.docker.compose.oneoff=False' --label='com.docker.compose.project=docker' --label='com.docker.compose.project.config_files=D:\IdeaProjects\Jwordpress-parent-s02-81f3dea303558f4388ef39435a52ea1cfab22904\docker\docker-compose.yml' --label='com.docker.compose.project.working_dir=D:\IdeaProjects\Jwordpress-parent-s02-81f3dea303558f4388ef39435a52ea1cfab22904\docker' --label='com.docker.compose.service=mysql' --label='com.docker.compose.version=2.21.0' --runtime=runc -d registry.cn-hangzhou.aliyuncs.com/zhengqing/mysql:5.7

假设你已经在MySQL中创建了一个数据库,接下来创建用户和消息表。

CREATE DATABASE chat_db;

USE chat_db;

CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    username VARCHAR(255) UNIQUE NOT NULL,
    password VARCHAR(255) NOT NULL
);

CREATE TABLE messages (
    id INT AUTO_INCREMENT PRIMARY KEY,
    sender VARCHAR(255) NOT NULL,
    receiver VARCHAR(255) NOT NULL,
    message TEXT NOT NULL,
    timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
);

3. 创建Flask API服务

一个简单的Flask应用,提供注册、登录和发送消息的API。 详见main.py

from flask import Flask, request, jsonify
import pymysql
import bcrypt

app = Flask(__name__)

# Database connection configuration
db_config = {
    'host': 'localhost',
    'user': 'root',
    'password': '123456',
    'database': 'chat_db'
}


def get_db_connection():
    return pymysql.connect(**db_config)


@app.route('/register', methods=['POST'])
def register():
    data = request.json
    username = data['username']
    password = data['password']

    hashed = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())

    conn = get_db_connection()
    cursor = conn.cursor()

    try:
        cursor.execute("INSERT INTO users (username, password) VALUES (%s, %s)", (username, hashed))
        conn.commit()
        return jsonify({'message': 'Registration successful!'}), 201
    except pymysql.IntegrityError:
        return jsonify({'message': 'Username already exists!'}), 400
    finally:
        cursor.close()
        conn.close()


@app.route('/login', methods=['POST'])
def login():
    data = request.json
    username = data['username']
    password = data['password']

    conn = get_db_connection()
    cursor = conn.cursor()

    try:
        cursor.execute("SELECT password FROM users WHERE username = %s", (username,))
        result = cursor.fetchone()

        if result and bcrypt.checkpw(password.encode('utf-8'), result[0].encode('utf-8')):
            return jsonify({'message': 'Login successful!'}), 200
        else:
            return jsonify({'message': 'Incorrect username or password!'}), 401
    finally:
        cursor.close()
        conn.close()


@app.route('/send_message', methods=['POST'])
def send_message():
    data = request.json
    sender = data['sender']
    receiver = data['receiver']
    message = data['message']

    conn = get_db_connection()
    cursor = conn.cursor()

    cursor.execute("INSERT INTO messages (sender, receiver, message) VALUES (%s, %s, %s)", (sender, receiver, message))
    conn.commit()
    cursor.close()
    conn.close()

    return jsonify({'message': 'Message sent successfully!'}), 201


@app.route('/get_messages', methods=['GET'])
def get_messages():
    user1 = request.args.get('user1')
    user2 = request.args.get('user2')

    conn = get_db_connection()
    cursor = conn.cursor()

    cursor.execute('''SELECT sender, receiver, message, timestamp FROM messages 
                      WHERE (sender = %s AND receiver = %s) OR (sender = %s AND receiver = %s)
                      ORDER BY timestamp''', (user1, user2, user2, user1))

    messages = cursor.fetchall()
    cursor.close()
    conn.close()

    return jsonify(messages), 200


if __name__ == '__main__':
    app.run(debug=True)

4. 运行API服务

将上述代码保存为一个Python文件(例如main.py),然后运行:

python main.py

这将启动一个Flask开发服务器,你可以通过POST请求来注册和登录用户,通过GET请求来获取聊天记录。

5. 测试

post http://127.0.0.1:5000/register

{
    "username": "alice",
    "password": "password123"
}

post http://127.0.0.1:5000/register

{
    "username": "bob",
    "password": "password123"
}

post http://127.0.0.1:5000/send_message

{
    "sender": "alice",
    "receiver": "bob",
    "message": "Hello Bob!"
}

get http://127.0.0.1:5000/get_messages?user1=alice&user2=bob

[    [        "alice",        "bob",        "Hello Bob!",        "Fri, 15 Nov 2024 16:06:33 GMT"    ]
]

请注意,这个示例是一个基本实现,适用于学习和测试。在生产环境中,你需要考虑更多的安全性和性能优化,例如使用HTTPS、添加身份验证令牌等。

到此这篇关于Python3访问MySQL数据库的实现步骤的文章就介绍到这了,更多相关Python3访问MySQL内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python标准库与第三方库详解

    Python标准库与第三方库详解

    这篇文章主要介绍了Python标准库与第三方库,需要的朋友可以参考下
    2014-07-07
  • Python自动化下载Git仓库中的所有分支到本地

    Python自动化下载Git仓库中的所有分支到本地

    在日常的 Git 使用中,我们通常使用 git clone 命令克隆仓库,但这默认只会下载远程仓库的默认分支,下面我们就来看看如何使用Python自动化脚本实现下载 Git 仓库中的所有分支吧
    2025-10-10
  • OpenCV简单标准数字识别的完整实例

    OpenCV简单标准数字识别的完整实例

    这篇文章主要给大家介绍了关于OpenCV简单标准数字识别的相关资料,要通过opencv 进行数字识别离不开训练库的支持,需要对目标图片进行大量的训练,才能做到精准的识别出目标数字,需要的朋友可以参考下
    2021-09-09
  • Python如何将日志写入到数据表中

    Python如何将日志写入到数据表中

    这篇文章主要介绍了Python如何将日志写入到数据表中的实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2026-03-03
  • 利用python实现蝴蝶曲线

    利用python实现蝴蝶曲线

    这篇文章主要介绍了如何利用python实现蝴蝶曲线,自然界的很多现象都可以适当的简化用代数曲线和超越曲线来表达,下面我们就来利用去先画出小蝴蝶,需要的小伙伴可以参考一下
    2022-03-03
  • python实现把二维列表变为一维列表的方法分析

    python实现把二维列表变为一维列表的方法分析

    这篇文章主要介绍了python实现把二维列表变为一维列表的方法,结合实例形式总结分析了Python列表推导式、嵌套、循环等相关操作技巧,需要的朋友可以参考下
    2019-10-10
  • Python利用Turtle绘画简单图形

    Python利用Turtle绘画简单图形

    这篇文章主要介绍了Python利用Turtle绘画简单图形,文章围绕主题展开详细的内容介绍,具有一定的参考价值,需要的朋友可以参考一下
    2022-07-07
  • 跟老齐学Python之list和str比较

    跟老齐学Python之list和str比较

    list和str两种类型数据,有不少相似的地方,也有很大的区别。本讲对她们做个简要比较,同时也是对前面有关两者的知识复习一下,所谓“温故而知新”。
    2014-09-09
  • Python使用subprocess模块检测Linux用户是否存在

    Python使用subprocess模块检测Linux用户是否存在

    本文介绍了使用Python的subprocess模块检测和创建Unix系统用户的方法,通过调用id命令判断用户是否存在,若不存在则使用adduser命令创建用户,适用于Ubuntu/Debian系统,感兴趣的小伙伴可以了解下
    2026-01-01
  • Python 字符替换的四方法

    Python 字符替换的四方法

    本文主要介绍了Python 字符替换的四方法,主要包括replace、translate、maketrans 和正则这是四种方法,具有一定的参考价值,感兴趣的可以了解一下
    2024-01-01

最新评论