mirror of
https://github.com/Faraphel/Atlas-Install.git
synced 2025-07-04 19:58:26 +02:00
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
"""
|
|
this script allow you to more easily select map preview for each track.
|
|
"""
|
|
|
|
from selenium import webdriver
|
|
import keyboard
|
|
import shutil
|
|
import ctypes
|
|
import time
|
|
import glob
|
|
import os
|
|
|
|
get_filename = lambda file: ".".join(file.split(".")[:-1])
|
|
get_nodir = lambda file: file.replace("\\", "/").split("/")[-1]
|
|
move_cursor_to = ctypes.windll.user32.SetCursorPos
|
|
|
|
chrome_option = webdriver.ChromeOptions()
|
|
driver = webdriver.Chrome("./map preview/chromedriver.exe", options=chrome_option)
|
|
driver.get("https://noclip.website/")
|
|
driver.fullscreen_window()
|
|
time.sleep(5)
|
|
|
|
driver.execute_script("var element = arguments[0]; element.parentNode.removeChild(element);",
|
|
driver.find_element_by_id("Panel"))
|
|
|
|
tracks = glob.iglob("../file/Track/*.szs")
|
|
track = "none.png"
|
|
|
|
|
|
def save_screenshot():
|
|
global track
|
|
driver.save_screenshot(filename=f"./map preview/image/{get_filename(get_nodir(track))}.png")
|
|
skip_to_next()
|
|
|
|
|
|
def skip_to_next():
|
|
global track
|
|
if os.path.exists("./map preview/tmp/" + get_nodir(track)):
|
|
os.remove("./map preview/tmp/" + get_nodir(track))
|
|
|
|
track = next(tracks)
|
|
while os.path.exists(f"./map preview/image/{get_filename(get_nodir(track))}.png"):
|
|
track = next(tracks)
|
|
|
|
shutil.copy(track, "./map preview/tmp/" + get_nodir(track))
|
|
|
|
|
|
skip_to_next()
|
|
keyboard.add_hotkey('h', save_screenshot)
|
|
|
|
while True:
|
|
time.sleep(1)
|