Python高级——用多线程实现UDP聊天器

思路分析
1.功能：(1)发送消息 (2)接收消息 (3)退出系统
2.架构：
(1)发送消息：send_msg()
(2)接收消息：recv_msg()
(3)主入口：main()

发送消息：(1)用户手动输入信息 (2)将消息编码并发送
接收消息：(1)循环接收消息 (2)解码并显示
主入口：(1)创建套接字，绑定端口 (2)创建一个线程，执行接收消息 (3)选择功能

代码实现

# 导入模块
import socket
import threading


# 发送信息
def send(udp_socket):
    ip = input("请输入目标IP地址：\n")
    if len(ip) == 0:
        ip = '192.168.43.17'
        print("默认为：192.168.43.17")
    port = input("请输入目标端口：\n")
    if len(port) == 0:
        port = '8080'
        print("默认为：8080")
    content = input("请输入要发送的内容：\n")
    udp_socket.sendto(content.encode(), (ip, int(port)))


# 接收信息
def recv(udp_socket):
    # 循环接收消息
    while True:
        content_data, ip_port = udp_socket.recvfrom(1024)
        print("【新消息】", ip_port, ":", content_data.decode())


if __name__ == '__main__':
    # 创建UDP套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 绑定端口
    udp_socket.bind(("", 8080))
    # 创建一个子线程，执行接收消息的方法
    t1 = threading.Thread(target=recv, args=(udp_socket,))
    # 设置线程守护，主线程停止，子线程也停止
    t1.setDaemon(True)
    # 开启线程
    t1.start()
    while True:
        print("**********************")
        print("*****  1、发送信息  ****")
        print("*****  2、退出系统  ****")
        print("**********************")
        # 4）接收用户输入的菜单编号
        sel_num = input("请输入菜单编号:\n")
        # 5）根据编号判断要执行的功能
        if sel_num == '1':
            send(udp_socket)
        elif sel_num == '2':
            print("系统正在退出...")
            print("退出成功!")
            break
        else:
            print("输入错误，请重新输入")
        # 6）关闭套接字
    udp_socket.close()

执行结果


用多线程改写UDP聊天器后，不用手动去接收消息，子线程会自动接收消息，并实时显示出来。