域名登陆管理平台

from flask import Flask, request, redirect, url_for, session
import hashlib

app = Flask(__name__)
app.secret_key = 'your_secret_key'

# 模拟用户数据库
users = {
    'user1': {'password': 'hashed_password_1', 'roles': ['admin']},
    'user2': {'password': 'hashed_password_2', 'roles': ['user']}
}

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        username = request.form['username']
        password = request.form['password']
        hashed_password = hashlib.sha256(password.encode()).hexdigest()
        if username in users and users[username]['password'] == hashed_password:
            session['username'] = username
            return redirect(url_for('dashboard'))
        else:
            return 'Invalid credentials'
    return '''
        <form method="post">
            <input type="text" name="username" placeholder="Username">
            <input type="password" name="password" placeholder="Password">
            <button type="submit">Login</button>
        </form>
    '''

@app.route('/dashboard')
def dashboard():
    if 'username' not in session:
        return redirect(url_for('login'))
    username = session['username']
    user_roles = users[username]['roles']
    return f'Welcome, {username}! Your roles: {", ".join(user_roles)}'

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