ipdate

devplacepy/routers/admin/devii_tasks.py

@@ -11,7 +11,9 @@ from devplacepy.schemas import AdminDeviiTasksOut
 from devplacepy.seo import base_seo_context, site_url, website_schema
 from devplacepy.services.audit import record as audit
 from devplacepy.services.devii import config as devii_config
+from devplacepy.services.devii.tasks import limits
 from devplacepy.services.devii.tasks.guards import DEFAULT_MAX_PER_OWNER
+from devplacepy.services.devii.tasks.schedule import now_utc
 from devplacepy.services.devii.tasks.store import TABLE, TaskStore
 from devplacepy.utils import not_found, require_admin
 
@@ -43,9 +45,25 @@ def _rows(state: str) -> list[dict]:
     return rows
 
 
+def _quotas(owner_uids: set[str]) -> dict[str, dict]:
+    reference = now_utc()
+    quotas = {}
+    for owner_uid in owner_uids:
+        runs = limits.run_quota(db, "user", owner_uid, reference)
+        creations = limits.create_quota(db, "user", owner_uid, reference)
+        quotas[owner_uid] = {
+            "runs_used": runs.used,
+            "runs_limit": runs.limit,
+            "creates_used": creations.used,
+            "creates_limit": creations.limit,
+        }
+    return quotas
+
+
 def _items(rows: list[dict]) -> list[dict]:
     owners = get_users_by_uids([row.get("owner_id") for row in rows if row.get("owner_id")])
     admins = get_admin_uids()
+    quotas = _quotas({str(row.get("owner_id") or "") for row in rows if row.get("owner_id")})
     items = []
     for row in rows:
         owner_uid = row.get("owner_id") or ""
@@ -57,6 +75,7 @@ def _items(rows: list[dict]) -> list[dict]:
                 "owner_uid": owner_uid,
                 "owner": owner["username"] if owner else owner_uid,
                 "owner_is_admin": owner_uid in admins,
+                "quota": quotas.get(owner_uid, {}),
                 "schedule": _schedule_text(row),
                 "status": row.get("status") or "",
                 "enabled": bool(row.get("enabled")),
@@ -85,7 +104,7 @@ async def admin_devii_tasks(request: Request, state: str = "active"):
         state = "active"
     rows = _rows(state)
     items = _items(rows)
-    limits = {
+    bounds = {
         "max_concurrent": get_int_setting(
             devii_config.FIELD_TASK_MAX_CONCURRENT,
             devii_config.DEFAULT_TASK_MAX_CONCURRENT,
@@ -100,6 +119,10 @@ async def admin_devii_tasks(request: Request, state: str = "active"):
         "idle_days": get_int_setting(
             devii_config.FIELD_TASK_IDLE_DAYS, devii_config.DEFAULT_TASK_IDLE_DAYS
         ),
+        "member_create_24h": limits.create_limit(False),
+        "member_runs_24h": limits.run_limit(False),
+        "admin_create_24h": limits.create_limit(True),
+        "admin_runs_24h": limits.run_limit(True),
     }
     tabs = [
         {"key": key, "label": key.capitalize(), "active": key == state}
@@ -127,7 +150,7 @@ async def admin_devii_tasks(request: Request, state: str = "active"):
             "items": items,
             "state": state,
             "tabs": tabs,
-            "limits": limits,
+            "limits": bounds,
             "admin_section": "devii-tasks",
         },
         model=AdminDeviiTasksOut,