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=['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)