Python UDP聊天器

注释：下面的逻辑分析思路来自李伟老师，感谢在python学习道路上提供的帮助。

程序需求说明：
在一个客户端中编写个程序，有3个基本的功能：

• 1.首先获取键盘数据，并将其发送给对方
• 2.对方接收数据并显示
• 3.退出聊天系统

接下来我们分析下思路

功能： 1、发送消息 2、接收消息 3、退出系统

设计：

• 发送消息 send_msg()
• 接收消息 recv_msg()
• 程序入口 main()

主入口：
1、导入模块 2、创建套接字 3、绑定端口 发送消息：

• 1、保存接收方信息（用户输入）
• 2、保存接收方端口号（用户输入）
• 3、保存要发送的内容（用户输入）
• 4、把用户输入的内容编码并发送

接收消息：
1、使用socket recvfrom 接收数据
2、解码数据并且输出

代码实现：

定义一个send_msg()发送消息和recv_msg()接受消息的函数，是最基本的消息功能,下面通讯所用的是底层socket套接字。

import socket

def send_msg(udp_socket):
    """发送消息的方法"""
    # 1、输入接收方的ip地址
    ipaddr = input("请输入接收方地址:\n")
    if len(ipaddr) == 0:    #如果不输入对应的ip地址，默认设置为本机的ip地址
        ipaddr = "192.168.1.1."   #使用 ifconfig 命令 查看本机的ip地址
        print("默认设置为：%s" % ipaddr)
    # 2、输入接收方的端口
    port = input("请输入接收方端口号:\n")
    if len(port) == 0:      #  端口不写，提供一个默认值
        port = "2438"
        print("默认设置为：%s" % port)
    # 3、要求输入要发送的内容
    content = input("请输入要发送的内容:\n")
    # 4、发送数据
    udp_socket.sendto(content.encode(), (ipaddr, int(port)))

def recv_msg(udp_socket):
    # 1、接收数据  1024表示本次接受的最大的字节数
    recv_data = udp_socket.recvfrom(1024)
    # 2、把接收到的数据（二进制）解码并且显示出来
    re_text = recv_data[0].decode()
    print("接收到消息为：%s" % re_text)
    ip_port = recv_data[1]
    # 3、打印显示发送方的IP和端口信息
    print(ip_port)

main()函数是主程序的入口，上面是定义，下面我们来调用

if __name__ == '__main__':
    
# 创建套接字 
udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

# 绑定端口为2438
udp_socket.bind(("", 2438))

while True:
    # 给出提示，显示聊天器的主要功能
    print("*************************")
    print("****　　　1.发送消息　　****")
    print("****　　　2.接收消息　　****")
    print("****　　　3.退出系统　　****")
    print("*************************")

    # 1、提示用户选择功能
    num = int(input("请选择功能:\n"))
    # 2、判断用户选择
    # 如果选择1,调用发送消息的函数
    if num == 1:
        # print("您选择了发送消息")
        send_msg(udp_socket)
    # 如果选择2,调用接收消息的函数
    elif num == 2:
        # print("您
        recv_msg(udp_socket)

    # 如果选择3,退出程序执行
    else:
        print("程序正在退出...")
        break
        print("程序已退出！～")

# 关闭套接字
udp_socket.close()

注意：
这个程序是简单的了解下udp的通信过程，有许多的bug。比如说当我们选择了接受消息的时候，可是这个时候并没有消息，会造成堵塞的问题，解决的方法是接受发送消息一起进行，这个问题我们先放在这里以后再解决。