q792602257/Douyin_Web_Live
Archived
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

兼容未设置的情况

Browse Source
This commit is contained in:
Jerry Yan 2022-06-12 01:11:18 +08:00
parent 6bbb5cddb2
commit 2c652189fa
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
browser/manager.py
View File

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