dekun
109 lines
3.0 KiB
Python
109 lines
3.0 KiB
Python
"""
|
|
多账户 API 密钥管理工具 — Flask 后端
|
|
端口: 5200 (0.0.0.0 局域网可访问) | 数据: data.json
|
|
"""
|
|
import json
|
|
import uuid
|
|
from pathlib import Path
|
|
|
|
from flask import Flask, jsonify, request, send_from_directory
|
|
|
|
app = Flask(__name__)
|
|
|
|
BASE_DIR = Path(__file__).resolve().parent
|
|
DATA_FILE = BASE_DIR / "data.json"
|
|
VALID_EXCHANGES = frozenset({"binance", "okx", "gate"})
|
|
|
|
|
|
def load_accounts():
|
|
if not DATA_FILE.exists():
|
|
return []
|
|
try:
|
|
with open(DATA_FILE, "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
if isinstance(data, list):
|
|
return data
|
|
return []
|
|
except (json.JSONDecodeError, OSError):
|
|
return []
|
|
|
|
|
|
def save_accounts(accounts):
|
|
with open(DATA_FILE, "w", encoding="utf-8") as f:
|
|
json.dump(accounts, f, ensure_ascii=False, indent=2)
|
|
|
|
|
|
def validate_account_payload(payload):
|
|
exchange = (payload.get("exchange") or "").strip().lower()
|
|
username = (payload.get("username") or "").strip()
|
|
api_key = (payload.get("api_key") or "").strip()
|
|
api_secret = (payload.get("api_secret") or "").strip()
|
|
|
|
if exchange not in VALID_EXCHANGES:
|
|
return None, "请选择交易所：binance / okx / gate"
|
|
if not username:
|
|
return None, "账户名称不能为空"
|
|
if not api_key:
|
|
return None, "API Key 不能为空"
|
|
if not api_secret:
|
|
return None, "API Secret 不能为空"
|
|
|
|
account = {
|
|
"exchange": exchange,
|
|
"username": username,
|
|
"api_key": api_key,
|
|
"api_secret": api_secret,
|
|
}
|
|
|
|
if exchange == "okx":
|
|
password = (payload.get("password") or "").strip()
|
|
if not password:
|
|
return None, "OKX 密码（Passphrase）不能为空"
|
|
account["password"] = password
|
|
|
|
return account, None
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return send_from_directory(BASE_DIR, "index.html")
|
|
|
|
|
|
@app.route("/api/accounts", methods=["GET"])
|
|
def list_accounts():
|
|
accounts = load_accounts()
|
|
exchange = (request.args.get("exchange") or "").strip().lower()
|
|
if exchange:
|
|
if exchange not in VALID_EXCHANGES:
|
|
return jsonify({"error": "无效交易所"}), 400
|
|
accounts = [a for a in accounts if a.get("exchange") == exchange]
|
|
return jsonify(accounts)
|
|
|
|
|
|
@app.route("/api/accounts", methods=["POST"])
|
|
def create_account():
|
|
body = request.get_json(silent=True) or {}
|
|
account, err = validate_account_payload(body)
|
|
if err:
|
|
return jsonify({"error": err}), 400
|
|
|
|
accounts = load_accounts()
|
|
account["id"] = str(uuid.uuid4())
|
|
accounts.append(account)
|
|
save_accounts(accounts)
|
|
return jsonify(account), 201
|
|
|
|
|
|
@app.route("/api/accounts/<account_id>", methods=["DELETE"])
|
|
def delete_account(account_id):
|
|
accounts = load_accounts()
|
|
new_accounts = [a for a in accounts if a.get("id") != account_id]
|
|
if len(new_accounts) == len(accounts):
|
|
return jsonify({"error": "账户不存在"}), 404
|
|
save_accounts(new_accounts)
|
|
return jsonify({"ok": True})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=5200, debug=False)
|