40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
|
"""Pro-Nutzer-Tageskontingent (Kostenkontrolle).
|
||
|
|
|
||
|
|
Limit aus Settings (`daily_request_limit`), pro Nutzer ueber `prefs.daily_request_limit`
|
||
|
|
ueberschreibbar. 0 bedeutet unbegrenzt.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from app.config import settings, Settings
|
||
|
|
from app.metrics import metrics
|
||
|
|
|
||
|
|
|
||
|
|
class QuotaExceededError(Exception):
|
||
|
|
def __init__(self, limit: int, count: int):
|
||
|
|
self.limit = limit
|
||
|
|
self.count = count
|
||
|
|
super().__init__(f"Daily request limit reached ({count}/{limit})")
|
||
|
|
|
||
|
|
|
||
|
|
def effective_limit(user, cfg: Settings = settings) -> int:
|
||
|
|
pref = user.prefs.get("daily_request_limit") if user and user.prefs else None
|
||
|
|
if pref is not None:
|
||
|
|
try:
|
||
|
|
return int(pref)
|
||
|
|
except (TypeError, ValueError):
|
||
|
|
pass
|
||
|
|
return cfg.daily_request_limit
|
||
|
|
|
||
|
|
|
||
|
|
def enforce_quota(user, store, cfg: Settings = settings) -> None:
|
||
|
|
"""Wirft QuotaExceededError, wenn das Tageslimit erreicht ist."""
|
||
|
|
limit = effective_limit(user, cfg)
|
||
|
|
if limit and limit > 0:
|
||
|
|
count = store.get_request_count(user.id)
|
||
|
|
if count >= limit:
|
||
|
|
metrics.inc("quota_exceeded_total")
|
||
|
|
raise QuotaExceededError(limit, count)
|
||
|
|
|
||
|
|
|
||
|
|
def record_usage(user, store, units: int = 0) -> None:
|
||
|
|
store.add_usage(user.id, units)
|
||
|
|
metrics.inc("turns_total")
|