From 2c652189faf629c61da612cb41c2f7498acf673d Mon Sep 17 00:00:00 2001
From: Jerry Yan <792602257@qq.com>
Date: Sun, 12 Jun 2022 01:11:18 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=BC=E5=AE=B9=E6=9C=AA=E8=AE=BE=E7=BD=AE?=
 =?UTF-8?q?=E7=9A=84=E6=83=85=E5=86=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 browser/manager.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/browser/manager.py b/browser/manager.py
index aa1942e..751559e 100644
--- a/browser/manager.py
+++ b/browser/manager.py
@@ -1,10 +1,10 @@
 import threading
+from typing import TYPE_CHECKING
 from urllib.parse import urlparse
 
-from config.helper import config
-from browser.edge import EdgeDriver
 from browser.chrome import ChromeDriver
-from typing import TYPE_CHECKING
+from browser.edge import EdgeDriver
+from config.helper import config
 
 if TYPE_CHECKING:
     from typing import Type, Optional, List
@@ -27,10 +27,11 @@ class BrowserManager():
         self._tabs: "List[TabInfo]" = []
 
     def init_browser(self):
-        _users = config()['live']['users']
+        _live_config = config().get("live", {})
+        _users = _live_config.get("users", [])
         if type(_users) is not list:
             _users = [_users]
-        _rooms = config()['live']['rooms']
+        _rooms = _live_config.get("rooms", [])
         if type(_rooms) is not list:
             _rooms = [_rooms]
         for _user in _users: