Change version check to only macos

This commit is contained in:
naibo 2023-12-22 04:16:36 +08:00
parent 56b06c467a
commit 1982cb3c63
4 changed files with 1308 additions and 1292 deletions

View File

@ -1835,16 +1835,20 @@ class BrowserThread(Thread):
content = self.browser.title content = self.browser.title
elif p["contentType"] == 7: elif p["contentType"] == 7:
# 获取整个网页的高度和宽度 # 获取整个网页的高度和宽度
height = self.browser.execute_script( size = self.browser.get_window_size()
"return document.body.scrollHeight") width = size["width"]
width = self.browser.execute_script( height = size["height"]
"return document.body.scrollWidth")
# 调整浏览器窗口的大小 # 调整浏览器窗口的大小
self.browser.set_window_size(width, height) self.browser.set_window_size(width, height)
element.screenshot("Data/Task_" + str(self.id) + "/" + self.saveName + element.screenshot("Data/Task_" + str(self.id) + "/" + self.saveName +
"/" + str(time.time()) + ".png") "/" + str(time.time()) + ".png")
# 截图完成后,将浏览器的窗口大小设置为原来的大小
self.browser.set_window_size(width, height)
elif p["contentType"] == 8: elif p["contentType"] == 8:
try: try:
size = self.browser.get_window_size()
width = size["width"]
height = size["height"]
screenshot = element.screenshot_as_png screenshot = element.screenshot_as_png
screenshot_stream = io.BytesIO(screenshot) screenshot_stream = io.BytesIO(screenshot)
# 使用Pillow库打开截图并转换为灰度图像 # 使用Pillow库打开截图并转换为灰度图像
@ -1858,6 +1862,7 @@ class BrowserThread(Thread):
image_bytes = f.read() image_bytes = f.read()
content = ocr.classification(image_bytes) content = ocr.classification(image_bytes)
os.remove(location) os.remove(location)
self.browser.set_window_size(width, height)
# 使用Tesseract OCR引擎识别图像中的文本 # 使用Tesseract OCR引擎识别图像中的文本
# content = pytesseract.image_to_string(image, lang='chi_sim+eng') # content = pytesseract.image_to_string(image, lang='chi_sim+eng')
except Exception as e: except Exception as e:
@ -2258,13 +2263,21 @@ if __name__ == '__main__':
tmp_user_data_folder = os.path.join(tmp_user_folder_parent, "user_data_" + str(id) + "_" + str(time.time()).replace(".","") + "_" + random_string) tmp_user_data_folder = os.path.join(tmp_user_folder_parent, "user_data_" + str(id) + "_" + str(time.time()).replace(".","") + "_" + random_string)
tmp_options[i]["tmp_user_data_folder"] = tmp_user_data_folder tmp_options[i]["tmp_user_data_folder"] = tmp_user_data_folder
if os.path.exists(tmp_user_data_folder): if os.path.exists(tmp_user_data_folder):
shutil.rmtree(tmp_user_data_folder) try:
shutil.rmtree(tmp_user_data_folder)
except:
pass
print(f"Copying user data folder to: {tmp_user_data_folder}, please wait...") print(f"Copying user data folder to: {tmp_user_data_folder}, please wait...")
print(f"正在复制用户信息目录到: {tmp_user_data_folder},请稍等...") print(f"正在复制用户信息目录到: {tmp_user_data_folder},请稍等...")
if os.path.exists(absolute_user_data_folder): if os.path.exists(absolute_user_data_folder):
shutil.copytree(absolute_user_data_folder, tmp_user_data_folder) try:
print("User data folder copied successfully, if you exit the program before it finishes, please delete the temporary user data folder manually.") shutil.copytree(absolute_user_data_folder, tmp_user_data_folder)
print("用户信息目录复制成功,如果程序在运行过程中被手动退出,请手动删除临时用户信息目录。") print("User data folder copied successfully, if you exit the program before it finishes, please delete the temporary user data folder manually.")
print("用户信息目录复制成功,如果程序在运行过程中被手动退出,请手动删除临时用户信息目录。")
except:
tmp_user_data_folder = absolute_user_data_folder
print("Copy user data folder failed, use the original folder.")
print("复制用户信息目录失败,使用原始目录。")
else: else:
tmp_user_data_folder = absolute_user_data_folder tmp_user_data_folder = absolute_user_data_folder
print("Cannot find user data folder, create a new folder.") print("Cannot find user data folder, create a new folder.")

Binary file not shown.

Binary file not shown.

File diff suppressed because it is too large Load Diff