Merge pull request #41 from muwenyan521/main

feat:尝试解决配置问题&优化代码
2026-01-13 00:27:32 +08:00 · 2024-10-01 14:02:08 +00:00
parent b1b7e720c0 cf0d508237
commit 22df5426d8
15 changed files with 372 additions and 255 deletions
--- a/common/check.py
+++ b/common/check.py
@@ -1,28 +1,30 @@
 import time
 import ujson as json
-
 from aiohttp import ClientError
 from .log import log
 from .stack_error import stack_error

-
 async def check_github_api_rate_limit(headers, session):
+    url = 'https://api.github.com/rate_limit'
+    
    try:
-        url = 'https://api.github.com/rate_limit'
-
        async with session.get(url, headers=headers, ssl=False) as r:
            r_json = await r.json()

            if r.status == 200:
-                rate_limit = r_json['rate']
-                remaining_requests = rate_limit['remaining']
-                reset_time = rate_limit['reset']
+                rate_limit = r_json.get('rate', {})
+                remaining_requests = rate_limit.get('remaining', 0)
+                reset_time = rate_limit.get('reset', 0)
                reset_time_formatted = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(reset_time))
+                
                log.info(f' 🔄 剩余请求次数: {remaining_requests}')
+
+                if remaining_requests == 0:
+                    log.warning(f'⚠ GitHub API 请求数已用尽，将在 {reset_time_formatted} 重置，建议生成一个填在配置文件里')
            else:
                log.error('⚠ Github请求数检查失败，网络错误')
-
-        if remaining_requests == 0:
-            log.warning(f'⚠ GitHub API 请求数已用尽，将在 {reset_time_formatted} 重置, 不想等生成一个填配置文件里')
+    
    except ClientError as e:
-        log.error(f'⚠ 检查Github API 请求数失败，{stack_error(e)}')
+        log.error(f'⚠ 检查Github API 请求数失败，{stack_error(e)}')
+    except Exception as e:
+        log.error(f'⚠ 发生错误: {stack_error(e)}')
--- a/common/config.py
+++ b/common/config.py
@@ -3,50 +3,55 @@ import sys
 import asyncio
 import ujson as json
 import aiofiles
-
 from .stack_error import stack_error
 from .log import log

+DEFAULT_CONFIG = {
+    "Github_Personal_Token": "",
+    "Custom_Steam_Path": "",
+    "QA1": "温馨提示：Github_Personal_Token可在Github设置的最底下开发者选项找到，详情看教程",
+    "教程": "https://ikunshare.com/Onekey_tutorial"
+}
+
+def validate_config(config):
+    # 检查配置文件的有效性
+    required_keys = DEFAULT_CONFIG.keys()
+    for key in required_keys:
+        if key not in config:
+            return False
+    return True

 async def gen_config_file():
    try:
-        qa1 = "温馨提示：Github_Personal_Token可在Github设置的最底下开发者选项找到，详情看教程"
-        tutorial = "https://ikunshare.com/Onekey_tutorial"
-        default_config ={
-                        "Github_Personal_Token": "",
-                        "Custom_Steam_Path": "",
-                        "QA1": qa1,
-                        "教程": tutorial
-                        }
-        async with aiofiles.open("./config.json",
-                                mode="w",
-                                encoding="utf-8") as f:
-            await f.write(json.dumps(default_config,
-                                    indent=2,
-                                    ensure_ascii=False,
-                                    escape_forward_slashes=False))
-            await f.close()
-        log.info('🖱️ 程序可能为第一次启动，请填写配置文件后重新启动程序')
+        async with aiofiles.open("./config.json", mode="w", encoding="utf-8") as f:
+            await f.write(json.dumps(DEFAULT_CONFIG, indent=2, ensure_ascii=False, escape_forward_slashes=False))
+        
+        log.info('🖱️ 程序可能为第一次启动或配置重置，请填写配置文件后重新启动程序')
    except Exception as e:
        log.error(f'❗ 配置文件生成失败，{stack_error(e)}')

-
 async def load_config():
    if not os.path.exists('./config.json'):
        await gen_config_file()
        os.system('pause')
        sys.exit()
-    else:
-        try:
-            async with aiofiles.open("./config.json",
-                                     mode="r",
-                                     encoding="utf-8") as f:
-                config = json.loads(await f.read())
-                return config
-        except Exception as e:
-            log.error(f"配置文件加载失败，原因: {stack_error(e)}")
-            os.remove("./config.json")
-            os.system('pause')

-        
+    try:
+        async with aiofiles.open("./config.json", mode="r", encoding="utf-8") as f:
+            config = json.loads(await f.read())
+
+            if not validate_config(config):
+                log.error("配置文件格式无效，正在重置为默认配置...")
+                await gen_config_file()
+                os.system('pause')
+                sys.exit()
+
+            return config
+    except Exception as e:
+        log.error(f"配置文件加载失败，原因: {stack_error(e)}")
+        os.remove("./config.json")
+        await gen_config_file()
+        os.system('pause')
+        sys.exit()
+
 config = asyncio.run(load_config())
--- a/common/dkey_merge.py
+++ b/common/dkey_merge.py
@@ -1,27 +1,40 @@
 import asyncio
 import aiofiles
 import vdf
-
+from pathlib import Path
 from .log import log

 lock = asyncio.Lock()

-
-async def depotkey_merge(config_path, depots_config):
+async def depotkey_merge(config_path: Path, depots_config: dict) -> bool:
    if not config_path.exists():
        async with lock:
-            log.error(' 👋 Steam默认配置不存在，可能是没有登录账号')
-        return
-    async with aiofiles.open(config_path, encoding='utf-8') as f:
-        content = await f.read()
-    config = vdf.loads(content)
-    software = config['InstallConfigStore']['Software']
-    valve = software.get('Valve') or software.get('valve')
-    steam = valve.get('Steam') or valve.get('steam')
-    if 'depots' not in steam:
-        steam['depots'] = {}
-    steam['depots'].update(depots_config['depots'])
-    async with aiofiles.open(config_path, mode='w', encoding='utf-8') as f:
-        new_content = vdf.dumps(config, pretty=True)
-        await f.write(new_content)
-    return True
+            log.error('👋 Steam默认配置不存在，可能是没有登录账号')
+        return False
+
+    try:
+        async with aiofiles.open(config_path, encoding='utf-8') as f:
+            content = await f.read()
+        
+        config = vdf.loads(content)
+        steam = config.get('InstallConfigStore', {}).get('Software', {}).get('Valve') or \
+                config.get('InstallConfigStore', {}).get('Software', {}).get('valve')
+
+        if steam is None:
+            log.error('⚠ 找不到Steam配置，请检查配置文件')
+            return False
+        
+        depots = steam.setdefault('depots', {})
+        depots.update(depots_config.get('depots', {}))
+        
+        async with aiofiles.open(config_path, mode='w', encoding='utf-8') as f:
+            new_context = vdf.dumps(config, pretty=True)
+            await f.write(new_context)
+
+        log.info('✅ 成功合并')
+        return True
+        
+    except Exception as e:
+        async with lock:
+            log.error(f'❗ 合并失败，原因: {e}')
+        return False
--- a/common/get_manifest_info.py
+++ b/common/get_manifest_info.py
@@ -7,28 +7,37 @@ from .manifest_down import get
 from .stack_error import stack_error


-async def get_manifest(sha, path, steam_path: Path, repo, session):
+async def get_manifest(sha: str, path: str, steam_path: Path, repo: str, session) -> list:
    collected_depots = []
+    depot_cache_path = steam_path / 'depotcache'
+    
    try:
+        depot_cache_path.mkdir(exist_ok=True)
+
        if path.endswith('.manifest'):
-            depot_cache_path = steam_path / 'depotcache'
-            if not depot_cache_path.exists():
-                depot_cache_path.mkdir(exist_ok=True)
            save_path = depot_cache_path / path
            if save_path.exists():
-                log.warning(f'👋已存在清单: {path}')
+                log.warning(f'👋 已存在清单: {save_path}')
                return collected_depots
+
            content = await get(sha, path, repo, session)
-            log.info(f' 🔄 清单下载成功: {path}')
+            log.info(f'🔄 清单下载成功: {path}')
+
            async with aiofiles.open(save_path, 'wb') as f:
                await f.write(content)
+
        elif path == 'Key.vdf':
            content = await get(sha, path, repo, session)
-            log.info(f' 🔄 密钥下载成功: {path}')
-            depots_config = vdf.loads(content.decode(encoding='utf-8'))
-            for depot_id, depot_info in depots_config['depots'].items():
-                collected_depots.append((depot_id, depot_info['DecryptionKey']))
+            log.info(f'🔄 密钥下载成功: {path}')
+
+            depots_config = vdf.loads(content.decode('utf-8'))
+            collected_depots = [
+                (depot_id, depot_info['DecryptionKey'])
+                for depot_id, depot_info in depots_config['depots'].items()
+            ]
+
    except Exception as e:
-        log.error(f'处理失败: {path} - {stack_error(e)}')
+        log.error(f'❗ 处理失败: {path} - {stack_error(e)}')
        raise
-    return collected_depots
+
+    return collected_depots
--- a/common/get_steam_path.py
+++ b/common/get_steam_path.py
@@ -1,23 +1,20 @@
 import os
-
 import winreg
-
 from pathlib import Path
 from .log import log
 from .config import config
 from .stack_error import stack_error

-def get_steam_path():
+def get_steam_path() -> Path:
    try:
        key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'Software\Valve\Steam')
        steam_path = Path(winreg.QueryValueEx(key, 'SteamPath')[0])
-        custom_steam_path = config["Custom_Steam_Path"]
-        if not custom_steam_path == '':
-            return Path(custom_steam_path)
-        else:
-            return steam_path
+        
+        custom_steam_path = config.get("Custom_Steam_Path", "").strip()
+        return Path(custom_steam_path) if custom_steam_path else steam_path
    except Exception as e:
        log.error(f'Steam路径获取失败, {stack_error(e)}')
        os.system('pause')
-    
-steam_path = get_steam_path()
+        return Path()
+
+steam_path = get_steam_path()
--- a/common/gl_unlock.py
+++ b/common/gl_unlock.py
@@ -1,28 +1,32 @@
 from .get_steam_path import steam_path
+from pathlib import Path

-async def greenluma_add(depot_id_list):
+async def greenluma_add(depot_id_list: list) -> bool:
    app_list_path = steam_path / 'AppList'
-    if app_list_path.exists() and app_list_path.is_file():
-        app_list_path.unlink(missing_ok=True)
-    if not app_list_path.is_dir():
+
+    try:
        app_list_path.mkdir(parents=True, exist_ok=True)
-    depot_dict = {}
-    for i in app_list_path.iterdir():
-        if i.stem.isdecimal() and i.suffix == '.txt':
-            with i.open('r', encoding='utf-8') as f:
-                app_id_ = f.read().strip()
-                depot_dict[int(i.stem)] = None
-                if app_id_.isdecimal():
-                    depot_dict[int(i.stem)] = int(app_id_)
-    for depot_id in depot_id_list:
-        if int(depot_id) not in depot_dict.values():
-            index = max(depot_dict.keys()) + 1 if depot_dict.keys() else 0
-            if index != 0:
-                for i in range(max(depot_dict.keys())):
-                    if i not in depot_dict.keys():
-                        index = i
-                        break
-            with (app_list_path / f'{index}.txt').open('w', encoding='utf-8') as f:
-                f.write(str(depot_id))
-            depot_dict[index] = int(depot_id)
-    return True
+
+        for file in app_list_path.glob('*.txt'):
+            file.unlink(missing_ok=True)
+
+        depot_dict = {
+            int(i.stem): int(i.read_text(encoding='utf-8').strip())
+            for i in app_list_path.iterdir() if i.is_file() and i.stem.isdecimal() and i.suffix == '.txt'
+        }
+
+        for depot_id in map(int, depot_id_list):
+            if depot_id not in depot_dict.values():
+                index = max(depot_dict.keys(), default=-1) + 1
+                while index in depot_dict:
+                    index += 1
+
+                (app_list_path / f'{index}.txt').write_text(str(depot_id), encoding='utf-8')
+
+                depot_dict[index] = depot_id
+
+        return True
+
+    except Exception as e:
+        print(f'❗ 处理时出错: {e}')
+        return False
--- a/common/init_text.py
+++ b/common/init_text.py
@@ -5,15 +5,20 @@ init()
 from .log import log

 def init():
-    print(f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT}  _____   __   _   _____   _   _    _____  __    __ {Style.RESET_ALL}")
-    print(f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} /  _  \\ |  \\ | | | ____| | | / /  | ____| \\ \\  / /{Style.RESET_ALL}")
-    print(f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} | | | | |   \\| | | |__   | |/ /   | |__    \\ \\/ /{Style.RESET_ALL}")
-    print(f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} | | | | | |\\   | |  __|  | |\\ \\   |  __|    \\  / {Style.RESET_ALL}")
-    print(f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} | |_| | | | \\  | | |___  | | \\ \\  | |___    / /{Style.RESET_ALL}")
-    print(f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} \\_____/ |_|  \\_| |_____| |_|  \\_\\ |_____|  /_/{Style.RESET_ALL}")
+    banner_lines = [
+        f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT}  _____   __   _   _____   _   _    _____  __    __ {Style.RESET_ALL}",
+        f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} /  _  \\ |  \\ | | | ____| | | / /  | ____| \\ \\  / /{Style.RESET_ALL}",
+        f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} | | | | |   \\| | | |__   | |/ /   | |__    \\ \\/ /{Style.RESET_ALL}",
+        f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} | | | | | |\\   | |  __|  | |\\ \\   |  __|    \\  / {Style.RESET_ALL}",
+        f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} | |_| | | | \\  | | |___  | | \\ \\  | |___    / /{Style.RESET_ALL}",
+        f"{Fore.GREEN}{Back.BLACK}{Style.BRIGHT} \\_____/ |_|  \\_| |_____| |_|  \\_\\ |_____|  /_/{Style.RESET_ALL}",
+    ]
+    for line in banner_lines:
+        print(line)
+    
    log.info('作者：ikun0014')
    log.info('本项目采用GNU General Public License v3开源许可证')
-    log.info('版本：1.2.3')
+    log.info('版本：1.2.4')
    log.info('项目仓库：https://github.com/ikunshare/Onekey')
    log.info('官网：ikunshare.com')
    log.warning('本项目完全开源免费，如果你在淘宝，QQ群内通过购买方式获得，赶紧回去骂商家死全家\n交流群组：\nhttps://qm.qq.com/q/d7sWovfAGI\nhttps://t.me/ikunshare_group')
--- a/common/log.py
+++ b/common/log.py
@@ -1,22 +1,28 @@
 import logging
 import colorlog

+LOG_FORMAT = '%(log_color)s[%(name)s][%(levelname)s]%(message)s'
+LOG_COLORS = {
+    'INFO': 'cyan',
+    'WARNING': 'yellow',
+    'ERROR': 'red',
+    'CRITICAL': 'purple',
+}

-def init_log():
+def init_log(level=logging.DEBUG) -> logging.Logger:
    logger = logging.getLogger('Onekey')
-    logger.setLevel(logging.DEBUG)
+    logger.setLevel(level)
+
    stream_handler = logging.StreamHandler()
-    stream_handler.setLevel(logging.DEBUG)
-    fmt_string = '%(log_color)s[%(name)s][%(levelname)s]%(message)s'
-    log_colors = {
-        'INFO': 'cyan',
-        'WARNING': 'yellow',
-        'ERROR': 'red',
-        'CRITICAL': 'purple'
-    }
-    fmt = colorlog.ColoredFormatter(fmt_string, log_colors=log_colors)
+    stream_handler.setLevel(level)
+
+    fmt = colorlog.ColoredFormatter(LOG_FORMAT, log_colors=LOG_COLORS)
    stream_handler.setFormatter(fmt)
-    logger.addHandler(stream_handler)
+
+    # 避免重复添加处理器
+    if not logger.handlers:
+        logger.addHandler(stream_handler)
+
    return logger


--- a/common/main_func.py
+++ b/common/main_func.py
@@ -1,5 +1,4 @@
 import os
-
 from aiohttp import ClientSession

 from common.config import config
@@ -16,65 +15,78 @@ from common.stack_error import stack_error
 isGreenLuma = any((steam_path / dll).exists() for dll in ['GreenLuma_2024_x86.dll', 'GreenLuma_2024_x64.dll', 'User32.dll'])
 isSteamTools = (steam_path / 'config' / 'stUI').is_dir()

-async def main(app_id, repos):
+async def fetch_branch_info(session, url, headers):
+    try:
+        async with session.get(url, headers=headers, ssl=False) as response:
+            return await response.json()
+    except Exception as e:
+        log.error(f'⚠ 获取信息失败: {stack_error(e)}')
+        return None
+
+async def get_latest_repo_info(session, repos, app_id, headers):
+    latest_date = None
+    selected_repo = None
+
+    for repo in repos:
+        url = f'https://api.github.com/repos/{repo}/branches/{app_id}'
+        r_json = await fetch_branch_info(session, url, headers)
+        if r_json and 'commit' in r_json:
+            date = r_json['commit']['commit']['author']['date']
+            if latest_date is None or date > latest_date:
+                latest_date = date
+                selected_repo = repo
+
+    return selected_repo, latest_date
+
+async def main(app_id: str, repos: list) -> bool:
    app_id_list = list(filter(str.isdecimal, app_id.strip().split('-')))
    if not app_id_list:
-        log.error(f'⚠  App ID无效')
+        log.error(f'⚠ App ID无效')
        return False
-    else:
-        app_id = app_id_list[0]
-    
+    app_id = app_id_list[0]
+
    async with ClientSession() as session:
-        github_token = config["Github_Personal_Token"]
+        github_token = config.get("Github_Personal_Token", "")
        headers = {'Authorization': f'Bearer {github_token}'} if github_token else None
-        latest_date = None
-        selected_repo = None
        await check_github_api_rate_limit(headers, session)
-        for repo in repos:
-            url = f'https://api.github.com/repos/{repo}/branches/{app_id}'
-            try:
-                async with session.get(url, headers=headers, ssl=False) as r:
-                    r_json = await r.json()
-                    if 'commit' in r_json:
-                        date = r_json['commit']['commit']['author']['date']
-                        if latest_date is None or date > latest_date:
-                            latest_date = date
-                            selected_repo = repo
-            except Exception as e:
-                log.error(f'⚠ 获取分支信息失败: {stack_error(e)}')
+
+        selected_repo, latest_date = await get_latest_repo_info(session, repos, app_id, headers)

        if selected_repo:
-            log.info(f' 🔄 选择清单仓库：{selected_repo}')
+            log.info(f'🔄 选择清单仓库：{selected_repo}')
            url = f'https://api.github.com/repos/{selected_repo}/branches/{app_id}'
-            async with session.get(url, headers=headers, ssl=False) as r:
-                r_json = await r.json()
-                if 'commit' in r_json:
-                    sha = r_json['commit']['sha']
-                    url = r_json['commit']['commit']['tree']['url']
-                    async with session.get(url, headers=headers, ssl=False) as r2:
-                        r2_json = await r2.json()
-                        if 'tree' in r2_json:
-                            collected_depots = []
-                            for i in r2_json['tree']:
-                                result = await get_manifest(sha, i['path'], steam_path, selected_repo, session)
-                                collected_depots.extend(result)
-                            if collected_depots:
-                                if isSteamTools:
-                                    await migrate(st_use=True, session=session)
-                                    await stool_add(collected_depots, app_id)
-                                    log.info(' ✅ 找到SteamTools，已添加解锁文件')
-                                if isGreenLuma:
-                                    await migrate(st_use=False, session=session)
-                                    await greenluma_add([app_id])
-                                    depot_config = {'depots': {depot_id: {'DecryptionKey': depot_key} for depot_id, depot_key in collected_depots}}
-                                    await depotkey_merge(steam_path / 'config' / 'config.vdf', depot_config)
-                                    if await greenluma_add([int(i) for i in depot_config['depots'] if i.isdecimal()]):
-                                        log.info('✅ 找到GreenLuma，已添加解锁文件')
-                                log.info(f'✅ 清单最后更新时间：{date}')
-                                log.info(f'✅ 入库成功: {app_id}')
-                                os.system('pause')
-                                return True
-                            
+            r_json = await fetch_branch_info(session, url, headers)
+
+            if r_json and 'commit' in r_json:
+                sha = r_json['commit']['sha']
+                url = r_json['commit']['commit']['tree']['url']
+                r2_json = await fetch_branch_info(session, url, headers)
+                
+                if r2_json and 'tree' in r2_json:
+                    collected_depots = []
+                    for item in r2_json['tree']:
+                        result = await get_manifest(sha, item['path'], steam_path, selected_repo, session)
+                        collected_depots.extend(result)
+
+                    if collected_depots:
+                        if isSteamTools:
+                            await migrate(st_use=True, session=session)
+                            await stool_add(collected_depots, app_id)
+                            log.info('✅ 找到SteamTools，已添加解锁文件')
+
+                        if isGreenLuma:
+                            await migrate(st_use=False, session=session)
+                            await greenluma_add([app_id])
+                            depot_config = {'depots': {depot_id: {'DecryptionKey': depot_key} for depot_id, depot_key in collected_depots}}
+                            await depotkey_merge(steam_path / 'config' / 'config.vdf', depot_config)
+                            if await greenluma_add([int(i) for i in depot_config['depots'] if i.isdecimal()]):
+                                log.info('✅ 找到GreenLuma，已添加解锁文件')
+
+                        log.info(f'✅ 清单最后更新时间：{latest_date}')
+                        log.info(f'✅ 入库成功: {app_id}')
+                        os.system('pause')
+                        return True
+            
        log.error(f'⚠ 清单下载或生成失败: {app_id}')
        os.system('pause')
-        return False
+        return False
--- a/common/manifest_down.py
+++ b/common/manifest_down.py
@@ -1,38 +1,55 @@
-from aiohttp import ClientError
+from aiohttp import ClientError, ClientResponse
 from tqdm.asyncio import tqdm_asyncio
+from typing import Union

 from .log import log


-async def get(sha, path, repo, session):
+async def get(sha: str, path: str, repo: str, session, chunk_size: int = 1024) -> bytearray:
    url_list = [
+        f'https://jsdelivr.pai233.top/gh/{repo}@{sha}/{path}',
        f'https://cdn.jsdmirror.com/gh/{repo}@{sha}/{path}',
-        f'https://jsd.onmicrosoft.cn/gh/{repo}@{sha}/{path}',
+        f'https://raw.kkgithub.com/{repo}/{sha}/{path}',
        f'https://raw.dgithub.xyz/{repo}/{sha}/{path}',
-        f'https://raw.gitmirror.com/{repo}/{sha}/{path}',
        f'https://raw.githubusercontent.com/{repo}/{sha}/{path}'
    ]
+    '''
+    下载时间 (20MB 从小到大):
+    https://jsdelivr.pai233.top/gh/{repo}@{sha}/{path} - 0.95秒
+    https://cdn.jsdmirror.com/gh/{repo}@{sha}/{path} - 6.74秒
+    https://raw.kkgithub.com/{repo}/{sha}/{path} - 6.76秒
+    https://raw.dgithub.xyz/{repo}/{sha}/{path} - 8.30秒
+    https://raw.gitmirror.com/{repo}/{sha}/{path} - 15.60秒
+    https://ghproxy.net/https://raw.githubusercontent.com/{repo}/{sha}/{path} - 16.59秒
+    https://fastly.jsdelivr.net/gh/{repo}@{sha}/{path} - 20.08秒
+    https://jsd.onmicrosoft.cn/gh/{repo}@{sha}/{path} - 22.07秒
+    https://gitdl.cn/https://raw.githubusercontent.com/{repo}/{sha}/{path} - 47.33秒
+    https://ghp.ci/https://raw.githubusercontent.com/{repo}/{sha}/{path} - 96.56秒
+    https://raw.githubusercontent.com/{repo}/{sha}/{path} - 458.75秒
+    https://cdn.jsdelivr.net/gh/{repo}@{sha}/{path} - 下载时出错
+    '''
    retry = 3
-    while retry:
+    while retry > 0:
        for url in url_list:
            try:
-                # log.debug(f"{url}")
-                async with session.get(url, ssl=False) as r:
-                    if r.status == 200:
-                        total_size = int(r.headers.get('Content-Length', 0))
-                        chunk_size = 1024
+                async with session.get(url, ssl=False) as response:
+                    if response.status == 200:
+                        total_size = int(response.headers.get('Content-Length', 0))
                        content = bytearray()

                        with tqdm_asyncio(total=total_size, unit='B', unit_scale=True, desc=f'下载 {path}', colour='#ffadad') as pbar:
-                            async for chunk in r.content.iter_chunked(chunk_size):
+                            async for chunk in response.content.iter_chunked(chunk_size):
                                content.extend(chunk)
                                pbar.update(len(chunk))
+                        
                        return content
                    else:
-                        log.error(f'🔄 获取失败: {path} - 状态码: {r.status}')
-            except ClientError:
-                log.error(f'🔄 获取失败: {path} - 连接错误')
+                        log.error(f'🔄 获取失败: {path} - 状态码: {response.status}')
+            except ClientError as e:
+                log.error(f'🔄 获取失败: {path} - 连接错误: {str(e)}')
+        
        retry -= 1
        log.warning(f'🔄 重试剩余次数: {retry} - {path}')
+    
    log.error(f'🔄 超过最大重试次数: {path}')
    raise Exception(f'🔄 无法下载: {path}')
--- a/common/migration.py
+++ b/common/migration.py
@@ -3,52 +3,58 @@ import subprocess
 import aiofiles
 from pathlib import Path
 from tqdm.asyncio import tqdm
-
 from .log import log
 from .get_steam_path import steam_path

-directory = Path(steam_path / "config" / "stplug-in")
+directory = Path(steam_path) / "config" / "stplug-in"
+temp_path = Path('./temp')
+setup_url = 'https://steamtools.net/res/SteamtoolsSetup.exe'
+setup_file = temp_path / 'SteamtoolsSetup.exe'

-async def migrate(st_use, session):
-    if st_use == True:
-        log.info('🔄 检测到你正在使用SteamTools，尝试迁移旧文件')
-        if os.path.exists(directory):
-            for filename in os.listdir(directory):
-                if filename.startswith("Onekey_unlock_"):
-                    new_filename = filename[len("Onekey_unlock_"):]
+async def download_setup_file(session) -> None:
+    log.info('🔄 开始下载 SteamTools 安装程序...')
+    try:
+        async with session.get(setup_url, stream=True) as r:
+            if r.status == 200:
+                total_size = int(r.headers.get('Content-Length', 0))
+                chunk_size = 8192
+                progress = tqdm(total=total_size, unit='B', unit_scale=True, desc='下载安装程序')

-                    old_file = os.path.join(directory, filename)
-                    new_file = os.path.join(directory, new_filename)
+                async with aiofiles.open(setup_file, mode='wb') as f:
+                    async for chunk in r.content.iter_chunked(chunk_size):
+                        await f.write(chunk)
+                        progress.update(len(chunk))
+
+                progress.close()
+                log.info('✅ 安装程序下载完成')
+            else:
+                log.error('⚠ 网络错误，无法下载安装程序')
+    except Exception as e:
+        log.error(f'⚠ 下载失败: {e}')
+
+async def migrate(st_use: bool, session) -> None:
+    if st_use:
+        log.info('🔄 检测到你正在使用 SteamTools，尝试迁移旧文件')
+
+        if directory.exists():
+            for file in directory.iterdir():
+                if file.is_file() and file.name.startswith("Onekey_unlock_"):
+                    new_filename = file.name[len("Onekey_unlock_"):]

                    try:
-                        os.replace(old_file, new_file)
-                        log.info(f'Renamed: {filename} -> {new_filename}')
+                        file.rename(directory / new_filename)
+                        log.info(f'Renamed: {file.name} -> {new_filename}')
                    except Exception as e:
-                        log.error(f'Failed to rename {filename} -> {new_filename}: {e}')
+                        log.error(f'⚠ 重命名失败 {file.name} -> {new_filename}: {e}')
        else:
-            log.error('⚠ 故障，正在重新安装SteamTools')
-            temp_path = './temp'
-            if not os.path.exists(temp_path):
-                os.mkdir(temp_path)
-            down_url = 'https://steamtools.net/res/SteamtoolsSetup.exe'
-            out_path = './temp/SteamtoolsSetup.exe'
+            log.error('⚠ 故障，正在重新安装 SteamTools')
+            temp_path.mkdir(parents=True, exist_ok=True)

-            async with session.get(down_url, stream=True) as r:
-                if r.status == 200:
-                    total_size = int(await  r.headers.get('Content-Length', 0))
-                    chunk_size = 8192
-                    progress = tqdm(total=total_size, unit='B', unit_scale=True)
+            await download_setup_file(session)

-                    async with aiofiles.open(out_path, mode='wb') as f:
-                        async for chunk in r.content.iter_chunked(chunk_size=chunk_size):
-                            await f.write(chunk)
-                            await progress.update(len(chunk))
-
-                    await progress.close()
-                else:
-                    log.error('⚠ 网络错误')
-
-            subprocess.run(str(out_path))
-            os.rmdir(temp_path)
+            subprocess.run(str(setup_file), check=True)
+            for file in temp_path.iterdir():
+                file.unlink()
+            temp_path.rmdir()
    else:
-        log.info('✅ 未使用SteamTools，停止迁移')
+        log.info('✅ 未使用 SteamTools，停止迁移')
--- a/common/st_unlock.py
+++ b/common/st_unlock.py
@@ -2,24 +2,43 @@ import os
 import asyncio
 import subprocess
 import aiofiles
+from pathlib import Path

 from .log import log
 from .get_steam_path import steam_path

 lock = asyncio.Lock()

-async def stool_add(depot_data, app_id):
+async def stool_add(depot_data: list, app_id: str) -> bool:
    lua_filename = f"{app_id}.lua"
    lua_filepath = steam_path / "config" / "stplug-in" / lua_filename

    async with lock:
-        log.info(f'✅ SteamTools解锁文件生成: {lua_filepath}')
-        async with aiofiles.open(lua_filepath, mode="w", encoding="utf-8") as lua_file:
-            await lua_file.write(f'addappid({app_id}, 1, "None")\n')
-            for depot_id, depot_key in depot_data:
-                await lua_file.write(f'addappid({depot_id}, 1, "{depot_key}")\n')
+        log.info(f'✅ SteamTools 解锁文件生成: {lua_filepath}')
+        try:
+            async with aiofiles.open(lua_filepath, mode="w", encoding="utf-8") as lua_file:
+                await lua_file.write(f'addappid({app_id}, 1, "None")\n')
+                for depot_id, depot_key in depot_data:
+                    await lua_file.write(f'addappid({depot_id}, 1, "{depot_key}")\n')

-    luapacka_path = steam_path / "config" / "stplug-in" / "luapacka.exe"
-    subprocess.run([str(luapacka_path), str(lua_filepath)])
-    os.remove(lua_filepath)
-    return True
+            luapacka_path = steam_path / "config" / "stplug-in" / "luapacka.exe"
+            log.info(f'🔄 正在处理文件: {lua_filepath}')
+
+            result = subprocess.run(
+                [str(luapacka_path), str(lua_filepath)], 
+                capture_output=True
+            )
+            if result.returncode != 0:
+                log.error(f'⚠ 调用失败: {result.stderr.decode()}')
+                return False
+
+            log.info('✅ 处理完成')
+        except Exception as e:
+            log.error(f'❗ 处理过程出现错误: {e}')
+            return False
+        finally:
+            if lua_filepath.exists():
+                os.remove(lua_filepath)
+                log.info(f'🗑️ 删除临时文件: {lua_filepath}')
+
+    return True
--- a/common/stack_error.py
+++ b/common/stack_error.py
@@ -1,5 +1,5 @@
 import traceback

-def stack_error(exception):
+def stack_error(exception: Exception) -> str:
    stack_trace = traceback.format_exception(type(exception), exception, exception.__traceback__)
-    return ''.join(stack_trace)
+    return ''.join(stack_trace)
--- a/main.py
+++ b/main.py
@@ -1,6 +1,8 @@
 import os
 import time
+import sys
 import asyncio
+import re

 from colorama import Fore, Back, Style
 from colorama import init as cinit
@@ -15,20 +17,40 @@ init()
 cinit()

 repos = [
-         'ikun0014/ManifestHub',
-         'Auiowu/ManifestAutoUpdate',
-         'tymolu233/ManifestAutoUpdate',
-        ]
+    'ikun0014/ManifestHub',
+    'Auiowu/ManifestAutoUpdate',
+    'tymolu233/ManifestAutoUpdate',
+]
+
+def prompt_app_id():
+        app_id = input(f"{Fore.CYAN}{Back.BLACK}{Style.BRIGHT}🤔 请输入游戏AppID：{Style.RESET_ALL}").strip()
+        if re.match(r'^\d+$', app_id):
+            return app_id
+        else:
+            print(f"{Fore.RED}⚠ 无效的AppID，请输入数字！{Style.RESET_ALL}")
+
+async def main_loop():
+    while True:
+        try:
+            app_id = prompt_app_id()
+            await main(app_id, repos)
+        except EOFError:
+            break
+
+async def run():
+    try:
+        log.info('❗ App ID可以在SteamDB或Steam商店链接页面查看')
+        await main_loop()
+    except KeyboardInterrupt:
+        log.info("👋 程序已退出")
+    except Exception as e:
+        log.error(f' ⚠ 发生错误: {stack_error(e)}，将在5秒后退出')
+        await asyncio.sleep(5)
+    finally:
+        asyncio.get_event_loop().stop()

 if __name__ == '__main__':
    try:
-        log.info('❗ App ID可以在SteamDB或Steam商店链接页面查看')
-        while True:
-            app_id = input(f"{Fore.CYAN}{Back.BLACK}{Style.BRIGHT}🤔 请输入游戏AppID：{Style.RESET_ALL}").strip()
-            asyncio.run(main(app_id, repos))
-    except KeyboardInterrupt:
-        exit()
-    except Exception as e:
-        log.error(f' ⚠ 发生错误: {stack_error(e)}，将在5秒后退出')
-        time.sleep(5)
-        os.system('pause')
+        asyncio.run(run())
+    except SystemExit:
+        sys.exit()
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
  "name": "onekey",
-  "version": "1.2.3",
+  "version": "1.2.4",
  "description": "一个Steam仓库清单下载器",
  "main": "index.js",
  "scripts": {