锦中人工智能助手

import sqlite3

from flask import Flask, request, session, redirect, url_for

app = Flask(__name__)

app.secret_key = 'your_secret_key'

# 创建数据库连接

def get_db():

return sqlite3.connect('users.db')

# 初始化数据库

def init_db():

with app.app_context():

db = get_db()

db.execute('CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, username TEXT UNIQUE, password TEXT)')

db.commit()

@app.route('/register', methods=['POST'])

def register():

username = request.form['username']

password = request.form['password']

db = get_db()

try:

db.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, password))

db.commit()

return "注册成功"

except sqlite3.IntegrityError:

return "用户名已存在"

@app.route('/login', methods=['POST'])

def login():

username = request.form['username']

password = request.form['password']

db = get_db()

user = db.execute("SELECT * FROM users WHERE username = ?", (username,)).fetchone()

if user and user[2] == password: # 假设密码未加密

session['user'] = username

return "登录成功"

else:

return "用户名或密码错误"

@app.route('/logout')

def logout():

session.pop('user', None)

return "已退出登录"

if __name__ == '__main__':

init_db()

app.run(debug=True)

from werkzeug.security import generate_password_hash, check_password_hash

# 修改注册函数

@app.route('/register', methods=['POST'])

def register():

username = request.form['username']

password = request.form['password']

hashed_pw = generate_password_hash(password)

db = get_db()

try:

db.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, hashed_pw))

db.commit()

return "注册成功"

except sqlite3.IntegrityError:

return "用户名已存在"

# 修改登录函数

@app.route('/login', methods=['POST'])

def login():

username = request.form['username']

password = request.form['password']

db = get_db()

user = db.execute("SELECT * FROM users WHERE username = ?", (username,)).fetchone()

if user and check_password_hash(user[2], password):

session['user'] = username

return "登录成功"

else:

return "用户名或密码错误"