Final main file

core/classes/modules/rpc/rpc_user.py

@@ -1,15 +1,20 @@
 from typing import TYPE_CHECKING, Optional
 
+from starlette.responses import JSONResponse
+from core.classes.interfaces.irpc_endpoint import IRPC
+from core.classes.modules.rpc.rpc_errors import JSONRPCErrorCode
+
 if TYPE_CHECKING:
     from core.loader import Loader
     from core.definition import MUser
 
-class RPCUser:
+class RPCUser(IRPC):
     def __init__(self, loader: 'Loader'):
-        self._ctx = loader
+        super().__init__(loader)
     
-    def user_list(self, **kwargs) -> list[dict]:
-        users = self._ctx.User.UID_DB.copy()
+    def user_list(self, **kwargs) -> JSONResponse:
+        self.reset()
+        users = self.ctx.User.UID_DB.copy()
         copy_users: list['MUser'] = []
 
         for user in users:
@@ -17,14 +22,24 @@ class RPCUser:
             copy_user.connexion_datetime = copy_user.connexion_datetime.strftime('%d-%m-%Y')
             copy_users.append(copy_user)
         
-        return [user.to_dict() for user in copy_users]
+        self.response_model['result'] = [user.to_dict() for user in copy_users]
 
-    def user_get(self, **kwargs) -> Optional[dict]:
-        uidornickname = kwargs.get('uid_or_nickname', None)
-        user = self._ctx.User.get_user(uidornickname)
+        return JSONResponse(self.response_model)
+
+    def user_get(self, **kwargs) -> JSONResponse:
+        self.reset()
+        uidornickname = kwargs.get('uid_or_nickname', '')
+
+        if not uidornickname:
+            self.response_model['error'] = self.create_error_response(JSONRPCErrorCode.INVALID_PARAMS, {'uid_or_nickname': 'The param to use is uid_or_nickname'})
+            return JSONResponse(self.response_model, self.http_status_code.HTTP_405_METHOD_NOT_ALLOWED)
+
+        user = self.ctx.User.get_user(uidornickname)
         if user:
             user_copy = user.copy()
             user_copy.connexion_datetime = user_copy.connexion_datetime.strftime('%d-%m-%Y')
-            return user_copy.to_dict()
+            self.response_model['result'] = user_copy.to_dict()
+            return JSONResponse(self.response_model)
         
-        return None
+        self.response_model['result'] = 'User not found!'
+        return JSONResponse(self.response_model, self.http_status_code.HTTP_204_NO_CONTENT)