Python实现多任务版的udp聊天器

import socket
import threading

def send_msg(udp_socket):
 """获取键盘数据，并将其发送给对方"""
 while True:
  # 1. 从键盘输入数据
  msg = input("\n请输入要发送的数据:")
  # 2. 输入对方的ip地址
  dest_ip = input("\n请输入对方的ip地址:")
  # 3. 输入对方的port
  dest_port = int(input("\n请输入对方的port:"))
  # 4. 发送数据
  udp_socket.sendto(msg.encode("utf-8"), (dest_ip, dest_port))

def recv_msg(udp_socket):
 """接收数据并显示"""
 while True:
  # 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))
 # 3. 创建一个子线程用来接收数据
 t = threading.Thread(target=recv_msg, args=(udp_socket,))
 t.start()
 # 4. 让主线程用来检测键盘数据并且发送
 send_msg(udp_socket)

if __name__ == "__main__":
 main()