feature: Use @ttl_cache for ratings

Author: worldbeater

webapp/managers.py

@@ -31,6 +31,7 @@
     TaskStatusRepository,
     VariantRepository
 )
+from webapp.utils import ttl_cache
 
 
 class AppConfigManager:
@@ -138,6 +139,7 @@ def get_group_statuses(self, group_id: int, hide_pending: bool) -> GroupDto:
             dtos.append(dto)
         return GroupDto(group, tasks, dtos)
 
+    @ttl_cache(duration=15, maxsize=1)
     def get_group_rating(self) -> dict[int, list[GroupInRatingDto]]:
         def key(info: tuple[Group, int]):
             group, _ = info
@@ -154,6 +156,7 @@ def key(info: tuple[Group, int]):
             places[earned].append(GroupInRatingDto(group, earned))
         return dict(sorted(places.items(), reverse=True))
 
+    @ttl_cache(duration=15, maxsize=1)
     def get_rating(self) -> dict[int, list[StudentInRatingDto]]:
         def key(info: tuple[Group, TaskStatus]):
             _, status = info

webapp/utils.py

@@ -1,6 +1,8 @@
+import functools
 import json
 import os
 import sys
+import time
 import traceback
 from datetime import datetime
 from functools import wraps
@@ -11,12 +13,23 @@
 
 from flask import Request, redirect
 
-from webapp.managers import AppConfigManager
 from webapp.models import Student
 from webapp.repositories import StudentRepository
 
 
-def logout(config: AppConfigManager, path: str, auth_redirect=True):
+def ttl_cache(duration: int, maxsize=128, typed=False):
+    def decorator(func):
+        @functools.lru_cache(maxsize=maxsize, typed=typed)
+        def cached(*args, __time, **kwargs):
+            return func(*args, **kwargs)
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            return cached(*args, **kwargs, __time=int(time.time() / duration))
+        return wrapper
+    return decorator
+
+
+def logout(config, path, auth_redirect=True):
     def wrapper(function):
         @wraps(function)
         def decorator(*args, **kwargs):