udp聊天器

import socket
"""导入套接字模块"""


def menu():
    """显示菜单"""
    print("$" * 50)
    print("[1].发送信息")
    print("[2].接受信息")
    print("[3].退出聊天")
    print("$" * 50)


def send_data(udp_socket):
    """发送消息"""
    data = input("请输入消息:")
    ip = input("请输入ip地址:")
    port = int(input("请输入端口号:"))
    udp_socket.sendto(data.encode("gbk"), (ip, port))


def recv_data(udp_socket):
    """接受消息"""
    data, remote_adress = udp_socket.recvfrom(4396)
    print("收到了来自%s的信息:%s" % (str(remote_adress), data.decode("gbk")))


def main():
    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 绑定端口
    udp_socket.bind(('', 4397))
    while True:
        # 显示菜单
        menu()
        op = input("请输入")
        if op == "1":
            send_data(udp_socket)
        elif op == "2":
            recv_data(udp_socket)
        elif op == "3":
            break
        else:
            print("error")
    # 关闭套接字
    udp_socket.close()

if __name__ == '__main__':
    main()