python 网络编程 TCP套接字实例

一. 服务器端代码tcpServer.py:

from socket import *
from time import ctime

#HOST变量为空，表示bind()函数可以绑定在所有有效的地址上
HOST = ''
PORT = 21234
#设置缓冲大小为128，可以根据网络情况和需求来进行修改
BUFSIZ = 128
ADDR = (HOST, PORT)

#创建TCP套接字
tcpSerSock = socket(AF_INET, SOCK_STREAM)
tcpSerSock.bind(ADDR)
#listen()函数的参数表示最多允许同时接收的连接数，超过这个数目的连接会被拒绝掉
tcpSerSock.listen(5)

while True:
    print 'waiting for connection...'
    tcpCliSock, addr = tcpSerSock.accept()
    print '...connected from:', addr

    while True:
        #接收数据
        data = tcpCliSock.recv(BUFSIZ)
        if not data:
            break
        #发送数据
        tcpCliSock.send('From server response:[%s] %s' % (ctime(), data))

tcpCliSock.close()
tcpSerSock.close()
			

客户端代码tcpClient.py:

from socket import *

HOST = 'localhost'
PORT = 21234
BUFSIZ = 128
ADDR = (HOST, PORT)

#创建客户端TCP套接字
tcpCliSock = socket(AF_INET, SOCK_STREAM)
#连接服务器
tcpCliSock.connect(ADDR)

while True:
    data = raw_input('>')
    if not data:
        break
    #发送数据
    tcpCliSock.send(data)
    #接收数据
    data = tcpCliSock.recv(BUFSIZ)
    if not data:
        break
    print data
	
tcpCliSock.close()
	

二、 使用SocketServer模块实现TCP服务器端和客户端

SocketServer是标准库中的一个高级模块，用于简化服务器端与客户端的实现。

客户端代码，server.py:

from SocketServer import (TCPServer as TCP, StreamRequestHandler as SRH)
from time import ctime

HOST = ''
PORT = 20123
ADDR = (HOST, PORT)

#继承SRH类，并重写handle()函数。当有客户消息进来时，handle()函数就会被调用。StreamRequestHandler类支持像操作文件一样对套接字进行输入输出
#操作。通过readline()函数得到客户消息，write()函数把消息返回给客户。
class MyRequestHandler(SRH):
    def handle(self):
        print '...connect from:', self.client_address
        self.wfile.write('[%s] %s' % (ctime(), self.rfile.readline()))

#利用主机信息和请求处理类来创建TCP服务器		
tcpServ = TCP(ADDR, MyRequestHandler)
print 'waiting for connection...'
#无限循环，等待与处理客户请求
tcpServ.serve_forever()

客户端代码，client.py:

from socket import *

HOST = 'localhost'
PORT = 20123
BUFSIZ = 128
ADDR = (HOST, PORT)

while True:
    client = socket(AF_INET, SOCK_STREAM)
#由于SocketServer的请求处理器的默认行为是接受连接，得到请求，然后关闭连接。这使得程序在运行时不能一直保持连接状态，因此，每次发送数据前都要
创建一个新的套接字
    client.connect(ADDR)
    data = raw_input('>')
    if not data:
        break
#由于服务器端的处理器类像文件一样操作套接字，所以每次都要发送行结束符（回车与换行）。服务器保留并重用客户端发送的行结束符。当从服务器端得到数据
时，使用strip()函数去掉它们
    client.send('%s\r\n' % data)
    data = client.recv(BUFSIZ)
    if not data:
        break
    print data.strip()
	
client.close()