socket网络编程之udp聊天器

 1 from socket import *
 2 
 3 def send_msg(udp_socket):
 4     ‘‘‘发送消息‘‘‘
 5     # 获取要发送的内容
 6     dest_ip = input("请输入对方IP：")
 7     dest_port = int(input("请输入对方port："))
 8     send_data = input("请输入要发送的消息：")
 9     udp_socket.sendto(send_data.encode(‘utf-8‘),(dest_ip,dest_port))
10 
11 def recv_msg(udp_socket):
12     """接收消息"""
13     recv_data = udp_socket.recvfrom(1024)
14     print("%s:%s" % (str(recv_data[1]), recv_data[0].decode(‘utf-8‘)))
15 
16 def main():
17     # 创建套接字
18     udp_socket = socket(AF_INET, SOCK_DGRAM)
19     udp_socket.bind(("", 7899))
20     # while循环来处理事情
21     while True:
22         print("----微Q聊天器-----")
23         print("1|发送消息")
24         print("2|接收消息")
25         print("3|退出程序")
26         op = input("请选择功能")
27         if op == "1":
28             send_msg(udp_socket)
29         elif op == "2":
30             recv_msg(udp_socket)
31         elif op == "0":
32             break
33         else:
34             print("输入有误")
35 
36 
37 if __name__ == ‘__main__‘:
38     main()