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)