【测试开发】python系列教程：socket模块

上一篇

【测试开发】python系列教程：threading模块

这次我们分享python的socket库。

1.概念

Python中的socket模块是底层的网络接口， socket模块提供了标准的BSD SocketAPI和服务器中心类，可以简化网络服务器的开发。该模块提供了两个基本的Socket模块：服务端Socket和客户端Socket。当创建了一个服务端Socket之后，这个Socket就会在本机的一个端口上等待连接，客户端Socket会访问这个端口，当两者完成连接之后，就可以进行交互了。

2.基本用法

1. Socket实例化首先看一下如何实例化一个Socket。Socket实例化的格式为：

socket(family,type[,protocal])

其中，三个参数中的family是要使用的地址族，以下是常用的协议族

第二个参数type用来指明Socket类型，这里可以使用一下参数

第三个参数是指使用的协议，这个参数是可选额，通常赋值“0”，由系统自动选择。如果希望初始化一个TCP类型的Socket，可以使用以下语句。

s=socket.socket()

这条语句相当于socket.socket(socket.AF_INET,socket.SOCK_STREAM)。这里因为都是默认值，所以可以省略。而如果想要初始化一个UDP类型的Socket，则可以使用如下语句。

s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

3.Socket常用函数

socket类中主要提供如下所示常用的函数。

四、代码演示

思路

• 1 创建套接字，绑定套接字到本地IP与端口 　 s=socket.socket(socket.AF_INET,socket.SOCK_STREAM) s.bind() 2 开始监听连接 s.listen() 3 进入循环，不断接受客户端的连接请求 s.accept() 4 然后接收传来的数据，并发送给对方数据 s.recv() , s.sendall() 5 传输完毕后，关闭套接字 s.close()
• TCP客户端: 1 创建套接字，连接远端地址 socket.socket(socket.AF_INET,socket.SOCK_STREAM) , s.connect() 2 连接后发送数据和接收数据 # s.sendall(), s.recv() 3 传输完毕后，关闭套接字 s.close()

demo演示 TCP服务端

import socket               # 导入 socket 模块

s = socket.socket()         # 创建 socket 对象
host = socket.gethostname() # 获取本地主机名

port = 8082                # 设置端口
s.bind((host, port))        # 绑定端口

s.listen(5)                 # 等待客户端连接
while True:
    c, addr = s.accept()     # 建立客户端连接。
    print('连接地址：', addr)
    c.send('leizi')
    c.close()  # 关闭连接

demo演示客户端

import socket               # 导入 socket 模块

s = socket.socket()         # 创建 socket 对象
host = socket.gethostname() # 获取本地主机名
port = 8082               # 设置端口好

s.connect((host, port))
print(s.recv(1024))
s.close()

这里首先启动服务端

再启动客户端

可以看到，客户端已经接到消息了，看下服务端

这样证明我们这里写的已经实现了。

演示一个UDP的服务端

from socket import *

HOST = '192.168.1.123'
PORT = 8888

s = socket(AF_INET, SOCK_DGRAM)
s.connect((HOST, PORT))
while True:
    message = input('send message:>>')
    s.sendall(message)
    data = s.recv(1024)
    print(data)

    s.close() 

这样简单的demo就实现了。