feat: implement configfile
Basically still ConfigParser
This commit is contained in:
parent
e20bbab5d7
commit
25ca6cf2cb
4
vollerei/common/__init__.py
Normal file
4
vollerei/common/__init__.py
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
from vollerei.common.configfile import ConfigFile
|
||||||
|
from vollerei.common.telemetry import block_telemetry, check_telemetry
|
||||||
|
|
||||||
|
__all__ = ["ConfigFile", "block_telemetry", "check_telemetry"]
|
15
vollerei/common/configfile.py
Normal file
15
vollerei/common/configfile.py
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
from configparser import ConfigParser
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
|
||||||
|
class ConfigFile(ConfigParser):
|
||||||
|
path: Path
|
||||||
|
|
||||||
|
def __init__(self, path, **kwargs):
|
||||||
|
super().__init__(**kwargs)
|
||||||
|
self.path = Path(path)
|
||||||
|
self.read(self.path)
|
||||||
|
|
||||||
|
def save(self):
|
||||||
|
with self.path.open("w") as f:
|
||||||
|
self.write(f)
|
@ -2,6 +2,7 @@ from hashlib import md5
|
|||||||
from os import PathLike
|
from os import PathLike
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
|
from vollerei.common import ConfigFile
|
||||||
from vollerei.abc.launcher.game import GameABC
|
from vollerei.abc.launcher.game import GameABC
|
||||||
from vollerei.hsr.constants import MD5SUMS
|
from vollerei.hsr.constants import MD5SUMS
|
||||||
|
|
||||||
@ -89,6 +90,24 @@ class Game(GameABC):
|
|||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
def _get_version_config(self) -> tuple[int, int, int]:
|
||||||
|
cfg_file = self._path.joinpath("config.ini")
|
||||||
|
if not cfg_file.exists():
|
||||||
|
return (0, 0, 0)
|
||||||
|
cfg = ConfigFile(cfg_file)
|
||||||
|
if "General" not in cfg.sections():
|
||||||
|
return (0, 0, 0)
|
||||||
|
if "game_version" not in cfg["General"]:
|
||||||
|
return (0, 0, 0)
|
||||||
|
version_str = cfg["General"]["game_version"]
|
||||||
|
if version_str.count(".") != 2:
|
||||||
|
return (0, 0, 0)
|
||||||
|
try:
|
||||||
|
version = tuple(int(i) for i in version_str.split("."))
|
||||||
|
except Exception:
|
||||||
|
return (0, 0, 0)
|
||||||
|
return version
|
||||||
|
|
||||||
def get_version(self) -> tuple[int, int, int]:
|
def get_version(self) -> tuple[int, int, int]:
|
||||||
"""
|
"""
|
||||||
Get the current installed game version.
|
Get the current installed game version.
|
||||||
@ -116,6 +135,7 @@ class Game(GameABC):
|
|||||||
version_bytes: list[list[bytes]] = [[], [], []]
|
version_bytes: list[list[bytes]] = [[], [], []]
|
||||||
version_ptr = 0
|
version_ptr = 0
|
||||||
correct = True
|
correct = True
|
||||||
|
try:
|
||||||
with self.data_folder().joinpath("data.unity3d").open("rb") as f:
|
with self.data_folder().joinpath("data.unity3d").open("rb") as f:
|
||||||
f.seek(0x7D0) # 2000 in decimal
|
f.seek(0x7D0) # 2000 in decimal
|
||||||
for byte in f.read(10000):
|
for byte in f.read(10000):
|
||||||
@ -145,7 +165,10 @@ class Game(GameABC):
|
|||||||
version_bytes[version_ptr].append(byte)
|
version_bytes[version_ptr].append(byte)
|
||||||
else:
|
else:
|
||||||
correct = False
|
correct = False
|
||||||
return (0, 0, 0)
|
except Exception:
|
||||||
|
pass
|
||||||
|
# Fallback to config.ini
|
||||||
|
return self._get_version_config()
|
||||||
|
|
||||||
def get_version_str(self) -> str:
|
def get_version_str(self) -> str:
|
||||||
"""
|
"""
|
||||||
|
Loading…
Reference in New Issue
Block a user