Optimize some content

This commit is contained in:
zhaoyafan 2025-02-24 02:05:25 +08:00
parent cd007774a0
commit 60729db9d7
1 changed files with 13 additions and 0 deletions

View File

@ -10,6 +10,7 @@ import signal
import random
import shutil
import socket
import psutil
import inspect
import uvicorn
import zipfile
@ -2392,6 +2393,14 @@ class MainRunner:
def _read_file_text(src):
return open(file=src, mode='r', encoding='utf-8').read() if os.path.exists(src) else ''
@staticmethod
def preprocessing():
for proc in psutil.process_iter():
try:
proc.name().lower() == 'chromedriver.exe' and proc.terminate()
except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
pass
def install_chrome(self):
chrome_install_source = os.path.join(self.app_root, 'Chrome')
chrome_install_source_version = os.path.join(chrome_install_source, 'chrome.version')
@ -2421,6 +2430,10 @@ class MainRunner:
sys.path.append(os.path.join(os.path.dirname(__file__), 'site-packages.zip'))
sys.path.append(os.path.join(self.app_data, 'Packages'))
sys.path.append(os.path.join(self.app_data, 'site-packages.zip'))
try:
self.preprocessing()
except Exception:
pass
try:
self.install_chrome() and print('Chrome is now installed.', file=sys.stderr)
except PermissionError: