Improve create_tab() and create _id_zero attribute at Protocol

plotly · Aug 15, 2024 · 75b752c · 75b752c
1 parent 5bf493a
commit 75b752c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/devtools/protocol.py b/devtools/protocol.py
@@ -8,11 +8,16 @@ def __init__(self, browser_pipe):
         self.browser_session = Session(self, session_id="")
         self.tabs = OrderedDict()
         self.browser_pipe = browser_pipe
+        self._id_zero = True
 
     def create_tab(self):
         tab_obj = Tab()
-        tab_obj.id =  next(reversed(self.tabs.keys())) + 1
-        self.tabs[tab_obj.id] = tab_obj
+        if self._id_zero and 0 not in self.tabs.keys():
+            self._id_zero = False
+            self.tabs[tab_obj.id] = tab_obj
+        else:
+            tab_obj.id =  next(reversed(self.tabs.keys())) + 1
+            self.tabs[tab_obj.id] = tab_obj
         print(f"New Tab Created: {tab_obj.id}")
         return tab_obj