Improve .tokens output

This commit is contained in:
Zed 2023-08-29 23:45:18 +02:00
parent 7630f57f17
commit 4ccf350dc7

View file

@ -1,5 +1,5 @@
#i hate begging for this too em SPDX-License-Identifier: AGPL-3.0-only #SPDX-License-Identifier: AGPL-3.0-only
import asyncdispatch, times, json, random, strutils, tables import asyncdispatch, times, json, random, strutils, tables, sets
import types import types
# max requests at a time per account to avoid race conditions # max requests at a time per account to avoid race conditions
@ -19,14 +19,16 @@ proc getPoolJson*(): JsonNode =
list = newJObject() list = newJObject()
totalReqs = 0 totalReqs = 0
totalPending = 0 totalPending = 0
totalLimited = 0 limited: HashSet[string]
reqsPerApi: Table[string, int] reqsPerApi: Table[string, int]
let now = epochTime().int let now = epochTime().int
for account in accountPool: for account in accountPool:
totalPending.inc(account.pending) totalPending.inc(account.pending)
list[account.id] = %*{
var includeAccount = false
let accountJson = %*{
"apis": newJObject(), "apis": newJObject(),
"pending": account.pending, "pending": account.pending,
} }
@ -36,17 +38,18 @@ proc getPoolJson*(): JsonNode =
apiStatus = account.apis[api] apiStatus = account.apis[api]
obj = %*{} obj = %*{}
if apiStatus.limited:
obj["limited"] = %true
inc totalLimited
if apiStatus.reset > now.int: if apiStatus.reset > now.int:
obj["remaining"] = %apiStatus.remaining obj["remaining"] = %apiStatus.remaining
if "remaining" notin obj and not apiStatus.limited: if "remaining" notin obj and not apiStatus.limited:
continue continue
list[account.id]["apis"][$api] = obj if apiStatus.limited:
obj["limited"] = %true
limited.incl account.id
accountJson{"apis", $api} = obj
includeAccount = true
let let
maxReqs = maxReqs =
@ -64,9 +67,12 @@ proc getPoolJson*(): JsonNode =
reqsPerApi[$api] = reqsPerApi.getOrDefault($api, 0) + reqs reqsPerApi[$api] = reqsPerApi.getOrDefault($api, 0) + reqs
totalReqs.inc(reqs) totalReqs.inc(reqs)
if includeAccount:
list[account.id] = accountJson
return %*{ return %*{
"amount": accountPool.len, "amount": accountPool.len,
"limited": totalLimited, "limited": limited.card,
"requests": totalReqs, "requests": totalReqs,
"pending": totalPending, "pending": totalPending,
"apis": reqsPerApi, "apis": reqsPerApi,