11-应用：udp聊天器（二）

参考代码

import socket
# 聊天器主要功能
# 1、发送消息
# 2、接收消息
# 3、退出消息

def send_msg(udp_socket):
    """发送消息的方法"""
    # 1、输入接收方的ip地址
    ipaddr = input("请输入接收方地址:\n")
    if len(ipaddr) == 0:
        ipaddr = "192.168.31.163"
        print("默认设置为：%s" % ipaddr)
    # 2、输入接收方的端口
    port = input("请输入接收方端口号:\n")
    if len(port) == 0:
        port = "6666"
        print("默认设置为：%s" % port)
    # 3、要求输入要发送的内容
    content = input("请输入要发送的内容:\n")
    # 4、发送数据
    udp_socket.sendto(content.encode(), (ipaddr, int(port)))


def recv_msg(udp_socket):
    # 1、接收数据
    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)


if __name__ == '__main__':

    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

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

    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()

想一想

• 以上的程序如果选择了接收数据功能，并且此时没有数据，程序会堵塞在这，那么怎样才能让这个程序收发数据一起进行呢？别着急，学习完多任务知识之后就解决了O(∩_∩)O...