From 6cb1544c7de1c758aea9b38aa19e2ea31d1b9d4c Mon Sep 17 00:00:00 2001 From: zhbaor Date: Sun, 2 Feb 2025 10:16:40 +0800 Subject: [PATCH] =?UTF-8?q?/stat=E7=BB=9F=E8=AE=A1=E6=95=B0=E6=8D=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 22 ++++++---------------- 1 file changed, 6 insertions(+), 16 deletions(-) diff --git a/main.py b/main.py index e13f452..15a25c3 100644 --- a/main.py +++ b/main.py @@ -87,21 +87,11 @@ def upload(data: PostData, session: SessionDep) -> int: return ratio -@app.get("/user/today") -def user_today(session: SessionDep) -> dict[str, int]: - main_set = set() - fast_set = set() - slow_set = set() +@app.get("/stat") +def user_today(session: SessionDep): + data = {} yesterday = datetime.now(timezone.utc) - timedelta(days=1) for system in session.exec(select(System).where(System.upload_time > yesterday)): - if system.branch == "main": - main_set.add(system.user_id) - elif system.branch == "fast" and system.user_id not in main_set: - fast_set.add(system.user_id) - elif ( - system.branch == "slow" - and system.user_id not in main_set - and system.user_id not in fast_set - ): - slow_set.add(system.user_id) - return {"main": len(main_set), "fast": len(fast_set), "slow": len(slow_set)} + if system.branch in ["main", "fast", "slow"]: + data[system.user_id] = [system.branch, system.commit] + return list(data.values())