revert: ui

ikunshare · Oct 19, 2024 · 5d20b57 · 5d20b57
1 parent 015e96c
commit 5d20b57
Show file tree

Hide file tree

Showing 15 changed files with 142 additions and 119 deletions.
diff --git a/README.md b/README.md
@@ -11,10 +11,10 @@
 
 
 ## Onekey
-Onekey Steam Depot Manifest Downloader
+ Onekey Steam Depot Manifest Downloader
 
 ## 使用方法
-先去Release下最新发布，然后去steamtools官网下steamtools，日志会有点石介意别用
+ 先去Release下最新发布，然后去steamtools官网下steamtools，日志会有点石介意别用
 
 ## 开发
 本程序使用Python编程语言开发 
@@ -36,36 +36,35 @@ pip install -r requirements.txt
 ```
 
 ## 项目协议
-本项目基于 GPL-3.0 许可证发行，以下协议是对于 GPL-3.0 原协议的补充，如有冲突，以以下协议为准。
+ 本项目基于 GPL-3.0 许可证发行，以下协议是对于 GPL-3.0 原协议的补充，如有冲突，以以下协议为准。
 
-词语约定：“使用者”指签署本协议的使用者;“版权数据”指包括但不限于图像、音频、名字等在内的他人拥有所属版权的数据。
+ 词语约定：“使用者”指签署本协议的使用者;“版权数据”指包括但不限于图像、音频、名字等在内的他人拥有所属版权的数据。
 
-本项目的数据来源原理是从Steam官方的CDN服务器中拉取游戏清单数据，经过对数据简单地筛选与合并后进行展示，因此本项目不对数据的准确性负责。
-使用本项目的过程中可能会产生版权数据，对于这些版权数据，本项目不拥有它们的所有权，为了避免造成侵权，使用者务必在24 小时内清除使用本项目的过程中所产生的版权数据。
-由于使用本项目产生的包括由于本协议或由于使用或无法使用本项目而引起的任何性质的任何直接、间接、特殊、偶然或结果性损害（包括但不限于因商誉损失、停工、计算机故障或故障引起的损害赔偿，或任何及所有其他商业损害或损失）由使用者负责。
-本项目完全免费，且开源发布于 GitHub 面向全世界人用作对技术的学习交流，本项目不对项目内的技术可能存在违反当地法律法规的行为作保证，禁止在违反当地法律法规的情况下使用本项目，对于使用者在明知或不知当地法律法规不允许的情况下使用本项目所造成的任何违法违规行为由使用者承担，本项目不承担由此造成的任何直接、间接、特殊、偶然或结果性责任。
-而且，本项目已禁止使用于商业用途，以及不得进行未经允许的二次修改，否则必须同时发布源代码。
-若你使用了本项目，将代表你接受以上协议。
+ 本项目的数据来源原理是从Steam官方的CDN服务器中拉取游戏清单数据，经过对数据简单地筛选与合并后进行展示，因此本项目不对数据的准确性负责。
+ 使用本项目的过程中可能会产生版权数据，对于这些版权数据，本项目不拥有它们的所有权，为了避免造成侵权，使用者务必在24 小时内清除使用本项目的过程中所产生的版权数据。
+ 由于使用本项目产生的包括由于本协议或由于使用或无法使用本项目而引起的任何性质的任何直接、间接、特殊、偶然或结果性损害（包括但不限于因商誉损失、停工、计算机故障或故障引起的损害赔偿，或任何及所有其他商业损害或损失）由使用者负责。
+ 本项目完全免费，且开源发布于 GitHub 面向全世界人用作对技术的学习交流，本项目不对项目内的技术可能存在违反当地法律法规的行为作保证，禁止在违反当地法律法规的情况下使用本项目，对于使用者在明知或不知当地法律法规不允许的情况下使用本项目所造成的任何违法违规行为由使用者承担，本项目不承担由此造成的任何直接、间接、特殊、偶然或结果性责任。
+ 而且，本项目已禁止使用于商业用途，以及不得进行未经允许的二次修改，否则必须同时发布源代码。
+ 若你使用了本项目，将代表你接受以上协议。
 
-Steam正版平台不易，请尊重版权，支持正版。 
-本项目仅用于对技术可行性的探索及研究，不接受任何商业（包括但不限于广告等）合作。 
+ Steam正版平台不易，请尊重版权，支持正版。 
+ 本项目仅用于对技术可行性的探索及研究，不接受任何商业（包括但不限于广告等）合作。 
 
 ## Star 趋势图
 
-[![Stargazers over time](https://starchart.cc/ikunshare/Onekey.svg)](https://starchart.cc/ikunshare/Onekey)
+ [![Stargazers over time](https://starchart.cc/ikunshare/Onekey.svg)](https://starchart.cc/ikunshare/Onekey)
 
 ## 贡献者
 
-<a href="https://github.com/ikunshare/Onekey/graphs/contributors">
- <img src="https://contrib.rocks/image?repo=ikunshare/Onekey" />
-</a>
+ <a href="https://github.com/ikunshare/Onekey/graphs/contributors">
+  <img src="https://contrib.rocks/image?repo=ikunshare/Onekey" />
+ </a>
 
 ## 常见问题解答（FAQ）
 查看 [FAQ](https://ikunshare.com/d/49) 获取常见问题的解答。
 
 ## 社区和支持
 加入我们的社区，参与讨论和支持：
-- [ikun论坛](https://ikunshare.com)
 - [GitHub Discussions](https://github.com/ikunshare/Onekey/discussions)
+- [QQ](https://qm.qq.com/q/d7sWovfAGI)
 - [Telegram](https://t.me/ikunshare_qun)
-- [Telegram Channel](https://t.me/ikunshare_channel)
diff --git a/common/check.py b/common/check.py
@@ -4,8 +4,10 @@
 from .stack_error import stack_error
 
 async def check_github_api_rate_limit(headers, session):
+ url = 'https://api.github.com/rate_limit'
+
  try:
- async with session.get('https://api.github.com/rate_limit', headers=headers, ssl=False) as r:
+ async with session.get(url, headers=headers, ssl=False) as r:
  r_json = await r.json()
 
  if r.status == 200:
@@ -14,17 +16,18 @@ async def check_github_api_rate_limit(headers, session):
  reset_time = rate_limit.get('reset', 0)
  reset_time_formatted = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(reset_time))
 
- log(f'🔄 剩余请求次数: {remaining_requests}')
+ log.info(f'🔄 剩余请求次数: {remaining_requests}')
 
  if remaining_requests == 0:
- log(f'⚠ GitHub API 请求数已用尽, 将在 {reset_time_formatted} 重置,建议生成一个填在配置文件里')
+ log.warning(f'⚠ GitHub API 请求数已用尽, 将在 {reset_time_formatted} 重置,建议生成一个填在配置文件里')
  else:
- log('⚠ Github请求数检查失败, 网络错误')
+ log.error('⚠ Github请求数检查失败, 网络错误')
+
  except KeyboardInterrupt:
- log("👋 程序已退出")
+ log.info("\n👋 程序已退出")
  except ClientError as e:
- log(f'⚠ 检查Github API 请求数失败,{stack_error(e)}')
+ log.error(f'⚠ 检查Github API 请求数失败,{stack_error(e)}')
  except ConnectionTimeoutError as e:
- log(f'⚠ 检查Github API 请求数超时: {stack_error(e)}')
+ log.error(f'⚠ 检查Github API 请求数超时: {stack_error(e)}')
  except Exception as e:
- log(f'⚠ 发生错误: {stack_error(e)}')
+ log.error(f'⚠ 发生错误: {stack_error(e)}')
diff --git a/common/checkcn.py b/common/checkcn.py
@@ -3,23 +3,24 @@
 from .log import log
 from .stack_error import stack_error
 
-def checkcn() -> bool:
+def checkcn():
  try:
  req = requests.get('https://mips.kugou.com/check/iscn?&format=json')
  req.raise_for_status()
  body = req.json()
  scn = bool(body['flag'])
  if not scn:
- log(f"❌ 您在非中国大陆地区({body['country']})上使用了项目, 已自动切换回Github官方下载CDN")
+ log.info(f"❌ 您在非中国大陆地区({body['country']})上使用了项目, 已自动切换回Github官方下载CDN")
  os.environ['IS_CN'] = 'no'
  return False
  else:
  os.environ['IS_CN'] = 'yes'
  return True
+
  except KeyboardInterrupt:
- log("👋 程序已退出")
+ log.info("\n👋 程序已退出")
  except requests.RequestException as e:
  os.environ['IS_CN'] = 'yes'
- log('❗ 检查服务器位置失败，已忽略，自动认为你在中国大陆')
- log(stack_error(e))
+ log.warning('❗ 检查服务器位置失败，已忽略，自动认为你在中国大陆')
+ log.warning(stack_error(e))
  return False
diff --git a/common/config.py b/common/config.py
@@ -9,33 +9,35 @@
 DEFAULT_CONFIG = {
  "Github_Personal_Token": "",
  "Custom_Steam_Path": "",
- "QA1": "温馨提示: Github_Personal_Token可在Github设置的最底下开发者选项找到, 详情看教程",
+ "QA1": "温馨提示：Github_Personal_Token可在Github设置的最底下开发者选项找到，详情看教程",
  "教程": "https://ikunshare.com/Onekey_tutorial"
 }
 
 async def gen_config_file():
  try:
  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('🖱️ 程序可能为第一次启动或配置重置,请填写配置文件后重新启动程序')
+
+ log.info('🖱️ 程序可能为第一次启动或配置重置,请填写配置文件后重新启动程序')
  except KeyboardInterrupt:
- log("👋 程序已退出")
+ log.info("\n👋 程序已退出")
  except Exception as e:
- log(f'❌ 配置文件生成失败,{stack_error(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()
+
  try:
  async with aiofiles.open("./config.json", mode="r", encoding="utf-8") as f:
  config = json.loads(await f.read())
  return config
  except KeyboardInterrupt:
- log("👋 程序已退出")
+ log.info("\n👋 程序已退出")
  except Exception as e:
- log(f"配置文件加载失败，原因: {stack_error(e)},重置配置文件中...")
+ log.error(f"配置文件加载失败，原因: {stack_error(e)},重置配置文件中...")
  os.remove("./config.json")
  await gen_config_file()
  os.system('pause')

diff --git a/common/dkey_merge.py b/common/dkey_merge.py
@@ -9,8 +9,9 @@
 async def depotkey_merge(config_path: Path, depots_config: dict) -> bool:
  if not config_path.exists():
  async with lock:
- log('👋 Steam默认配置不存在, 可能是没有登录账号')
+ log.error('👋 Steam默认配置不存在, 可能是没有登录账号')
  return False
+
  try:
  async with aiofiles.open(config_path, encoding='utf-8') as f:
  content = await f.read()
@@ -20,7 +21,7 @@ async def depotkey_merge(config_path: Path, depots_config: dict) -> bool:
  config.get('InstallConfigStore', {}).get('Software', {}).get('valve')
 
  if steam is None:
- log('⚠ 找不到Steam配置, 请检查配置文件')
+ log.error('⚠ 找不到Steam配置, 请检查配置文件')
  return False
 
  depots = steam.setdefault('depots', {})
@@ -30,11 +31,12 @@ async def depotkey_merge(config_path: Path, depots_config: dict) -> bool:
  new_context = vdf.dumps(config, pretty=True)
  await f.write(new_context)
 
- log('✅ 成功合并')
+ log.info('✅ 成功合并')
  return True
+
  except KeyboardInterrupt:
- log("👋 程序已退出")
+ log.info("\n👋 程序已退出")
  except Exception as e:
  async with lock:
- log(f'❌ 合并失败, 原因: {e}')
+ log.error(f'❌ 合并失败, 原因: {e}')
  return False
diff --git a/common/get_manifest_info.py b/common/get_manifest_info.py
@@ -17,18 +17,18 @@ async def get_manifest(sha: str, path: str, steam_path: Path, repo: str, session
  if path.endswith('.manifest'):
  save_path = depot_cache_path / path
  if save_path.exists():
- log(f'👋 已存在清单: {save_path}')
+ log.warning(f'👋 已存在清单: {save_path}')
  return collected_depots
 
  content = await get(sha, path, repo, session)
- log(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(f'🔄 密钥下载成功: {path}')
+ log.info(f'🔄 密钥下载成功: {path}')
 
  depots_config = vdf.loads(content.decode('utf-8'))
  collected_depots = [
@@ -37,9 +37,9 @@ async def get_manifest(sha: str, path: str, steam_path: Path, repo: str, session
  ]
 
  except KeyboardInterrupt:
- log("👋 程序已退出")
+ log.info("\n👋 程序已退出")
  except Exception as e:
- log(f'❌ 处理失败: {path} - {stack_error(e)}')
+ log.error(f'❌ 处理失败: {path} - {stack_error(e)}')
  raise
 
  return collected_depots
diff --git a/common/get_steam_path.py b/common/get_steam_path.py
@@ -13,9 +13,9 @@ def get_steam_path() -> Path:
  custom_steam_path = config.get("Custom_Steam_Path", "").strip()
  return Path(custom_steam_path) if custom_steam_path else steam_path
  except KeyboardInterrupt:
- log("👋 程序已退出")
+ log.info("👋 程序已退出")
  except Exception as e:
- log(f'❌ Steam路径获取失败, {stack_error(e)}, 请检查是否正确安装Steam')
+ log.error(f'❌ Steam路径获取失败, {stack_error(e)}, 请检查是否正确安装Steam')
  os.system('pause')
  return Path()
 

diff --git a/common/init_text.py b/common/init_text.py
@@ -1,27 +1,25 @@
-from terminaltexteffects.effects.effect_print import Print
+from colorama import Fore, Back, Style, init
+
+init()
 
 from .log import log
 
 def init():
- text = (
- " _____ __ _ _____ _ _ _____ __ __ \n"
- " / _ \\ | \\ | | | ____| | | / / | ____| \\ \\ / /\n"
- " | | | | | \\| | | |__ | |/ / | |__ \\ \\/ /\n"
- " | | | | | |\\ | | __| | |\\ \\ | __| \\ / \n"
- " | |_| | | | \\ | | |___ | | \\ \\ | |___ / / \n"
- " \\_____/ |_| \\_| |_____| |_| \\_\\ |_____| /_/ "
- )
-
- effect = Print(text)
-
- with effect.terminal_output() as terminal:
- for frame in effect:
- terminal.print(frame)
+ 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(f'作者: ikun0014')
- log(f'本项目采用GNU General Public License v3开源许可证，请勿用于商业用途')
- log('版本: 1.3.0')
- log(f'项目Github仓库: https://github.com/ikunshare/Onekey')
- log('官网: ikunshare.com')
- log('本项目完全开源免费, 如果你在淘宝, QQ群内通过购买方式获得, 赶紧回去骂商家死全家\n交流群组:\n https://t.me/ikunshare_qun')
- log('如果本项目中的Emoji(即表情包)无法正常显示, 请使用支持Emoji的终端(例如Windows Terminal)')
+ log.info(f'作者: {Fore.RED}{Back.BLACK}{Style.BRIGHT}ikun0014{Style.RESET_ALL}')
+ log.info(f'{Fore.RED}{Back.BLACK}{Style.BRIGHT}本项目采用GNU General Public License v3开源许可证，请勿用于商业用途{Style.RESET_ALL}')
+ log.info('版本: 1.2.9')
+ log.info(f'{Fore.RED}{Back.BLACK}{Style.BRIGHT}项目Github仓库: https://github.com/ikunshare/Onekey{Style.RESET_ALL}')
+ log.info('官网: ikunshare.com')
+ log.warning('本项目完全开源免费, 如果你在淘宝, QQ群内通过购买方式获得, 赶紧回去骂商家死全家\n交流群组:\n https://t.me/ikunshare_qun')
+ log.warning('如果本项目中的Emoji(即表情包)无法正常显示, 请使用支持Emoji的终端(例如Windows Terminal)')
diff --git a/common/log.py b/common/log.py
@@ -1,11 +1,29 @@
-from terminaltexteffects.effects.effect_print import Print
+import logging
+import colorlog
 
-def logger(text):
- effect = Print(text)
-
- with effect.terminal_output() as terminal:
- for frame in effect:
- terminal.print(frame)
+LOG_FORMAT = '%(log_color)s[%(name)s][%(levelname)s]%(message)s'
+LOG_COLORS = {
+ 'INFO': 'cyan',
+ 'WARNING': 'yellow',
+ 'ERROR': 'red',
+ 'CRITICAL': 'purple',
+}
 
+def init_log(level=logging.DEBUG) -> logging.Logger:
+ logger = logging.getLogger('Onekey')
+ logger.setLevel(level)
 
-log = logger
+ stream_handler = logging.StreamHandler()
+ stream_handler.setLevel(level)
+
+ fmt = colorlog.ColoredFormatter(LOG_FORMAT, log_colors=LOG_COLORS)
+ stream_handler.setFormatter(fmt)
+
+ # 避免重复添加处理器
+ if not logger.handlers:
+ logger.addHandler(stream_handler)
+
+ return logger
+
+
+log = init_log()