Improve send_command() using write_jsons() at Session

plotly · Aug 16, 2024 · cc0a034 · cc0a034
1 parent e30ac01
commit cc0a034
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/devtools/session.py b/devtools/session.py
@@ -28,11 +28,24 @@ def send_command(self, command, params=None, cb=None):
             "method": command,
         }
 
+        if self.session_id != "":
+            json_command["session_id"] = self.session_id
+
         if params:
             json_command["params"] = params
 
-        if self.session_id != "":
-            json_command["session_id"] = self.session_id
+            self.parent_connection.browser_pipe.write_jsons(
+                message_id=json_command["id"],
+                method=json_command["method"],
+                params=json_command["params"],
+                session_id=self.session_id,
+            )
+        else:
+            self.parent_connection.browser_pipe.write_jsons(
+                message_id=json_command["id"],
+                method=json_command["method"],
+                session_id=self.session_id,
+            )
 
         self.message_id += 1