mirror of
https://github.com/LTTLabsOSS/markbench-tests.git
synced 2026-01-08 05:33:52 -05:00
36c542be4a
this is a ford of jd/harness-fixes that is currently live on the benches, the ONLY thing i changed is im rounding the timestamps to the nearest second please approve --------- Co-authored-by: J-Doiron <139803019+J-Doiron@users.noreply.github.com>
70 lines
2.8 KiB
Python
70 lines
2.8 KiB
Python
"""Utility functions for Total War: Warhammer III test script"""
|
|
import os
|
|
import re
|
|
import logging
|
|
import sys
|
|
import shutil
|
|
from pathlib import Path
|
|
sys.path.insert(1, os.path.join(sys.path[0], '..'))
|
|
|
|
from harness_utils.steam import get_app_install_location
|
|
|
|
SCRIPT_DIRECTORY = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
STEAM_GAME_ID = 312670
|
|
EXE_PATH = os.path.join(get_app_install_location(STEAM_GAME_ID), "bin")
|
|
PROCESS_NAME = "StrangeBrigade.exe"
|
|
LOCALAPPDATA = os.getenv("LOCALAPPDATA")
|
|
CONFIG_LOCATION = f"{LOCALAPPDATA}\\Strange Brigade"
|
|
CONFIG_FILENAME = "GraphicsOptions.ini"
|
|
|
|
def read_current_resolution():
|
|
"""Reads resolutions settings from local game file"""
|
|
height_pattern = re.compile(r"Resolution_Width = (\d+);")
|
|
width_pattern = re.compile(r"Resolution_Height = (\d+);")
|
|
cfg = f"{CONFIG_LOCATION}\\{CONFIG_FILENAME}"
|
|
height_value = 0
|
|
width_value = 0
|
|
with open(cfg, encoding="utf-8") as file:
|
|
lines = file.readlines()
|
|
for line in lines:
|
|
height_match = height_pattern.search(line)
|
|
width_match = width_pattern.search(line)
|
|
if height_match is not None:
|
|
height_value = height_match.group(1)
|
|
if width_match is not None:
|
|
width_value = width_match.group(1)
|
|
return (height_value, width_value)
|
|
|
|
def replace_exe():
|
|
"""Replaces the Strange Brigade launcher exe with the Vulkan exe for immediate launching
|
|
"""
|
|
check_backup = Path(f"{EXE_PATH}\\StrangeBrigade_launcher.exe")
|
|
launcher_exe = Path(f"{EXE_PATH}\\StrangeBrigade.exe")
|
|
vulkan_exe = Path(f"{EXE_PATH}\\StrangeBrigade_Vulkan.exe")
|
|
if not os.path.exists(check_backup):
|
|
os.rename(launcher_exe, check_backup)
|
|
shutil.copy(vulkan_exe, launcher_exe)
|
|
logging.info(f"Replacing launcher file in {EXE_PATH}")
|
|
elif os.path.exists(check_backup):
|
|
if not os.path.exists(launcher_exe):
|
|
shutil.copy(vulkan_exe, launcher_exe)
|
|
logging.info(f"Replacing launcher file in {EXE_PATH}")
|
|
else:
|
|
logging.info("Launcher already replaced with Vulkan exe.")
|
|
|
|
def restore_exe():
|
|
"""Restores the launcher exe back to the original exe name to close the loop.
|
|
"""
|
|
check_backup = Path(f"{EXE_PATH}\\StrangeBrigade_launcher.exe")
|
|
launcher_exe = Path(f"{EXE_PATH}\\StrangeBrigade.exe")
|
|
if not os.path.exists(check_backup):
|
|
logging.info(f"Launcher already restored or file does not exist.")
|
|
elif os.path.exists(check_backup):
|
|
if not os.path.exists(launcher_exe):
|
|
os.rename(check_backup, launcher_exe)
|
|
logging.info(f"Restoring launcher file in {EXE_PATH}")
|
|
else:
|
|
os.remove(launcher_exe)
|
|
os.rename(check_backup, launcher_exe)
|
|
logging.info(f"Restoring launcher file in {EXE_PATH}") |