python---socket自动化交互

socket 的自动化交互

假设服务端运行程序为：

# coding:utf8
# python3

import random
import socket
import sys

# 创建 socket 对象
serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

port = 9999

# 绑定端口号
serversocket.bind(('', port))

# 设置最大连接数，超过后排队
serversocket.listen(5)

while True:
    print('等待连接...')
    # 建立客户端连接
    clientsocket, addr = serversocket.accept()
    print("连接地址: ", addr)

    first_num = str(random.randint(0, 10))
    second_num = str(random.randint(0, 10))

    clientsocket.send('input {}: '.format(first_num).encode('utf8'))
    first_input = clientsocket.recv(10).decode('utf8').strip()
    clientsocket.send('input {}: '.format(second_num).encode('utf8'))
    second_input = clientsocket.recv(10).decode('utf8').strip()

    print(first_input)
    print(second_input)

    if first_num == first_input and second_num == second_input:
        clientsocket.send('Right!\n'.encode('utf8'))
    else:
        clientsocket.send('Wrong!\n'.encode('utf8'))
    clientsocket.shutdown(socket.SHUT_RDWR)
    clientsocket.close()
serversocket.close()

运行如下脚本即可达到自动化交互的目的：

# coding:utf-8
# python3

import sys
import re
import socket
from threading import Thread


class SocketInteraction:
    tcp_socket = None

    def __init__(self, host, port):
        self.tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.tcp_socket.connect((host, port))

    def __del__(self):
        self.tcp_socket.close()

    def recvline(self):
        return self.recvuntil('\n')
    
    def recv_n(self, n):
        return self.tcp_socket.recv(n).decode('utf8')

    def recvuntil(self, want_end_str):
        current_str = ''
        while True:
            current_str += self.recv_n(1)
            if current_str.endswith(want_end_str):
                return current_str

    def recvuntil_re(self, want_re_str):
        current_str = ''
        while True:
            current_str += self.recv_n(1)
            s = re.search(want_re_str, current_str)
            if s:
                return [current_str, s]

    def send(self, send_str):
        final_bytes = send_str.encode('utf8')
        self.tcp_socket.send(final_bytes)

    def sendline(self, send_str):
        self.send(send_str+'\n')
    
    def interact(self):
        def recv_loop():
            while True:
                c = self.recv_n(1)
                # print 写到控制台会有延时，直接用系统io写
                sys.stdout.write(c)
                sys.stdout.flush()

        def send_loop():
            while True:
                send_str = input()
                self.sendline(send_str)

        recv_thread = Thread(target=recv_loop)
        send_thread = Thread(target=send_loop)

        recv_thread.start()
        send_thread.start()

        recv_thread.join()
        send_thread.join()



si = SocketInteraction('127.0.0.1', 9999)
for i in range(2):
    content = si.recvuntil_re(r'input (\d+): ')
    print(content[0])
    num = content[1].group(1)
    print(repr(num))
    si.sendline(num)
content = si.recvline()
print(content)