实战|轻松用 Python 开发一个简单有趣的聊天小程序

前言

Internet 协议集支持一个无连接的传输协议，该协议称为用户数据报协议（UDP，User Datagram Protocol）。

UDP 为应用程序提供了一种无需建立连接就可以发送封装的 IP 数据包的方法。

UDP 的特性：它不属于连接型协议，因而具有资源消耗小，处理速度快的优点，所以通常音频、视频和普通数据在传送时使用 UDP 较多，因为它们即使偶尔丢失一两个数据包，也不会对接收结果产生太大影响。

比如我们聊天用的 QQ 就是使用的 UDP 协议。

——摘自百度百科

今天的分享内容是使用 Python 结合 UDP传输协议 来开发一个简单的聊天小程序

环境准备

1. 准备工具：网络调试助手

如下图所示，选择UDP测试这个 tab 栏

2. 本地&目标 ip 与 port 配置

1）本地端口这栏需要配置一个指定的端口并绑定，因为在 Python 程序中需要指定这个 ip 地址和绑定的 port 号来接收信息

2）目标 IP 与目标端口也需要配置，用于 Python 程序接收从网络调试助手页面发送的消息，目标 IP 我这里和本地 IP 是一致的，目标端口则是 Python 程序中代码指定的

代码实战

import socketclass UdpMessage:def send_message(self, udp_socket):# 接收控制台输入的需发送消息的内容msg = input('请输入您要发送的信息：')try:# 接收控制台输入的发送对象的实际ip地址dest_ip = input('请输入您要发送对象的ip地址：')# 接收控制台输入的发送对象的实际port地址，port必须是int类型，使用int强转dest_port = int(input('请输入您发送对象的端口号：'))# 调用sendto方法先将消息内容进行utf-8的编码，后再将消息发送至目标对象udp_socket.sendto(msg.encode('utf-8'), (dest_ip, dest_port))except Exception as err:# 输出错误异常信息print('请输入正确的端口号，错误类型为：%s' % err)def receive_message(self, udp_socket):# 接收数据的最大信息字节数为1024recv_msg = udp_socket.recvfrom(1024)# 将接收的消息进行解码recv_ip_port = recv_msg[1]recv_msg = recv_msg[0].decode('gbk')print(f'收到消息内容：{recv_msg}, 对方的地址与端口：{recv_ip_port}')def entrance(self):# 创建套接字, AF_INET表示：ipv4的ip协议类型, SOCK_DGRAM表示：udp的套接字类型udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)# 绑定本机信息，当ip地址为空字符串时默认绑定本机的ip，接收发送方传递的信息udp_socket.bind(('', 1230))# 死循环保证当前程序一直在运行中，可以随时接收和发送消息while True:print('若发送消息，请输入:send')print('若接收消息，请输入:receive')op_num = input('请问您是想发送还是接收信息？')if op_num == 'send':self.send_message(udp_socket)elif op_num == 'receive':self.receive_message(udp_socket)else:print('您输入的指令非send&receive，请重新输入！')if __name__ == '__main__':UdpMessage().entrance()

实现思路

1.实现 send_message 方法逻辑

接收用户输入需要发送的消息，发送消息对象的 ip 地址, port 信息，对消息内容进行编码，使用try...except代码段来捕获异常

2.实现 receive_message 方法逻辑

通过recvfrom方法接收消息，并指定能接收的最大信息字节数，对接收到的消息进行解码，打印出收到的消息与对方发送该条消息的 ip 地址和 port 信息

3.实现 entrance 方法逻辑

这个方法是整个聊天程序的入口主文件，send_message与receive_message方法都将在此文件中被调用

绑定本机 ip 地址和 1230 端口，发送消息时这个 ip 和 port 就是出口 ip 与 port 信息，在网络调试助手中目标 ip 与 port 要与此设置值保持一致
这里是以死循环来保证聊天程序一直运行中状态，通过用户输入的指令来判断是走接收消息逻辑还是发送消息的逻辑

实现效果

1. 发送消息

配置正确的情况下，按照提示输入指定的信息，就能发送消息成功

2. 接收消息

网络调试助手中配置的目标IP，目标端口信息要与程序中设置一样，接收消息需在控制台输入指令：receive，则可收到 8987 端口发送的消息了

欢迎关注无量测试之道公众号，后台回复：udp 即可领取源码，回复：领取资源

Python+Unittest框架API自动化、

Python+Pytest框架API自动化、

Python+Pandas+Pyecharts大数据分析、

Python+Selenium框架Web的UI自动化、

Python+Appium框架APP的UI自动化、

Python编程学习资源干货、

Vue前端组件化框架开发、

资源和代码免费送啦~
文章下方有公众号二维码，可直接微信扫一扫关注即可。

备注：我的个人公众号已正式开通，致力于IT互联网技术的分享。

包含：数据分析、大数据、机器学习、测试开发、API接口自动化、测试运维、UI自动化、性能测试、代码检测、编程技术等。

微信搜索公众号：无量测试之道

添加关注，让我们一起共同成长！