32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from flask import Flask, render_template
|
|
from flask import flash, redirect, session, request, url_for
|
|
|
|
app = Flask(__name__)
|
|
app.secret_key = 'Your_secret_key&^52@!' # 设置secret_key
|
|
|
|
|
|
@app.route('/home')
|
|
@app.route('/')
|
|
def home_page():
|
|
username = session.get('username')
|
|
# 判断session是否存储username的数据
|
|
if 'username' in session:
|
|
return render_template('hw_t1_home_page.html', username=username)
|
|
return redirect(url_for('login')) # 重定向到login页面
|
|
|
|
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
if request.method == 'POST':
|
|
if request.form['username'] != 'admin' or \
|
|
request.form['password'] != '123':
|
|
flash('用户名或密码错误', category='error')
|
|
else:
|
|
session['username'] = request.form['username']
|
|
session['password'] = request.form['password']
|
|
flash('恭喜您，登录成功', category='info')
|
|
return redirect(url_for('home_page'))
|
|
return render_template('hw_t1_login.html')
|
|
|
|
if __name__ == '__main__':
|
|
app.run(port=80) |