UDP聊天器案例

"""
功能:
1. 发送信息
2. 接收信息
3. 退出系统

架构:

1. 发送信息 send_msg()
2. 接收信息 recv_msg()
3. 程序的主入口 main()
4. 当程序独立运行时，才启动聊天器

实现步骤
1. 发送信息 send_msg()
1). 定义变量接收用户与输入的接收方的 IP地址
2). 定义变量接收用户与输入的接收方端口
3). 定义变量接收用户与输入的接收方内容
4). 使用 socket 的 sendto() 发送信息

2. 接收信息 recv_msg()

1). 使用 socket 接收数据
2). 解码
3). 输出显示


3. main() 主入口
1). 创建套接字
2). 绑定端口
3). 打印菜单
4). 接收用户输入的选项
5). 判断用户的选择，并且调用对应的函数
6). 退出套节字
"""
import socket

def send_msg(udp_socket):
    """发送信息的函数"""
    # 1. 发送信息 send_msg()
    # 1).定义变量接收用户与输入的接收方的 IP地址
    ipaddr = input("请输入对方接收 IP 地址:\n")

    # 2).定义变量接收用户与输入的接收方端口
    port = input("请输入对方的端口号:\n")

    # 3).定义变量接收用户与输入的接收方内容
    content = input("请输入要发送的内容:\n")
    # 4).使用 socket 的 sendto() 发送信息
    udp_socket.sendto(content.encode(), (ipaddr, int(port)))



def recv_msg(udp_socket):
    """接收信息的函数"""
    # 2. 接收信息 recv_msg()
    #
    # 1). 使用 socket 接收数据
    recv_data, ip_port = udp_socket.recvfrom(1024)
    # 2). 解码
    recv_data.decode()
    # 3). 输出显示
    print("接收到 [%s] 的信息: %s" % (str(ip_port), recv_data.decode("utf-8")))


def main():
    """程序的主入口"""
    # 1).创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 2).绑定端口
    udp_socket.bind(("", 8080))
    # 3).打印菜单
    while True:
        print("*********************************")
        print("*********** 1、发送信息 ***********")
        print("*********** 2、接收信息 ***********")
        print("*********** 3、退出系统 ***********")
        print("*********************************")
        # 4).接收用户输入的选项
        sel_num = int(input("请输入选项:\n"))
        if sel_num == 1:
            print("您选择的是发送信息:")
            send_msg(udp_socket)
        elif sel_num == 2:
            print("您选择的是接收信息:")
            recv_msg(udp_socket)
        elif sel_num == 3:
            print("系统正在退出中 ...")
            print("系统退出完成!")

        # 5).判断用户的选择，并且调用对应的函数
    udp_socket.close()
    # 6). 关闭套接字

if __name__ == "__main__":
    main()

这记录学习第一个 UDP 聊天器的案例过程, 只是实现简单的功能。