mower-ng/stats
5
0
Fork 0
Code Activity

用户数量统计按分支分类

Browse source
This commit is contained in:
zhbaor 2025-01-30 19:55:08 +08:00
parent dfa2373008
commit 2d04b55152
1 changed files with 15 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
main.py
View file

@ -88,18 +88,20 @@ def upload(data: PostData, session: SessionDep) -> int:
@app.get("/user/today")
def user_today(session: SessionDep) -> int:
user_set = set()
def user_today(session: SessionDep) -> dict[str, int]:
main_set = set()
fast_set = set()
slow_set = set()
yesterday = datetime.now(timezone.utc) - timedelta(days=1)
for system in session.exec(select(System).where(System.upload_time > yesterday)):
user_set.add(system.user_id)
return len(user_set)
@app.get("/user/week")
def user_week(session: SessionDep) -> int:
user_set = set()
last_week = datetime.now(timezone.utc) - timedelta(weeks=1)
for system in session.exec(select(System).where(System.upload_time > last_week)):
user_set.add(system.user_id)
return len(user_set)
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)}