| from app import app
|
| from flask import render_template, request, jsonify
|
| import uuid
|
| import utils.globals as globals
|
| from utils.globals import *
|
| from utils.tools import *
|
| from werkzeug.security import generate_password_hash
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| @app.route('/user')
|
| @admin_required
|
| def user_management():
|
| return render_template('user_management.html')
|
|
|
|
|
|
|
|
|
| @app.route('/api/users', methods=['GET'])
|
| @admin_required
|
| def get_users():
|
|
|
| return jsonify([{k: v for k, v in user.items() if k != 'password' and k != 'bind_token'} for user in globals.users])
|
|
|
|
|
| @app.route('/api/users', methods=['POST'])
|
| @admin_required
|
| def create_user():
|
| data = request.get_json()
|
|
|
|
|
| if any(user['username'] == data['username'] for user in globals.users):
|
| return jsonify({'success': False, 'message': '用户名已存在'}), 400
|
|
|
| new_user = {
|
| 'id': str(uuid.uuid4()),
|
| 'username': data['username'],
|
| 'password': generate_password_hash(data['password']),
|
| 'role': data['role'],
|
| 'bind_token': '',
|
| 'bind_email': '',
|
| 'expiration_time': data['expiration_time'],
|
| 'bind_claude_token': '',
|
| 'bind_claude_email': '',
|
| 'claude_expiration_time': data['claude_expiration_time']
|
| }
|
|
|
| globals.users.append(new_user)
|
| save_users(globals.users)
|
|
|
| return jsonify({'success': True, 'message': '用户创建成功'})
|
|
|
|
|
| @app.route('/api/users/<user_id>', methods=['PUT'])
|
| @admin_required
|
| def update_user(user_id):
|
| data = request.get_json()
|
|
|
| user_index = next((i for i, user in enumerate(globals.users) if user['id'] == user_id), None)
|
| if user_index is None:
|
| return jsonify({'success': False, 'message': '用户不存在'}), 404
|
|
|
|
|
| if any(user['username'] == data['username'] and user['id'] != user_id for user in globals.users):
|
| return jsonify({'success': False, 'message': '用户名已存在'}), 400
|
|
|
|
|
| globals.users[user_index]['username'] = data['username']
|
| globals.users[user_index]['role'] = data['role']
|
|
|
|
|
| if data.get('password'):
|
| globals.users[user_index]['password'] = generate_password_hash(data['password'])
|
|
|
| globals.users[user_index]['expiration_time'] = data['expiration_time']
|
| globals.users[user_index]['claude_expiration_time'] = data['claude_expiration_time']
|
|
|
| save_users(globals.users)
|
| return jsonify({'success': True, 'message': '用户更新成功'})
|
|
|
|
|
| @app.route('/api/bind/<user_id>', methods=['PUT'])
|
| @admin_required
|
| def bind_account(user_id):
|
| data = request.get_json()
|
| user_index = next((i for i, user in enumerate(globals.users) if user['id'] == user_id), None)
|
| token_index = next((i for i, token in enumerate(globals.chatToken) if token['email'] == data['email']), None)
|
| res = set_seedmap(user_id,globals.chatToken[token_index]['access_token'])
|
| if res == 200:
|
| globals.users[user_index]['bind_email'] = data['email']
|
| globals.users[user_index]['bind_token'] = globals.chatToken[token_index]['access_token']
|
| save_users(globals.users)
|
| return jsonify({'success': True, 'message': '账号绑定成功'})
|
| else:
|
| return jsonify({'success': False, 'message': '账号绑定失败'})
|
|
|
|
|
| @app.route('/api/del_bind/<user_id>', methods=['DELETE'])
|
| @admin_required
|
| def del_bind_account(user_id):
|
| res = del_seedmap(user_id)
|
| if res == 200:
|
| user_index = next((i for i, user in enumerate(globals.users) if user['id'] == user_id), None)
|
| globals.users[user_index]['bind_email'] = ''
|
| globals.users[user_index]['bind_token'] = ''
|
| save_users(globals.users)
|
| return jsonify({'success': True, 'message': '账号解绑成功'})
|
| else:
|
| return jsonify({'success': False, 'message': '账号解绑失败'})
|
|
|
|
|
|
|
| @app.route('/api/all_email', methods=['GET'])
|
| @admin_required
|
| def all_email():
|
|
|
| return jsonify([token['email'] for token in globals.chatToken if 'email' in token])
|
|
|
|
|
| @app.route('/api/bindClaude/<user_id>', methods=['PUT'])
|
| @admin_required
|
| def bind_claude_account(user_id):
|
| data = request.get_json()
|
| user_index = next((i for i, user in enumerate(globals.users) if user['id'] == user_id), None)
|
| token_index = next((i for i, token in enumerate(globals.cluadeToken) if token['email'] == data['email']), None)
|
| globals.users[user_index]['bind_claude_email'] = data['email']
|
| globals.users[user_index]['bind_claude_token'] = globals.cluadeToken[token_index]['skToken']
|
| save_users(globals.users)
|
| return jsonify({'success': True, 'message': '账号绑定成功'})
|
|
|
|
|
| @app.route('/api/del_bindClaude/<user_id>', methods=['DELETE'])
|
| @admin_required
|
| def del_bind_claude_account(user_id):
|
|
|
| user_index = next((i for i, user in enumerate(globals.users) if user['id'] == user_id), None)
|
| globals.users[user_index]['bind_claude_email'] = ''
|
| globals.users[user_index]['bind_claude_token'] = ''
|
| save_users(globals.users)
|
| return jsonify({'success': True, 'message': '账号解绑成功'})
|
|
|
|
|
|
|
| @app.route('/api/all_claude_email', methods=['GET'])
|
| @admin_required
|
| def all_claude_email():
|
|
|
| return jsonify([token['email'] for token in globals.cluadeToken if 'email' in token])
|
|
|
|
|
| @app.route('/api/users/<user_id>', methods=['DELETE'])
|
| @admin_required
|
| def delete_user(user_id):
|
|
|
|
|
| updated_users = [user for user in globals.users if user['id'] != user_id]
|
|
|
| user = next((user for user in globals.users if user['id'] == user_id), None)
|
|
|
| if len(updated_users) == len(globals.users):
|
| return jsonify({'success': False, 'message': '用户不存在'}), 404
|
|
|
| globals.users = updated_users
|
| save_users(globals.users)
|
| if user['bind_token'] != '':
|
| del_seedmap(user_id)
|
| return jsonify({'success': True, 'message': '用户删除成功'}) |
