TOOLBOX/Webhook.py at master · YUDONGLING/TOOLBOX · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
def Lark(Token: str, Topic: str = '', Message: list = None) -> dict:
    '''
    Send Message to Lark Chat Group.
    '''
    raise NotImplementedError()


def Slack(Token: str, Topic: str = '', Message: list = None) -> dict:
    '''
    Send Message to Slack Chat Group.
    '''
    raise NotImplementedError()


def WeChat(Token: str, Topic: str = '', Message: list = None) -> dict:
    '''
    Send Message to Enterprise WeChat Chat Group. `Token` Should Likes `{{ACCESS_TOKEN}}|{{CHAT_ID}}`.
    '''
    raise NotImplementedError()


def DingTalk(Token: str, Topic: str = '', Message: list = None) -> dict:
    '''
    Send Message to DingTalk Chat Group.
    '''
    import time
    import requests

    if not __package__:
          from  Init import DotAccessDict; from  Log import MakeErrorMessage
    else: from .Init import DotAccessDict; from .Log import MakeErrorMessage

    Response = DotAccessDict({
        'Ec': 0, 'Em': ''
    })

    if Message == None: Message = []

    try:
        MessageBody = '<font color=#6A65FF>**%s**</font> \n\n %s \n\n' % (Topic, time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()))

        for Item in Message:
            if Item.get('Title', '') == '':
                MessageBody += ' --- \n\n '
            else:
                Color = Item.get('Color', 'UNKNOWN').upper()
                if not Color.startswith('#') or len(Color) != 7:
                    Color = {'PURPLE': '#6A65FF', 'RED': '#FF6666', 'GREEN': '#92D050', 'BLUE': '#76CCFF'}.get(Color, '#76CCFF')
                MessageBody += ' --- \n\n <font color=%s>**%s**</font> \n\n ' % (Color, Item.get('Title', ''))
            MessageBody += ' \n\n '.join(Item.get('Text', [])) + ' \n\n '

        MessageBody += '<font color=#6A65FF>%s</font>' % Topic
    except Exception as Error:
        Response['Ec'] = 50001; Response['Em'] = MakeErrorMessage(Error, Code = Response['Ec']); return Response

    try:
        Url = 'https://oapi.dingtalk.com/robot/send?access_token=%s' % Token.removeprefix('http://oapi.dingtalk.com/robot/send?access_token=').removeprefix('https://oapi.dingtalk.com/robot/send?access_token=')
        Hed = {
            'content-type': 'application/json'
        }
        Dat = {
            'msgtype'  : 'markdown',
            'markdown' : {
                'title': Topic,
                'text' : MessageBody
            }
        }
        Rsp = requests.post(Url, headers = Hed, json = Dat, timeout = 10).json()

        if Rsp.get('errcode', -1) != 0:
            raise ValueError('<Interface [%s]> %s' % (Rsp.get('errcode'), Rsp.get('errmsg') or Rsp or None))
    except Exception as Error:
        Response['Ec'] = 50002; Response['Em'] = MakeErrorMessage(Error, Code = Response['Ec']); return Response
    finally:
        try: Rsp.close()
        except: pass

    return Response


def Telegram(Token: str, Topic: str = '', Message: list = None) -> dict:
    '''
    Send Message via Telegram Bot.
    '''
    raise NotImplementedError()


def SeverChan(Token: str, Topic: str = '', Message: list = None) -> dict:
    '''
    Send Message via SeverChan.
    '''
    raise NotImplementedError()