47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from flask import Flask, request, redirect, url_for, render_template, session, jsonify
|
|
|
|
app = Flask(__name__)
|
|
app.secret_key = 'sUNiJ7QPulxrbmZD'
|
|
|
|
@app.route('/')
|
|
def index():
|
|
# 如果用户已登录，则重定向到主页；否则，重定向到登录页面
|
|
if 'username' in session:
|
|
return redirect(url_for('home'))
|
|
else:
|
|
return render_template('login.html')
|
|
@app.route('/register', methods=['GET','POST'])
|
|
def register():
|
|
return render_template('register.html')
|
|
|
|
|
|
@app.route('/login', methods=['POST'])
|
|
def login():
|
|
username = request.form['username']
|
|
password = request.form['password']
|
|
print(username,password)
|
|
# 验证用户名和密码...
|
|
if valid_login(username, password):
|
|
# 登录成功
|
|
session['username'] = username
|
|
return jsonify(success=True, message="登录成功")
|
|
else:
|
|
# 登录失败
|
|
return jsonify(success=False, message="无效的用户名或密码")
|
|
|
|
|
|
def valid_login(username, password):
|
|
# 这里应该是验证用户名和密码的逻辑，比如查询数据库等等
|
|
# 假设用户名是admin且密码是secret
|
|
return username == 'admin' and password == '1'
|
|
|
|
@app.route('/home')
|
|
def home():
|
|
if 'username' in session:
|
|
return '你已登录，欢迎, %s!' % session['username']
|
|
else:
|
|
return redirect(url_for('login')) # 如果用户未登录，重定向到登录页面
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True)
|