Optimized code

roc/request.py

@@ -34,3 +34,11 @@ def get_result(self) -> dict | None:
 
     def get_error(self) -> dict | None:
         return self.error
+
+
+def make_response(data: dict) -> Response:
+    for key in ["id", "result", "error"]:
+        if key not in data:
+            data[key] = None
+
+    return Response(data["id"], data["result"], data["error"])

roc/socket.py

@@ -11,6 +11,7 @@
 from roc.packet import Packet
 from roc.request import Request
 from roc.request import Response
+from roc.request import make_response
 
 
 class SocketException(Exception):
@@ -89,16 +90,7 @@ async def request(self, request: Request) -> Response:
 
         data = json.loads(res)
 
-        if "id" not in data:
-            data["id"] = None
-
-        if "result" not in data:
-            data["result"] = None
-
-        if "error" not in data:
-            data["error"] = None
-
-        return Response(data["id"], data["result"], data["error"])
+        return make_response(data)
 
     async def start(self):
         reader, writer = await asyncio.open_connection(self.host, self.port)