import socket def send_msg(udp_socket): """获取键盘数据,并将其发送给对方""" # 1. 从键盘输入数据 msg = input("\n请输入要发送的数据:") # 2. 输入对方的ip地址 addr = input("\n请输入对方的ip地址:") # 3. 输入对方的port port = int(input("\n请输入对方的port:")) # 4. 发送数据 udp_socket.sendto(msg.encode("utf-8"), (addr, port)) 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) # 2. 绑定本地信息 udp_socket.bind(("", 7890)) while True: # 3. 选择功能 print("=" * 30) print("1:发送消息") print("2:接收消息") print("=" * 30) num = input("请输入要操作的功能序号:") # 4. 根据选择调用相应的函数 if num == "1": send_msg(udp_socket) elif num == "2": recv_msg(udp_socket) else: print("输入有误,请重新输入...") if __name__ == "__main__": main()
截图测试