Python3訪問MySQL數據庫的實現步驟
前言
要實現一個簡單的IM(即時通訊)系統,支持用戶注冊、登錄和聊天記錄存儲,你可以使用Python和mysql數據庫。以下是一個基本的實現示例: 要使用MySQL創(chuàng)建表并通過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中創(chuàng)建了一個數據庫,接下來創(chuàng)建用戶和消息表。
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. 創(chuàng)建Flask API服務
一個簡單的Flask應用,提供注冊、登錄和發(fā)送消息的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開發(fā)服務器,你可以通過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" ] ]
請注意,這個示例是一個基本實現,適用于學習和測試。在生產環(huán)境中,你需要考慮更多的安全性和性能優(yōu)化,例如使用HTTPS、添加身份驗證令牌等。
到此這篇關于Python3訪問MySQL數據庫的實現步驟的文章就介紹到這了,更多相關Python3訪問MySQL內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
Python使用subprocess模塊檢測Linux用戶是否存在
本文介紹了使用Python的subprocess模塊檢測和創(chuàng)建Unix系統用戶的方法,通過調用id命令判斷用戶是否存在,若不存在則使用adduser命令創(chuàng)建用戶,適用于Ubuntu/Debian系統,感興趣的小伙伴可以了解下2026-01-01

