Python之TCP编程

参考原文

　　廖雪峰Python教程

客户端

　我们知道每一条TCP的连接有2个端点，这两个端点叫做套接字socket。如果我们要进行基于TCP的通信必须先创建套接字。在Python中可以这样创建套接字socket，并向服务端发送请求：

#导入socket库
import socket
import io

#创建一个socket：
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#建立链接
s.connect(('www.sina.com.cn', 80))
#客户端发送请求：
s.send(b'GET / HTTP/1.1\r\nHost: www.sina.com.cn\r\nConnection: close\r\n\r\n') #注意空格，回车换行

#接收从服务器返回的数据：
buffer = []
while True:
    d = s.recv(1024) #每次最多接收1K字节
    if d:
        buffer.append(d)
    else:
        break
data = b''.join(buffer)  #连接字符串
#关闭链接
s.close()
#将HTTP报文头和正文分离：
header, html = data.split(b'\r\n\r\n',1) #后面的参数是分割次数
#看下响应报文头：
print(header.decode('utf-8'))
#把接收到的正文写入文件：
with open('sina.html','wb') as f :
    f.write(html)

服务端

#导入socket库
import socket
import threading
import time
def tcplink(sock, addr):#新线程执行的函数
    print('Accept new connection from %s:%s...' % addr)
    sock.send(('Welcome!').encode('utf-8')) #发送数据给客户端
    while True:
        data = sock.recv(1024) #接收来自客户端的数据，最大（1k）,阻塞式等待
        time.sleep(1) # stop: 1s 避免过度占用CPU
        #如果客户端没有发送数据或发送’exit‘：关闭连接
        if not data or data.decode('utf-8') =='exit':
            break
        #发送编码后的数据：
        sock.send(('Hello, %s' % data.decode('utf-8')).encode('utf-8'))
    sock.close()
    print('Connection from %s:%s closed' % addr)

#创建一个基于IPv4和TCP协议的socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#绑定监听端口：（服务的端口号<1024时，需要管理员权限）
s.bind(('127.0.0.1', 9999)) #我们写的不是标准服务(标准服务HTTP，STMP)
#开始监听：
s.listen(5)  #param : 等待连接的最大数量
print('waiting for connecting')
#永久循环来接收来自客户端的连接：
while True:
    #接收一个新的连接：
    sock, addr = s.accept() #accept会等待并发返回一个客户端的连接
    #创建新线程（多线程）来处理这个连接：单线程在处理过程中，无法接收其他客户端的连接
    t = threading.Thread(target=tcplink, args=(sock,addr))
    t.start()  #启动线程

　　测试客户端代码：

import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1', 9999))
#接收欢迎数据：
print(s.recv(1024).decode('utf-8'))
#客户端发送数据：
for data in [('Michael').encode('utf-8'), ('Tracy').encode('utf-8'), b'Search']:
    s.send(data)
    #打印从服务端返回的数据：
    print(s.recv(1024).decode('utf-8'))
s.send(b'exit')
s.close()

　　运行效果：

Tips:用TCP协议进行Socket编程在Python中十分简单，对于客户端，要主动连接服务器的IP和指定端口，对于服务器，要首先监听指定端口，然后，对每一个新的连接，创建一个线程或进程来处理。通常，服务器程序会无限运行下去。 同一个端口，被一个Socket绑定了以后，就不能被别的Socket绑定了。