#!/usr/bin/env python

# -*- coding: utf-8 -*-

"""

pip install spyne

<targets>

<target type='WebService'

name='ws'

url='http://127.0.0.1:8000/logme'

protocol='HttpPost'

encoding='UTF-8'>

<parameter name='logger' type='System.String' layout='${logger}'/>

<parameter name='level' type='System.String' layout='${level}'/>

<parameter name='message' type='System.String' layout='${message}'/>

<parameter name='title' type='System.String' layout='${event-context:item=title}'/>

</target>

</targets>

<rules>

<logger name="*" minlevel="Info" writeTo="ws" />

</rules>

protected NLog.Logger log_wx = NLog.LogManager.GetLogger("wx");

log_wx.Warn("策略启动。");

"""

from spyne import Application, rpc, ServiceBase, String

from spyne.protocol.http import HttpRpc

from spyne.protocol.json import JsonDocument

from spyne.server.wsgi import WsgiApplication

from DtalkRobot import *

from BatchDelayQueue import DelayQueue

# 99simnow

webhook_a99ctp = "https://oapi.dingtalk.com/robot/send?access_token=11111"

# 98实

webhook_a98ctp = "https://oapi.dingtalk.com/robot/send?access_token=11111"

# S01策略

webhook_s01 = "https://oapi.dingtalk.com/robot/send?access_token=11111"

# 不明的消息都发到模拟平台上

webhook = webhook_a99ctp

ip = '0.0.0.0'

port = 8000

def send_msgs(logger, level, title, message):

# 利用logger,可以将消息发向不同的机器人

if logger == 'ALL':

_webhook = webhook

elif logger == 'A98CTP.A':

_webhook = webhook_a98ctp

elif logger == 'A98CTP.P':

_webhook = webhook_a98ctp

elif logger == 'S01':

_webhook = webhook_s01

else:

_webhook = webhook

if title is None or len(title) == 0:

robot.sendText_webhook(_webhook, message)

else:

msg = f"##### {title}\n{message}"

robot.sendMarkdown_webhook(_webhook, title, msg)

def process_log_msg(tuples):

msgs = {}

for tp in tuples:

logger, level, title, message = tp

key = (logger, level, title)

lst = msgs.get(key, [])

lst.append(message)

msgs[key] = lst

print("定时处理", len(msgs))

for k, v in msgs.items():

# 这里不能太长，否则会被截断

send_msgs(k[0], k[1], k[2], '\n'.join(v))

class LogService(ServiceBase):

@rpc(String, String, String, String, _returns=String)

def logme(ctx, logger, level, title, message):

delayQueue.add((logger, level, title, message))

return 'OK'

delayQueue = DelayQueue(10, process_log_msg)

application = Application([LogService],

tns='kan.logger.wechat',

in_protocol=HttpRpc(validator='soft'),

out_protocol=JsonDocument()

)

if __name__ == '__main__':

# You can use any Wsgi server. Here, we chose

# Python's built-in wsgi server but you're not

# supposed to use it in production.

from wsgiref.simple_server import make_server

wsgi_app = WsgiApplication(application)

server = make_server(ip, port, wsgi_app)

robot = DtalkRobot(webhook)

robot.sendText_webhook(webhook, "启动钉钉日志服务成功")

# 不退出

server.serve_forever()