File size: 2,679 Bytes
bab9c3b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | from app import app
import json
from flask import request, jsonify
import utils.globals as globals
from utils.globals import *
from utils.tools import *
# 加载Claude Token
@app.route('/get_Claude')
@admin_required
def get_Claude():
try:
return jsonify(globals.cluadeToken), 200
except json.JSONDecodeError:
return jsonify({"error": "Invalid JSON in tokens.json"}), 500
except Exception as e:
return jsonify({"error": str(e)}), 500
# 添加新账号
@app.route('/api/Claude', methods=['POST'])
@admin_required
def create_Claude():
data = request.get_json()
# 检查账号是否已存在
if any(token['email'] == data['email'] for token in globals.cluadeToken):
return jsonify({'success': False, 'message': '该账号已存在'}), 400
new_token = {
'email': data['email'],
'skToken': data['SkToken'],
'status': True,
'type':"/static/claude.png",
'PLUS': data['PLUS']
}
globals.cluadeToken.append(new_token)
save_cltoken(globals.cluadeToken)
return jsonify({'success': True, 'message': '用户创建成功'})
# 更新账号信息
@app.route('/api/Claude/<email>', methods=['PUT'])
@admin_required
def update_Claude(email):
data = request.get_json()
token_index = next((i for i, token in enumerate(globals.cluadeToken) if token['email'] == email), None)
if token_index is None:
return jsonify({'success': False, 'message': '账号不存在'}), 400
# 如果提供了邮箱，则更新邮箱
if data.get('email'):
globals.cluadeToken[token_index]['email'] = data['email']
# 如果提供了ReToken，则更新ReToken
if data.get('SkToken'):
globals.cluadeToken[token_index]['skToken'] = data['SkToken']
globals.cluadeToken[token_index]['status'] = True
if data.get('PLUS'):
globals.cluadeToken[token_index]['PLUS'] = data['PLUS']
save_cltoken(globals.cluadeToken)
return jsonify({'success': True, 'message': '账号更新成功'})
# 删除用户
@app.route('/api/Claude/<email>', methods=['DELETE'])
@admin_required
def delete_Claude(email):
# 过滤掉要删除的用户
updated_email = [token for token in globals.cluadeToken if token['email'] != email]
if len(updated_email) == len(globals.cluadeToken):
return jsonify({'success': False, 'message': '账号不存在'}), 404
globals.cluadeToken = updated_email
save_cltoken(globals.cluadeToken)
return jsonify({'success': True, 'message': '账号删除成功'}) |