Add base controller, catch controller, and history controller

2026-08-28 11:33:30 +00:00 · 2024-03-22 00:48:46 +02:00
parent f6b750247c
commit ca6c933fd9
3 changed files with 124 additions and 0 deletions
@@ -0,0 +1,73 @@
+from fastapi import Response, status
+from schema.answer import Answer
+from schema.request_data import RequestData
+from manager.history import History
+from controller.base_controller import BaseController
+from schema.status import Status
+
+
+class HistoryController(BaseController):
+    async def history(
+        self,
+        response: Response,
+        id: int | None = None,
+        namespace: str = "requests",
+    ) -> Answer | RequestData | list[RequestData]:
+        if not self.check_namespace(namespace=namespace):
+            response.status_code = status.HTTP_400_BAD_REQUEST
+            return Answer(
+                status=Status.error,
+                message="Invalid namespace name provided.",
+            )
+
+        history = History(namespace=namespace)
+
+        if id is not None:
+            item = history.get(id)
+            if item is None:
+                return Answer(
+                    status=Status.error,
+                    message="No requests found.",
+                )
+            return RequestData(**item)
+        return [RequestData(**d) for d in history.all()]
+
+    async def last_requests(
+        self,
+        response: Response,
+        namespace: str = "requests",
+    ) -> Answer | RequestData:
+        if not self.check_namespace(namespace=namespace):
+            response.status_code = status.HTTP_400_BAD_REQUEST
+            return Answer(
+                status=Status.error,
+                message="Invalid namespace name provided.",
+            )
+
+        last = History(namespace=namespace).last()
+
+        if last is None:
+            return Answer(
+                status=Status.error,
+                message="No requests found.",
+            )
+
+        return RequestData(**last)
+
+    async def clear_history(
+        self,
+        response: Response,
+        namespace: str = "requests",
+    ) -> Answer:
+        if not self.check_namespace(namespace=namespace):
+            response.status_code = status.HTTP_400_BAD_REQUEST
+            return Answer(
+                status=Status.error,
+                message="Invalid namespace name provided.",
+            )
+
+        History(namespace=namespace).clear()
+        return Answer(
+            status=Status.success,
+            message="History cleared.",
+        )