mirror of
https://github.com/LTTLabsOSS/markbench-tests.git
synced 2026-01-09 14:07:56 -05:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""3dmark test utils"""
|
|
from pathlib import Path
|
|
import psutil
|
|
import winreg
|
|
import re
|
|
|
|
SCRIPT_DIR = Path(__file__).resolve().parent
|
|
LOG_DIR = SCRIPT_DIR / "run"
|
|
|
|
def is_process_running(process_name):
|
|
"""check if given process is running"""
|
|
for process in psutil.process_iter(['pid', 'name']):
|
|
if process.info['name'] == process_name:
|
|
return process
|
|
return None
|
|
|
|
def regex_find_score_in_xml(result_regex):
|
|
"""Reads score from local game log"""
|
|
score_pattern = re.compile(result_regex)
|
|
cfg = f"{LOG_DIR}\\result.xml"
|
|
score_value = 0
|
|
with open(cfg, encoding="utf-8") as file:
|
|
lines = file.readlines()
|
|
for line in lines:
|
|
score_match = score_pattern.search(line)
|
|
if score_match is not None:
|
|
score_value = score_match.group(1)
|
|
return score_value
|
|
|
|
def get_install_path() -> str:
|
|
"""Gets the path to the Steam installation directory from the SteamPath registry key"""
|
|
reg_path = r"Software\UL\Procyon"
|
|
reg_key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_READ)
|
|
value, _ = winreg.QueryValueEx(reg_key, "InstallDir")
|
|
return value
|