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():
    if request.method == 'POST':
        # 解析JSON数据而不是表单数据
        data = request.get_json()
        nick_name = data['nickname']
        print(nick_name)
        # 处理注册逻辑...
        return jsonify({"success": True, "message": "注册成功"})  # 返回JSON响应
    else:
        return render_template('register.html')



@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        return render_template('login.html')
    else:
        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 == '1' and password == '1'

@app.route('/forget', methods=['GET', 'POST'])
def forget_page():
    return render_template('forget.html')
@app.route('/home')
def home():
    if 'username' in session:
        return render_template('home.html')
    else:
        return redirect("login")

@app.route('/logout')
def logout():
    # 清除session中的所有信息
    session.clear()
    # 返回一个响应，或者重定向到登录页面
    return redirect('/login')

if __name__ == '__main__':
    app.run(debug = True)