shikong
c59e3053c4
移除main.py中未使用的环境变量设置代码，减少冗余。在worker.py中新增模拟手机环境的设置，并优化日志输出，使用log.info替代print，提升代码可维护性。
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
from loguru import logger
|
|
logger.remove()
|
|
|
|
import sys
|
|
|
|
logger.add(sys.stdout,
|
|
filter=lambda record: record["extra"].get("name") == "runtime")
|
|
logger.add("log/runtime.log",
|
|
rotation="100 MB",
|
|
encoding="utf-8",
|
|
filter=lambda record: record["extra"].get("name") == "runtime")
|
|
|
|
log = logger.bind(name="runtime")
|
|
|
|
from DrissionPage import Chromium, ChromiumOptions, SessionPage
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
import atexit
|
|
|
|
import listener.console
|
|
import listener.network
|
|
|
|
# 初始化浏览器
|
|
options = ChromiumOptions()
|
|
# 无头模式
|
|
# options.headless()
|
|
|
|
browser = Chromium(options)
|
|
tab = browser.latest_tab
|
|
|
|
import os
|
|
# 初始化线程池
|
|
cpu_count = os.cpu_count()
|
|
cpu_count = cpu_count if cpu_count is not None else 1
|
|
executor = ThreadPoolExecutor(max_workers=cpu_count * 2)
|
|
executor.submit(listener.console.console_listener, tab)
|
|
executor.submit(listener.network.network_listener, tab)
|
|
|
|
def exit():
|
|
log.info('退出')
|
|
browser.quit()
|
|
executor.shutdown(False)
|
|
|
|
atexit.register(exit)
|
|
|
|
from worker import run_task;
|
|
try:
|
|
run_task(browser, executor, log)
|
|
except Exception as e:
|
|
log.error(e) |