对报错进行了修复, 新加几个装饰器,以及防撤回
This commit is contained in:
@@ -5,8 +5,7 @@ import functools
|
||||
import queue
|
||||
import traceback
|
||||
from threading import Thread
|
||||
from typing import Any, Callable
|
||||
|
||||
from typing import Callable, Any
|
||||
from wcfauto.wcf import WcfV2 as Wcf
|
||||
from wcfauto.wcf import WxMsgV2 as WxMsg
|
||||
|
||||
@@ -18,6 +17,9 @@ def load_function(cls):
|
||||
cls._processing_universal_func = _processing_universal_func
|
||||
cls.message_register = message_register
|
||||
cls.async_message_register = async_message_register
|
||||
cls.revoke_message_register = revoke_message_register
|
||||
cls.group_changed_register = group_changed_register
|
||||
cls.custom_message_register = custom_message_register
|
||||
cls.run = run
|
||||
cls.stop_receiving = stop_receiving
|
||||
return cls
|
||||
@@ -35,20 +37,33 @@ def _process_msg(self, wcf: Wcf):
|
||||
|
||||
|
||||
def _register(self,
|
||||
func: Callable[[Any], Any]):
|
||||
self._add_callback(func, self._wcf)
|
||||
# 此处必须返回被装饰函数原函数, 否则丢失被装饰函数信息
|
||||
return func
|
||||
kind: str,
|
||||
register_name: str,
|
||||
allow_other_receive: bool,
|
||||
judge_msg: Callable[[Any], Any]):
|
||||
def __register(func: Callable[[Any], Any]):
|
||||
self._add_callback(func,
|
||||
self._wcf,
|
||||
kind=kind,
|
||||
register_name=register_name,
|
||||
allow_other_rec=allow_other_receive,
|
||||
judge_msg=judge_msg)
|
||||
# 此处必须返回被装饰函数原函数, 否则丢失被装饰函数信息
|
||||
return func
|
||||
|
||||
return __register
|
||||
|
||||
|
||||
def _processing_async_func(self,
|
||||
isGroup: bool,
|
||||
isDivision: bool,
|
||||
isPyq: bool,):
|
||||
isPyq: bool,
|
||||
register_name: str,
|
||||
allow_other_receive: bool,
|
||||
judge_msg: Callable[[Any], bool]):
|
||||
def _async_func(func):
|
||||
|
||||
@functools.wraps(func)
|
||||
@self._register
|
||||
@self._register('async', register_name, allow_other_receive, judge_msg)
|
||||
async def __async_func(bot: Wcf, message: WxMsg):
|
||||
try:
|
||||
# 判断被装饰函数是否为协程函数, 本函数要求是协程函数
|
||||
@@ -71,17 +86,19 @@ def _processing_async_func(self,
|
||||
def _processing_universal_func(self,
|
||||
isGroup: bool,
|
||||
isDivision: bool,
|
||||
isPyq: bool, ):
|
||||
isPyq: bool,
|
||||
register_name: str,
|
||||
allow_other_receive: bool,
|
||||
judge_msg: Callable[[Any], bool]):
|
||||
def _universal_func(func):
|
||||
|
||||
@functools.wraps(func)
|
||||
@self._register
|
||||
@self._register('universal', register_name, allow_other_receive, judge_msg)
|
||||
def universal_func(bot: Wcf, message: WxMsg):
|
||||
try:
|
||||
# 判断被装饰函数是否为协程函数, 本函数要求是协程函数
|
||||
if asyncio.iscoroutinefunction(func):
|
||||
raise ValueError(
|
||||
f'这里应使用非协程函数, 而被装饰函数-> ({func.__name__}) <-协程函数')
|
||||
raise ValueError(f'这里应使用非协程函数, 而被装饰函数-> ({func.__name__}) <-协程函数')
|
||||
if message.is_pyq() and isPyq:
|
||||
return func(bot, message)
|
||||
if not isDivision:
|
||||
@@ -92,7 +109,6 @@ def _processing_universal_func(self,
|
||||
return func(bot, message)
|
||||
except:
|
||||
traceback.print_exc()
|
||||
return None
|
||||
return universal_func
|
||||
return _universal_func
|
||||
|
||||
@@ -101,18 +117,80 @@ def message_register(self,
|
||||
isGroup: bool = False,
|
||||
isDivision: bool = False,
|
||||
isPyq: bool = False):
|
||||
return self._processing_universal_func(isGroup, isDivision, isPyq)
|
||||
return self._processing_universal_func(isGroup,
|
||||
isDivision,
|
||||
isPyq,
|
||||
register_name='common',
|
||||
allow_other_receive=True,
|
||||
judge_msg=lambda x: True)
|
||||
|
||||
|
||||
def revoke_message_register(self,
|
||||
isGroup: bool = False,
|
||||
isDivision: bool = False,
|
||||
isPyq: bool = False,
|
||||
allow_other_receive: bool = True):
|
||||
def judge_msg(msg):
|
||||
if msg['isRevokeMsg'] and msg['revokmsgid'] is not None:
|
||||
return True
|
||||
return False
|
||||
|
||||
return self._processing_async_func(isGroup,
|
||||
isDivision,
|
||||
isPyq,
|
||||
register_name='revokeMessage',
|
||||
allow_other_receive=allow_other_receive,
|
||||
judge_msg=judge_msg)
|
||||
|
||||
|
||||
def group_changed_register(self,
|
||||
isGroup: bool = False,
|
||||
isDivision: bool = False,
|
||||
isPyq: bool = False,
|
||||
allow_other_receive: bool = True):
|
||||
def judge_msg(msg):
|
||||
if msg['isGroup'] and msg['data']['type'] == 10000:
|
||||
if '加入了群聊' in msg['data']['content'] or '出了群聊' in msg['data']['content']:
|
||||
return True
|
||||
return False
|
||||
|
||||
return self._processing_async_func(isGroup,
|
||||
isDivision,
|
||||
isPyq,
|
||||
register_name='groupChanged',
|
||||
allow_other_receive=allow_other_receive,
|
||||
judge_msg=judge_msg)
|
||||
|
||||
|
||||
def custom_message_register(self,
|
||||
register_name: str,
|
||||
msg_judge_func: Callable[[WxMsg], bool],
|
||||
allow_other_receive: bool,
|
||||
isGroup: bool = False,
|
||||
isDivision: bool = False,
|
||||
isPyq: bool = False):
|
||||
return self._processing_async_func(isGroup,
|
||||
isDivision,
|
||||
isPyq,
|
||||
register_name=register_name,
|
||||
allow_other_receive=allow_other_receive,
|
||||
judge_msg=msg_judge_func)
|
||||
|
||||
|
||||
def async_message_register(self,
|
||||
isGroup: bool = False,
|
||||
isDivision: bool = False,
|
||||
isPyq: bool = False):
|
||||
return self._processing_async_func(isGroup, isDivision, isPyq)
|
||||
return self._processing_async_func(isGroup,
|
||||
isDivision,
|
||||
isPyq,
|
||||
register_name='common',
|
||||
allow_other_receive=True,
|
||||
judge_msg=lambda x: True)
|
||||
|
||||
|
||||
def run(self, *args, **kwargs):
|
||||
self._wcf.enable_receiving_msg(*args, pyq=True, **kwargs)
|
||||
def run(self, pyq=True):
|
||||
self._wcf.enable_receiving_msg(pyq=pyq)
|
||||
Thread(target=self._process_msg, name="GetMessage", args=(self._wcf,), daemon=True).start()
|
||||
self._LOG.debug("开始接受消息")
|
||||
self._wcf.keep_running()
|
||||
|
||||
Reference in New Issue
Block a user