From bc48125ff8e41cb3459a5b48df28ab930a742236 Mon Sep 17 00:00:00 2001 From: SarContDeli <48789988+SaiCateDoan@users.noreply.github.com> Date: Thu, 2 Feb 2023 21:36:32 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20Python=E6=8E=A5=E5=8F=A3=E7=9A=84MAA?= =?UTF-8?q?=E6=9B=B4=E6=96=B0=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/Python/asst.py | 279 +++++++++++++++++++++++++++++++++++++++++-- src/Python/sample.py | 5 +- 2 files changed, 274 insertions(+), 10 deletions(-) diff --git a/src/Python/asst.py b/src/Python/asst.py index bcfcb6861c..d49ab94640 100644 --- a/src/Python/asst.py +++ b/src/Python/asst.py @@ -1,11 +1,19 @@ -import ctypes import os -import pathlib -import platform import json +import ctypes +import pathlib +import zipfile +import platform +import multiprocessing -from typing import Union, Dict, List, Any, Type, Optional +from shutil import rmtree +from urllib import request +from urllib.error import * +from urllib.request import Request +from multiprocessing import queues +from multiprocessing import Process from enum import Enum, IntEnum, unique, auto +from typing import Union, Dict, List, Any, Type, Optional JSON = Union[Dict[str, Any], List[Any], int, str, float, bool, Type[None]] @@ -28,7 +36,8 @@ class Asst: """ @staticmethod - def load(path: Union[pathlib.Path, str], incremental_path: Optional[Union[pathlib.Path, str]] = None, user_dir: Optional[Union[pathlib.Path, str]] = None) -> bool: + def load(path: Union[pathlib.Path, str], incremental_path: Optional[Union[pathlib.Path, str]] = None, + user_dir: Optional[Union[pathlib.Path, str]] = None) -> bool: """ 加载 dll 及资源 @@ -57,6 +66,12 @@ class Asst: platform_type = platform.system().lower() if platform_type == 'windows': lib_import_func = ctypes.WinDLL + # 手动加载onnxruntime.dll以避免部分版本的python错误地从System32加载旧版本 + try: + lib_import_func(str(pathlib.Path(path) / 'onnxruntime.dll')) + except Exception as e: + print(e) + pass else: lib_import_func = ctypes.CDLL @@ -109,7 +124,6 @@ class Asst: return Asst.__lib.AsstSetInstanceOption(self.__ptr, int(option_type), option_value.encode('utf-8')) - def connect(self, adb_path: str, address: str, config: str = 'General'): """ 连接设备 @@ -136,7 +150,8 @@ class Asst: :return: 任务 ID, 可用于 set_task_params 接口 """ - return Asst.__lib.AsstAppendTask(self.__ptr, type_name.encode('utf-8'), json.dumps(params, ensure_ascii=False).encode('utf-8')) + return Asst.__lib.AsstAppendTask(self.__ptr, type_name.encode('utf-8'), + json.dumps(params, ensure_ascii=False).encode('utf-8')) def set_task_params(self, task_id: TaskId, params: JSON) -> bool: """ @@ -176,13 +191,13 @@ class Asst: @staticmethod def log(level: str, message: str) -> None: - ''' + """ 打印日志 :params: ``level``: 日志等级标签 ``message``: 日志内容 - ''' + """ Asst.__lib.AsstLog(level.encode('utf-8'), message.encode('utf-8')) @@ -279,3 +294,249 @@ class Message(Enum): SubTaskExtraInfo = auto() SubTaskStopped = auto() + + +@unique +class Version(Enum): + """ + 目标版本 + """ + Nightly = auto() + + Beta = auto() + + Stable = auto() + + +class Updater: + headers = { + 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) ' + 'AppleWebKit/537.36 (KHTML, like Gecko) ' + 'Chrome/97.0.4692.99 ' + 'Safari/537.36 ' + 'Edg/97.0.1072.76' + } + RequestUrl = "repos/MaaAssistantArknights/MaaRelease/releases" + StableRequestUrl = "repos/MaaAssistantArknights/MaaAssistantArknights/releases/latest" + MaaReleaseRequestUrlByTag = "repos/MaaAssistantArknights/MaaRelease/releases/tags/" + InfoRequestUrl = "repos/MaaAssistantArknights/MaaAssistantArknights/releases/tags/" + + @staticmethod + def custom_print(s): + """ + 可以被monkey patch的print，在其他GUI上使用可以被替换为任何需要的输出 + """ + print(s) + + @staticmethod + def _get_cur_version(path, q): + """ + 获取当前版本 + """ + Asst.load(path=path) + q.put(Asst().get_version()) + + def __init__(self, path, version): + self.path = path + self.version = version + self.latest_json = None + self.latest_version = None + self.assets_object = None + + # 使用子线程获取当前版本后关闭，避免占用dll + q = queues.Queue(1, ctx=multiprocessing) + p = Process(target=self._get_cur_version, args=(path, q,)) + p.start() + p.join() + self.cur_version = q.get() + + @staticmethod + def _request_github_api(url, retry): + request_resource = ["https://api.github.com/", "https://api.kgithub.com/"] + for _ in range(retry): + for resource in request_resource: + try: + response = request.urlopen(Request(url=resource + url, headers=Updater.headers), timeout=20) + data = response.read().decode('utf-8') + Updater.custom_print(f'访问成功，URL: {resource + url}') + return data + except (HTTPError, URLError) as e: + Updater.custom_print(f'访问成功，URL: {resource + url}') + Updater.custom_print(e) + if _ == retry - 1: + raise + + def _is_nightly_version(self, ver: Union[str, None]): + if ver is None: + ver = self.cur_version + + if '-' not in ver: + return False + last_id = ver.split('.')[-1] + return last_id.startswith('g') and len(last_id) >= 7 + + def _is_std_version(self, ver: Union[str, None]): + if ver is None: + ver = self.cur_version + + if ver == 'DEBUG VERSION': + return False + elif ver.startswith('c') or ver.startswith('20') or 'local' in ver: + return False + elif self._is_nightly_version(ver): + return False + return True + + @staticmethod + def _split_version(ver: str): + if '-' in ver: + pre, sub = ver.split('-', 1) + else: + pre = ver + sub = None + pre = pre.split('.') + pre[0] = pre[0][1:] + if sub: + sub = sub.split('.') + sub = sub[:-1] + if sub[0] == 'alpha': + sub[0] = '1' + elif sub[0] == 'beta': + sub[0] = '2' + elif sub[0] == 'rc': + sub[0] = '3' + out = pre + sub + else: + out = pre + for i in range(len(out)): + out[i] = int(out[i]) + return out + + @staticmethod + def _compare_version(a, b): + a = Updater._split_version(a) + b = Updater._split_version(b) + for i in range(len(a)): + try: + if a[i] < b[i]: + return True + elif a[i] > b[i]: + return False + except IndexError: + return False + if len(b) > len(a): + return True + return False + + def _check_update(self): + max_retry = 2 + if self.version == Version.Stable: + stable_response = self._request_github_api(self.StableRequestUrl, max_retry) + if len(stable_response) == 0: + raise HTTPError + + self.latest_json = json.loads(stable_response) + self.latest_version = self.latest_json['tag_name'] + + stable_response = self._request_github_api(self.MaaReleaseRequestUrlByTag + self.latest_version, max_retry) + if len(stable_response) == 0: + return False + self.latest_json = json.loads(stable_response) + else: + response = self._request_github_api(self.RequestUrl, max_retry) + if len(response) == 0: + raise HTTPError + release_array = json.loads(response) + self.latest_json = None + for item in release_array: + if not self.version == Version.Nightly and not self._is_std_version(item['tag_name']): + continue + self.latest_json = item + break + + if self.latest_json is None: + return False + self.latest_version = self.latest_json['tag_name'] + release_assets = self.latest_json['assets'] + + if self.version == Version.Nightly and self.cur_version == self.latest_version: + return False + elif not self._compare_version(self.cur_version, self.latest_version): + return False + + if self._is_std_version(self.latest_version): + info_responce = self._request_github_api(self.InfoRequestUrl + self.latest_version, max_retry) + if len(info_responce) == 0: + raise HTTPError + self.latest_json = json.loads(info_responce) + + for cur_assets in release_assets: + name = cur_assets['name'].lower() + if 'ota' in name and 'win' in name and f'{self.cur_version}_{self.latest_version}' in name: + self.assets_object = cur_assets + break + return True + + def _remove_file(self): + def remove_with_print(s): + self.custom_print(f'删除文件：{s}') + os.remove(s) + + removelist_path = os.path.join(self.path, 'removelist.txt') + if os.path.isfile(removelist_path): + with open(removelist_path, 'r') as f: + file_list = f.readlines() + for file in file_list: + file_path = os.path.join(self.path, file) + if os.path.isfile(file_path): + remove_with_print(file_path) + elif os.path.isdir(file_path): + self.custom_print(f'删除文件夹：{file_path}') + rmtree(file_path) + f.close() + remove_with_print(removelist_path) + + md5sum_path = os.path.join(self.path, 'md5sum.txt') + if os.path.isfile(md5sum_path): + remove_with_print(md5sum_path) + for file in os.listdir(self.path): + if 'OTA' in file: + file_path = os.path.join(self.path, file) + remove_with_print(file_path) + + def update(self): + """ + 更新主函数 + """ + max_retry = 3 + if not self._check_update(): + Updater.custom_print('目前不需要更新') + return False + + # 下载 + replace_list = [ + ('github.com', 'ota.maa.plus'), + ('github.com', 'download.fastgit.org') + ] + for i in range(max_retry): + url = self.assets_object['browser_download_url'] + file = os.path.join(self.path, url.split('/')[-1]) + if i < 2: + url = url.replace(replace_list[i][0], replace_list[i][1]) + try: + Updater.custom_print(f'开始下载更新包，URL：{url}') + request.urlretrieve(url, file) + break + except (HTTPError, URLError) as e: + Updater.custom_print(e) + + # 解压 + Updater.custom_print('开始安装更新，请不要关闭') + zfile = zipfile.ZipFile(file, 'r') + zfile.extractall(self.path) + zfile.close() + + # 删除 + self._remove_file() + + Updater.custom_print('更新完成') diff --git a/src/Python/sample.py b/src/Python/sample.py index 53030c2642..f41d294ac5 100644 --- a/src/Python/sample.py +++ b/src/Python/sample.py @@ -2,7 +2,7 @@ import json import pathlib import time -from asst import InstanceOptionType, Asst, Message +from asst import InstanceOptionType, Asst, Message, Updater, Version @Asst.CallBackType @@ -18,6 +18,9 @@ if __name__ == "__main__": # 请设置为存放 dll 文件及资源的路径 path = pathlib.Path.cwd().parent + # 设置更新器的路径和目标版本并更新 + Updater(path, Version.Stable).update() + # 外服需要再额外传入增量资源路径，例如 # incremental_path=path / 'resource' / 'global' / 'YoStarEN' Asst.load(path=path)