roozhouandmgleed
b457877623
* BugFix: Select online tab when server disconnected Issue #789 * Simplify online/offline tab selection logic * more robust recovery, reusable restart_or_exit() fn in main, move attempts logic, remove param, etc. Co-authored-by: mgleed <[email protected]>
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
import os, sys
|
|
import keyboard
|
|
import subprocess
|
|
|
|
import template_finder
|
|
from utils.misc import wait, set_d2r_always_on_top
|
|
from screen import get_offset_state, grab
|
|
from ui.main_menu import MAIN_MENU_MARKERS
|
|
|
|
|
|
def process_exists(process_name):
|
|
call = 'TASKLIST', '/FI', 'imagename eq %s' % process_name
|
|
# use buildin check_output right away
|
|
output = subprocess.check_output(call).decode()
|
|
# check in last line for process name
|
|
last_line = output.strip().split('\r\n')[-1]
|
|
# because Fail message could be translated
|
|
return last_line.lower().startswith(process_name.lower())
|
|
|
|
def safe_exit(error_code=0):
|
|
kill_game()
|
|
os._exit(error_code)
|
|
|
|
def kill_game():
|
|
while process_exists("D2R.exe"):
|
|
os.system("taskkill /f /im BlizzardError.exe")
|
|
os.system("taskkill /f /im D2R.exe")
|
|
wait(1.0, 1.5)
|
|
|
|
def restart_game(d2r_path, launch_options):
|
|
kill_game()
|
|
wait(1.0, 1.5)
|
|
# This method should function similar to opening the exe via double-click
|
|
os.startfile(f"{d2r_path}/D2R.exe", arguments = launch_options)
|
|
wait(4.4, 5.5)
|
|
for _ in range(20):
|
|
keyboard.send("space")
|
|
wait(0.5, 1.0)
|
|
success = False
|
|
attempts = 0
|
|
set_d2r_always_on_top()
|
|
while not success:
|
|
success = get_offset_state()
|
|
wait(0.5, 1.0)
|
|
|
|
while not template_finder.search(MAIN_MENU_MARKERS, grab(), best_match=True).valid:
|
|
keyboard.send("space")
|
|
wait(2.0, 4.0)
|
|
attempts += 1
|
|
if attempts >= 5:
|
|
return False
|
|
return True
|
|
|
|
# For testing
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1:
|
|
result = restart_game(sys.argv[1])
|
|
print(result)
|
|
else:
|
|
result = restart_game()
|
|
print(result)
|