1
- import json
2
-
3
1
from .frames import Headers
4
2
from .frames import AckMessage
5
3
from .frames import SystemMessage
6
4
from .frames import EventMessage
7
5
from .frames import CallbackMessage
8
- from .stream import DingTalkStreamClient
9
6
from .log import setup_default_logger
10
7
8
+ from typing import TYPE_CHECKING
9
+
10
+ if TYPE_CHECKING :
11
+ from .stream import DingTalkStreamClient
11
12
12
13
class CallbackHandler (object ):
13
14
TOPIC_CARD_CALLBACK = '/v1.0/card/instances/callback'
14
15
15
16
def __init__ (self ):
16
- self .dingtalk_client : DingTalkStreamClient = None
17
+ self .dingtalk_client : ' DingTalkStreamClient' = None
17
18
self .logger = setup_default_logger ('dingtalk_stream.handler' )
18
19
19
20
def pre_start (self ):
@@ -34,7 +35,7 @@ async def raw_process(self, callback_message: CallbackMessage):
34
35
35
36
class EventHandler (object ):
36
37
def __init__ (self ):
37
- self .dingtalk_client : DingTalkStreamClient = None
38
+ self .dingtalk_client : ' DingTalkStreamClient' = None
38
39
self .logger = setup_default_logger ('dingtalk_stream.handler' )
39
40
40
41
def pre_start (self ):
@@ -56,13 +57,13 @@ async def raw_process(self, event_message: EventMessage):
56
57
57
58
class SystemHandler (object ):
58
59
def __init__ (self ):
59
- self .dingtalk_client : DingTalkStreamClient = None
60
+ self .dingtalk_client : ' DingTalkStreamClient' = None
60
61
self .logger = setup_default_logger ('dingtalk_stream.handler' )
61
62
62
63
def pre_start (self ):
63
64
return
64
65
65
- async def process (self , message ):
66
+ async def process (self , message : SystemMessage ):
66
67
return AckMessage .STATUS_OK , 'OK'
67
68
68
69
async def raw_process (self , system_message : SystemMessage ):
0 commit comments