用UDP连接 创建一个简易聊天工具

269 阅读1分钟
import socket


 def recv_message(udp_socket): 

 """使用UDP进行数据接收"""

 recv_data, remote_address = udp_socket.recvfrom(4096) 

 print("接收到来自%s的数据:%s" % (str(remote_address), recv_data.decode())) 


 def send_message(udp_socket): 

 """使用UDP进行数据发送""" 

 # 输入数据 

 data = input("请输入需要发送的数据:") 

 # 输入IP 端口 

 ip = input("请输入接收人的IP地址:") 

 port = int(input("接收的端口:"))

 udp_socket.sendto(data.encode(), (ip, port)) 


 def main(): 

 # 创建UDP套接字 udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # 绑定端口  udp_socket.bind(('', 8888)) 

 while True: 

 # 获取用户输入

 op = input("请输入你要进行的操作:

 1 发送数据 

2接收数据 

3退出") 

 if op == '1': 

 send_message(udp_socket) 

 elif op == '2': 

 recv_message(udp_socket) 

 elif op == '3': break else: 

 print("亲 没有这个选项 请重新输入")

 if __name__ == '__main__': 

 main()