python 关于 udp聊天器

import socket
def send_msg(udp_socket):
    # 获取输入
    msg = input("请输入要发送的数据：")
    # 输入对方IP
    dest_ip = input("请输入对方ip：")
    # 输入对方port
    dest_port = int(input("请输入对方端口："))
    # 编码
    udp_b = msg.encode('utf-8')
    # 发送数据
    udp_socket.sendto(udp_b,(dest_ip,dest_port))

def recv_msg(udp_socket):
    recv_msg = udp_socket.recvfrom(1024)
    # 解码
    # udp_j = recv_msg.decode('utf-8')
    # 显示
    recv_msg = recv_msg[0].decode('utf-8')
    print(recv_msg)
def main():
    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    # 绑定本地信息
    udp_socket.bind(("",7789))
    while True:
        print("=" * 30)
        print("1.发送消息")
        print("2.接收消息")
        print("0.退出聊天")
        op_num = input("请输入要操作的功能：")
        if op_num == "1":
            send_msg(udp_socket)
        elif op_num == "2":
            recv_msg(udp_socket)
        elif op_num == "0":
            break
        else:
            print("输入有误，请重新输入")


    udp_socket.close()

if __name__== main():
    main()