张工：嘿，李工，最近我们被分配了一个任务——为南昌地区的数据中台系统开发登录功能。你觉得我们应该从哪里开始呢？
李工：首先得有一个数据库来存储用户信息吧。我们可以使用SQLite作为我们的本地数据库。
张工：好主意！那接下来怎么实现登录验证呢？
李工：我们可以创建一个简单的Python脚本来处理登录请求。首先需要导入必要的库。
import sqlite3
from flask import Flask, request, jsonify
李工：然后我们需要连接到SQLite数据库并检查用户的凭据。
app = Flask(__name__)
DATABASE = 'users.db'
def get_db_connection():
conn = sqlite3.connect(DATABASE)
conn.row_factory = sqlite3.Row
return conn
@app.route('/login', methods=['POST'])
def login():
data = request.get_json()
username = data['username']
password = data['password']
conn = get_db_connection()
user = conn.execute('SELECT * FROM users WHERE username = ?', (username,)).fetchone()
if user and user['password'] == password:
return jsonify({'message': 'Login successful'}), 200
else:
return jsonify({'message': 'Invalid credentials'}), 401
张工：这个代码看起来不错！但是我们还需要确保安全性，比如密码加密。
李工：是的，我们可以使用bcrypt库来加密密码。
import bcrypt
def hash_password(password):
salt = bcrypt.gensalt()
hashed = bcrypt.hashpw(password.encode('utf-8'), salt)
return hashed
def verify_password(stored_password, provided_password):
return bcrypt.checkpw(provided_password.encode('utf-8'), stored_password)
张工：太好了，这样我们就有了一个基本的数据中台登录功能了！接下来只需要测试一下。
李工：没错，记得测试各种情况，包括正确的用户名和密码，错误的用户名或密码等。
]]>