mirror of
https://github.com/ikunshare/Onekey.git
synced 2026-01-16 10:02:56 +08:00
2cf9af811a
Replaced httpx and json with aiohttp and ujson throughout the codebase for improved async performance and faster JSON handling. Updated requirements.txt to reflect new dependencies and removed unused ones. Refactored manifest handling to remove steam.client.cdn dependency and implemented custom manifest serialization. Updated logger to use loguru instead of logzero. Adjusted i18n keys and tray menu logic to match new window-based UI. Updated about.html to reflect backend technology change from HTTPX to AIOHTTP.
114 lines
4.1 KiB
Python
114 lines
4.1 KiB
Python
import os
|
|
import sys
|
|
import ujson
|
|
import winreg
|
|
from pathlib import Path
|
|
from typing import Dict, Optional
|
|
|
|
from .constants import CONFIG_FILE
|
|
from .models import AppConfig
|
|
from .utils.i18n import t
|
|
|
|
DEFAULT_CONFIG = {
|
|
"KEY": "",
|
|
"Port": 5000,
|
|
"Debug_Mode": False,
|
|
"Logging_Files": True,
|
|
"Show_Console": False,
|
|
"Custom_Steam_Path": "",
|
|
"Language": "zh",
|
|
}
|
|
|
|
|
|
class ConfigManager:
|
|
"""配置管理器"""
|
|
|
|
def __init__(self):
|
|
self.config_path = CONFIG_FILE
|
|
self._config_data: Dict = {}
|
|
self.app_config: AppConfig = AppConfig()
|
|
self.steam_path: Optional[Path] = None
|
|
self._load_config()
|
|
|
|
def _generate_config(self) -> None:
|
|
"""生成默认配置文件"""
|
|
try:
|
|
with open(self.config_path, "w", encoding="utf-8") as f:
|
|
ujson.dump(DEFAULT_CONFIG, f, indent=2, ensure_ascii=False)
|
|
print(t("config.generated"))
|
|
os.system("pause")
|
|
sys.exit(1)
|
|
except IOError as e:
|
|
print(t("config.create_failed", error=str(e)))
|
|
os.system("pause")
|
|
sys.exit(1)
|
|
|
|
def _load_config(self) -> None:
|
|
"""加载配置文件"""
|
|
if not self.config_path.exists():
|
|
self._generate_config()
|
|
|
|
try:
|
|
with open(self.config_path, "r", encoding="utf-8") as f:
|
|
self._config_data = ujson.load(f)
|
|
|
|
self.app_config = AppConfig(
|
|
key=self._config_data.get("KEY", ""),
|
|
port=self._config_data.get("Port", 5000),
|
|
custom_steam_path=self._config_data.get("Custom_Steam_Path", ""),
|
|
debug_mode=self._config_data.get("Debug_Mode", False),
|
|
logging_files=self._config_data.get("Logging_Files", True),
|
|
show_console=self._config_data.get("Show_Console", True),
|
|
language=self._config_data.get("Language", "zh"),
|
|
)
|
|
|
|
self.steam_path = self._get_steam_path()
|
|
except ujson.JSONDecodeError:
|
|
print(t("config.corrupted"))
|
|
self._generate_config()
|
|
print(t("config.regenerated"))
|
|
self.app_config = AppConfig(
|
|
key=DEFAULT_CONFIG.get("KEY", ""),
|
|
port=DEFAULT_CONFIG.get("Port", 5000),
|
|
custom_steam_path=DEFAULT_CONFIG.get("Custom_Steam_Path", ""),
|
|
debug_mode=DEFAULT_CONFIG.get("Debug_Mode", False),
|
|
logging_files=DEFAULT_CONFIG.get("Logging_Files", True),
|
|
show_console=DEFAULT_CONFIG.get("Show_Console", True),
|
|
language=DEFAULT_CONFIG.get("Language", "zh"),
|
|
)
|
|
try:
|
|
self.steam_path = self._get_steam_path()
|
|
except Exception:
|
|
self.steam_path = None
|
|
except Exception as e:
|
|
print(t("config.load_failed", error=str(e)))
|
|
print(t("config.use_default"))
|
|
self.app_config = AppConfig(
|
|
key=DEFAULT_CONFIG.get("KEY", ""),
|
|
port=DEFAULT_CONFIG.get("Port", 5000),
|
|
custom_steam_path=DEFAULT_CONFIG.get("Custom_Steam_Path", ""),
|
|
debug_mode=DEFAULT_CONFIG.get("Debug_Mode", False),
|
|
logging_files=DEFAULT_CONFIG.get("Logging_Files", True),
|
|
show_console=DEFAULT_CONFIG.get("Show_Console", True),
|
|
language=DEFAULT_CONFIG.get("Language", "zh"),
|
|
)
|
|
try:
|
|
self.steam_path = self._get_steam_path()
|
|
except Exception:
|
|
self.steam_path = None
|
|
|
|
def _get_steam_path(self) -> Optional[Path]:
|
|
"""获取Steam安装路径"""
|
|
try:
|
|
if self.app_config.custom_steam_path:
|
|
return Path(self.app_config.custom_steam_path)
|
|
|
|
with winreg.OpenKey(
|
|
winreg.HKEY_CURRENT_USER, r"Software\Valve\Steam"
|
|
) as key:
|
|
return Path(winreg.QueryValueEx(key, "SteamPath")[0])
|
|
except Exception as e:
|
|
print(t("config.steam_path_failed", error=str(e)))
|
|
print(t("config.continue_partial"))
|
|
return None
|