UDP的聊天器

“””
实现UDP的聊天器
1.获取键盘数据，并将其发送给对方
2.接受数据并显示
3.并且功能数据进行选择以上2个功能调用
“”“

import socket


def send_msg(udp_socket):
    """获取键盘数据，并将其发送给对方"""
    # 1. 从键盘输入数据
    msg = input("\n请输入要发送的数据:")
    # 2. 输入对方的ip地址
    dest_ip = input("\n请输入对方的ip地址:")
    # 3. 输入对方的port
    dest_port = int(input("\n请输入对方的port:"))
    # 4. 发送数据
    udp_socket.sendto(msg.encode("utf-8"), (dest_ip, dest_port))
    # windows 'gbk'   linux'utf-8'


def recv_msg(udp_socket):
    """接收数据并显示"""
    # 1. 接收数据
    recv_msg = udp_socket.recvfrom(1024)
    # 2. 解码
    recv_ip = recv_msg[1]
    recv_msg = recv_msg[0].decode("utf-8")
    # 3. 显示接收到的数据
    print(">>>%s:%s" % (str(recv_ip), recv_msg))


def main():
    # 1. 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # AF_INET 网络通信
    # AF_UNIX 本机通信
    # SOCK_STREAM 流失套接字 tcp协议
    # SOCK_DGRAM  数据报套接字 udp协议
    # 2. 绑定本地信息
    udp_socket.bind(("", 7890))
    # ''接收本机的ip
    while True:
        # 3. 选择功能
        print("=" * 30)
        print("1:发送消息")
        print("2:接收消息")
        print("=" * 30)
        op_num = input("请输入要操作的功能序号:")

        # 4. 根据选择调用相应的函数
        if op_num == "1":
            send_msg(udp_socket)
        elif op_num == "2":
            recv_msg(udp_socket)
        else:
            print("输入有误，请重新输入...")


if __name__ == "__main__":
    main()