socketserver模块

1. 创建 SocketServer TCP 服务器

$ tsTservSS.py
  1 #!/usr/bin/python3
  2 
  3 from socketserver import (TCPServer as TCP, StreamRequestHandler as SRH)
  4 from time import ctime
  5 
  6 HOST = ''
  7 PORT = 6666
  8 ADDR = (HOST, PORT)
  9 
 10 # 定义请求头类
 11 class MyRequestHandler(SRH):
 12 
 13     # 重写handle方法
 14     def handle(self):
 15         # 打印链接地址
 16         print('[+] ...connected from: %s: %d' % (self.client_address[0], self.client_address[1]))
 17         # 发送消息，self.rfile.readline(),读取数据包文件的一行发送回去，
 18         # self.wfile.write(),写入数据包里并发送回去
 19         self.wfile.write(b'[%s] %s' % (bytes(ctime(), 'utf-8'), self.rfile.readline()))
 20 
 21 
 22 tcpServ = TCP(ADDR, MyRequestHandler)
 23 print('[*] waiting for connection...')
 24 tcpServ.serve_forever()

2. 创建 SocketServer TCP 客户端

$ tsTclntSS.py
  1 #!/usr/bin/python3
  2 
  3 from socket import *
  4 
  5 HOST = 'localhost'
  6 PORT = 6666
  7 BUFSIZ = 1024
  8 ADDR = (HOST, PORT)
  9 
 10 while True:
 11     tcpCliSock = socket(AF_INET, SOCK_STREAM)
 12     tcpCliSock.connect(ADDR)
 13     data = input(">>>")
 14     if data == 'exit':
 15        break
 16     # 这里需要加\r\n终止符
 17     tcpCliSock.send(b'%s\r\n' % bytes(data, 'utf-8'))
 18     data = tcpCliSock.recv(BUFSIZ)
 19     if not data:
 20         break
 21     print(data.decode('utf-8').strip())
 22 
 23 tcpCliSock.close()

3. 执行 TCP 服务器和客户端

# 终端1
$ ./tsTservSS.py 
[*] waiting for connection...
[+] ...connected from: 127.0.0.1: 33166    #连接一次
[+] ...connected from: 127.0.0.1: 33168    #发送消息一次
[+] ...connected from: 127.0.0.1: 33170    #发送消息一次

# 终端2
$ ./tsTclntSS.py 
>>>hello
[Mon Nov 27 14:02:51 2017] hello
>>>good bey!
[Mon Nov 27 14:02:53 2017] good bey!
>>>exit