由于提供的代码已经非常接近完整,以下是一个核心函数的示例,展示了如何使用Flask和MySQL创建一个简单的用户注册功能:
from flask import Flask, request, jsonify
import mysql.connector
app = Flask(__name__)
# 连接到MySQL数据库
db_connection = mysql.connector.connect(
host="localhost",
user="your_username",
password="your_password",
database="your_database"
)
cursor = db_connection.cursor()
@app.route('/register', methods=['POST'])
def register():
data = request.get_json()
if not data:
return jsonify({'message': 'No input data provided'}), 400
try:
cursor.execute("INSERT INTO users (username, password) VALUES (%s, %s)",
(data['username'], data['password']))
db_connection.commit()
user_id = cursor.lastrowid
return jsonify({'user_id': user_id, 'message': 'User registered successfully'}), 201
except mysql.connector.Error as error:
print("Failed to insert record into table: {}".format(error))
return jsonify({'message': 'User already exists'}), 400
if __name__ == '__main__':
app.run(debug=True)
在这个示例中,我们创建了一个名为register
的路由,它接受POST请求并尝试将用户信息插入到名为users
的数据库表中。如果插入成功,它将返回新用户的ID和一个成功消息,如果用户已经存在,它将返回一个错误消息。这个简单的示例展示了如何在Flask应用程序中处理数据库操作,并且如何通过API接收和处理数据。