mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-19 10:32:19 +08:00
重新配置asst导入方式
This commit is contained in:
AutumnSound
@@ -317,9 +317,16 @@ class Battle:
|
||||
try:
|
||||
from asst import Asst, Message
|
||||
except ImportError:
|
||||
print("asst导入失败，请确保asst.py在同目录下，或者下载 "
|
||||
"https://github.com/MaaAssistantArknights/MaaAssistantArknights/blob/master/src/Python/asst.py")
|
||||
return
|
||||
import sys
|
||||
import pathlib
|
||||
sys.path.append(str(pathlib.Path.cwd().parent.parent/"src"/"Python"))
|
||||
try:
|
||||
from asst import Asst, Message
|
||||
except ImportError:
|
||||
print("asst导入失败，请自行解决，或者下载"
|
||||
" https://github.com/MaaAssistantArknights/MaaAssistantArknights/blob/master/src/Python/asst.py "
|
||||
"到同目录下")
|
||||
return
|
||||
|
||||
@Asst.CallBackType
|
||||
def callback(msg, details, arg):
|
||||
|
||||
@@ -1,207 +0,0 @@
|
||||
import ctypes
|
||||
import os
|
||||
import pathlib
|
||||
import platform
|
||||
import json
|
||||
|
||||
from typing import Union, Dict, List, Any, Type
|
||||
from enum import Enum, unique, auto
|
||||
|
||||
JSON = Union[Dict[str, Any], List[Any], int, str, float, bool, Type[None]]
|
||||
|
||||
|
||||
class Asst:
|
||||
CallBackType = ctypes.CFUNCTYPE(
|
||||
None, ctypes.c_int, ctypes.c_char_p, ctypes.c_void_p)
|
||||
"""
|
||||
回调函数，使用实例可参照 my_callback
|
||||
|
||||
:params:
|
||||
``param1 message``: 消息类型
|
||||
``param2 details``: json string
|
||||
``param3 arg``: 自定义参数
|
||||
"""
|
||||
|
||||
@staticmethod
|
||||
def load(path: Union[pathlib.Path, str]) -> bool:
|
||||
"""
|
||||
加载 dll 及资源
|
||||
|
||||
:params:
|
||||
``path``: DLL及资源所在文件夹路径
|
||||
"""
|
||||
if platform.system().lower() == 'windows':
|
||||
Asst.__libpath = pathlib.Path(path) / 'MeoAssistant.dll'
|
||||
os.environ["PATH"] += os.pathsep + str(path)
|
||||
Asst.__lib = ctypes.WinDLL(str(Asst.__libpath))
|
||||
else:
|
||||
Asst.__libpath = pathlib.Path(path) / 'libMeoAssistant.so'
|
||||
os.environ['LD_LIBRARY_PATH'] += os.pathsep + str(path)
|
||||
Asst.__lib = ctypes.CDLL(str(Asst.__libpath))
|
||||
Asst.__set_lib_properties()
|
||||
|
||||
return Asst.__lib.AsstLoadResource(str(path).encode('utf-8'))
|
||||
|
||||
def __init__(self, callback: CallBackType = None, arg=None):
|
||||
"""
|
||||
:params:
|
||||
``callback``: 回调函数
|
||||
``arg``: 自定义参数
|
||||
"""
|
||||
|
||||
if callback:
|
||||
self.__ptr = Asst.__lib.AsstCreateEx(callback, arg)
|
||||
else:
|
||||
self.__ptr = Asst.__lib.AsstCreate()
|
||||
|
||||
def __del__(self):
|
||||
Asst.__lib.AsstDestroy(self.__ptr)
|
||||
self.__ptr = None
|
||||
|
||||
def connect(self, adb_path: str, address: str, config: str = 'General'):
|
||||
"""
|
||||
连接设备
|
||||
|
||||
:params:
|
||||
``adb_path``: adb 程序的路径
|
||||
``address``: adb 地址+端口
|
||||
``config``: adb 配置，可参考 resource/config.json
|
||||
|
||||
:return: 是否连接成功
|
||||
"""
|
||||
return Asst.__lib.AsstConnect(self.__ptr,
|
||||
adb_path.encode('utf-8'), address.encode('utf-8'), config.encode('utf-8'))
|
||||
|
||||
TaskId = int
|
||||
|
||||
def append_task(self, type_name: str, params: JSON = {}) -> TaskId:
|
||||
"""
|
||||
添加任务
|
||||
|
||||
:params:
|
||||
``type_name``: 任务类型，请参考 docs/集成文档.md
|
||||
``params``: 任务参数，请参考 docs/集成文档.md
|
||||
|
||||
: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'))
|
||||
|
||||
def set_task_params(self, task_id: TaskId, params: JSON) -> bool:
|
||||
"""
|
||||
动态设置任务参数
|
||||
|
||||
:params:
|
||||
``task_id``: 任务 ID, 使用 append_task 接口的返回值
|
||||
``params``: 任务参数，同 append_task 接口，请参考 docs/集成文档.md
|
||||
|
||||
:return: 是否成功
|
||||
"""
|
||||
return Asst.__lib.AsstSetTaskParams(self.__ptr, task_id, json.dumps(params, ensure_ascii=False).encode('utf-8'))
|
||||
|
||||
def start(self) -> bool:
|
||||
"""
|
||||
开始任务
|
||||
|
||||
:return: 是否成功
|
||||
"""
|
||||
return Asst.__lib.AsstStart(self.__ptr)
|
||||
|
||||
def stop(self) -> bool:
|
||||
"""
|
||||
停止并清空所有任务
|
||||
|
||||
:return: 是否成功
|
||||
"""
|
||||
Asst.__lib.AsstStop.restype = ctypes.c_bool
|
||||
Asst.__lib.AsstStop.argtypes = (ctypes.c_void_p,)
|
||||
return Asst.__lib.AsstStop(self.__ptr)
|
||||
|
||||
@staticmethod
|
||||
def log(level: str, message: str) -> None:
|
||||
"""
|
||||
打印日志
|
||||
|
||||
:params:
|
||||
``level``: 日志等级标签
|
||||
``message``: 日志内容
|
||||
"""
|
||||
|
||||
Asst.__lib.AsstLog(level.encode('utf-8'), message.encode('utf-8'))
|
||||
|
||||
@staticmethod
|
||||
def get_version() -> str:
|
||||
"""
|
||||
获取DLL版本号
|
||||
|
||||
:return: 版本号
|
||||
"""
|
||||
return Asst.__lib.AsstGetVersion().decode('utf-8')
|
||||
|
||||
@staticmethod
|
||||
def __set_lib_properties():
|
||||
Asst.__lib.AsstLoadResource.restype = ctypes.c_bool
|
||||
Asst.__lib.AsstLoadResource.argtypes = (
|
||||
ctypes.c_char_p,)
|
||||
|
||||
Asst.__lib.AsstCreate.restype = ctypes.c_void_p
|
||||
Asst.__lib.AsstCreate.argtypes = ()
|
||||
|
||||
Asst.__lib.AsstCreateEx.restype = ctypes.c_void_p
|
||||
Asst.__lib.AsstCreateEx.argtypes = (
|
||||
ctypes.c_void_p, ctypes.c_void_p,)
|
||||
|
||||
Asst.__lib.AsstDestroy.argtypes = (ctypes.c_void_p,)
|
||||
|
||||
Asst.__lib.AsstConnect.restype = ctypes.c_bool
|
||||
Asst.__lib.AsstConnect.argtypes = (
|
||||
ctypes.c_void_p, ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p,)
|
||||
|
||||
Asst.__lib.AsstAppendTask.restype = ctypes.c_int
|
||||
Asst.__lib.AsstAppendTask.argtypes = (
|
||||
ctypes.c_void_p, ctypes.c_char_p, ctypes.c_char_p)
|
||||
|
||||
Asst.__lib.AsstSetTaskParams.restype = ctypes.c_bool
|
||||
Asst.__lib.AsstSetTaskParams.argtypes = (
|
||||
ctypes.c_void_p, ctypes.c_int, ctypes.c_char_p)
|
||||
|
||||
Asst.__lib.AsstStart.restype = ctypes.c_bool
|
||||
Asst.__lib.AsstStart.argtypes = (ctypes.c_void_p,)
|
||||
|
||||
Asst.__lib.AsstGetVersion.restype = ctypes.c_char_p
|
||||
|
||||
Asst.__lib.AsstLog.restype = None
|
||||
Asst.__lib.AsstLog.argtypes = (
|
||||
ctypes.c_char_p, ctypes.c_char_p)
|
||||
|
||||
|
||||
@unique
|
||||
class Message(Enum):
|
||||
"""
|
||||
回调消息
|
||||
|
||||
请参考 docs/回调消息.md
|
||||
"""
|
||||
InternalError = 0
|
||||
|
||||
InitFailed = auto()
|
||||
|
||||
ConnectionInfo = auto()
|
||||
|
||||
AllTasksCompleted = auto()
|
||||
|
||||
TaskChainError = 10000
|
||||
|
||||
TaskChainStart = auto()
|
||||
|
||||
TaskChainCompleted = auto()
|
||||
|
||||
TaskChainExtraInfo = auto()
|
||||
|
||||
SubTaskError = 20000
|
||||
|
||||
SubTaskStart = auto()
|
||||
|
||||
SubTaskCompleted = auto()
|
||||
|
||||
SubTaskExtraInfo = auto()
|
||||
Reference in New Issue
Block a user