RobotFramework二次开发——Socket推送实时日志

背景

上一章说道RobotFramework的执行监听，是通过Socket服务来实现的，那么现在就来实现一下简单的Socket服务

思路

Socket服务有一个client端和一个server端，Listener.py我们可以当成一个Client，因为在执行的时候，我们需要Listener来发送实时数据，因此，我们还要另写一段服务端的代码。

改造Listener.py

首先，要开启要一个Socket服务

    def __init__(self):        self.sock = socket.socket()        self.conn = self.sock.connect(("127.0.0.1", 50007))

在类的初始化构造中起一个连接，端口可以先暂时指定，如果有需求，再改造成参数化传入。
之前我们获取的实时数据都是用print的方式打印出来，现在我们要把每个print的内容变成send_socket。

那么就要构造一个发送消息的方法。

    def _send_socket(self, msg):        self.sock.sendall(msg)

类似这样的，往连接的端口发送数据。

再把所有的print内容用_send_socket方法发送出去。

    def log_message(self, message):        # print message['timestamp'] + " :   " + message['level'] + " : " + message['message']        self._send_socket(message['timestamp'] + " :   " + message['level'] + " : " + message['message'])

这样，我们的Listerner就改造完成了。

Server端

Server端就比较简单了，直接网上找一个代码改改就好。

import socketHOST = ''  # Symbolic name meaning all available interfacesPORT = 50007  # Arbitrary non-privileged ports = socket.socket(socket.AF_INET, socket.SOCK_STREAM)s.bind((HOST, PORT))s.listen(1)conn, addr = s.accept()print 'Connected by', addrwhile 1:    data = conn.recv(1024)    print data    if not data: breakconn.close()

说明一下，上面代码中拿到了数据，也是用print方法打印出来，这里是可以构造很多后续操作的，比如写文件，或者集成到web环境时，可以用yield返回stream数据，在web端实时展示数据。

最后没有数据返回的时候，就关闭连接。

完整代码

Listener.py# ecoding=utf-8# Author: Sven_Weng# Email : [e-mail]# Web   : http://wybblog.applinzi.comimport socketclass RobotListener(object):    ROBOT_LISTENER_API_VERSION = 2    def __init__(self):        self.sock = socket.socket()        self.conn = self.sock.connect(("127.0.0.1", 50007))    def start_suite(self, name, args):        self._send_socket("Starting Suite : " + name + "  " + args['source'])    def start_test(self, name, args):        self._send_socket("Starting test : " + name)        if args['template']:            print 'Template is : ' + args['template']    def end_test(self, name, args):        self._send_socket("Ending test:  " + args['longname'])        self._send_socket("Test Result is : " + args['status'])        self._send_socket("Test Time is: " + str(args['elapsedtime']))    def log_message(self, message):        self._send_socket(message['timestamp'] + " :   " + message['level'] + " : " + message['message'])    def _send_socket(self, msg):        self.sock.sendall(msg)

Server.pyimport socketHOST = ''  # Symbolic name meaning all available interfacesPORT = 50007  # Arbitrary non-privileged ports = socket.socket(socket.AF_INET, socket.SOCK_STREAM)s.bind((HOST, PORT))s.listen(1)conn, addr = s.accept()print 'Connected by', addrwhile 1:    data = conn.recv(1024)    print data    if not data: breakconn.close()