From 6273afb58203dfbcd28c3a0f9f9eb7bec19fe626 Mon Sep 17 00:00:00 2001 From: raphael60650 Date: Wed, 4 Aug 2021 21:12:40 +0200 Subject: [PATCH] first choose will be on the last track that got a screenshot --- scripts/ask_map_preview.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/scripts/ask_map_preview.py b/scripts/ask_map_preview.py index fa48334..9134950 100644 --- a/scripts/ask_map_preview.py +++ b/scripts/ask_map_preview.py @@ -24,8 +24,7 @@ driver.execute_script("var element = arguments[0]; element.parentNode.removeChil driver.find_element_by_id("Panel")) tracks = glob.iglob("../file/Track/*.szs") -track = next(tracks) -shutil.copy(track, "./map preview/tmp/" + get_nodir(track)) +track = "none.png" def save_screenshot(): @@ -36,7 +35,8 @@ def save_screenshot(): def skip_to_next(): global track - os.remove("./map preview/tmp/" + get_nodir(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"): @@ -45,6 +45,7 @@ def skip_to_next(): shutil.copy(track, "./map preview/tmp/" + get_nodir(track)) +skip_to_next() keyboard.add_hotkey('h', save_screenshot) while True: