python-udp聊天器

import socket


def send_udp(udp_socket, dest_ip, dest_port):
    # 发送
    send_data = input('请输入要发送的内容：')
    udp_socket.sendto(send_data.encode('utf-8'), (dest_ip, dest_port))

def recv_udp(udp_socket):
    # 接收返回的数据
    recv_data = udp_socket.recvfrom(1024)
    print(recv_data[0].decode('gbk'), recv_data[1])

def main():
    # 创建一个udp套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    #绑定信息
    udp_socket.bind(('', 7788))
    # 获取对方的ip/port
    dest_ip = input('请输入对方的IP:')
    dest_port = int(input('请输入对方的端口:'))

    while True:
        print('-----制作聊天器------')
        print('1: 发送消息')
        print('2：接收消息')
        print('0: 退出系统')
        udp_input = input('请输入功能：')
        if udp_input == '1':
            # 发送
            send_udp(udp_socket, dest_ip, dest_port)
        elif udp_input == '2':
            # 接收返回的数据
            recv_udp(udp_socket)
        elif udp_input == '0':
            break
        else:
            print('输入错误，请重新输入')


    #关闭套接字
    udp_socket.close()

if __name__ == '__main__':
    main()