案例·：udp聊天室（四）

案例·：udp聊天室

报一下环境：
win10 + pycharm（python3.7）+ mNetAssist

import socket


def send_msg(udp_socket):  # udp_sockrt 形参
    """发送消息"""
    dest_ip = input("请输入对方的ip：")
    dest_port = int(input("请输入对方的port："))
    send_data = input("请输入要发送的消息：")
    udp_socket.sendto(send_data.encode("gbk"), (dest_ip, dest_port))


def recv_msg(udp_socket):
    """接收数据"""
    recv_data = udp_socket.recvfrom(1024)
    print("%s: %s" % (str(recv_data[1]), recv_data[0].decode("gbk")))


def main():
    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 绑定信息
    udp_socket.bind(("", 7788))

    # 循环进行收发数据
    while True:
        print("---XXX聊天器---")
        print("1.发送消息")
        print("2.接收消息")
        print("0.退出系统")
        op = input("请输入功能：")

        if op == "1":
            # 发送，获取要发送的内容
            send_msg(udp_socket)
        elif op == "2":
            # 接收并显示
            recv_msg(udp_socket)
        elif op == "0":
            # 退出
            break
        else:
            print("请重新输入。。。")


if __name__ == "__main__":
    main()