多任务版 UDP 聊天

高级编程技巧 学习笔记

一、多任务版 UDP 聊天

1. 创建套接字

2. 绑定本地信息

3. 获取对方IP和端口

4. 发送、接收数据

5. 创建两个线程，去执行功能

import socket
import threading


def recv_msg(udp_socket):
    # 接收数据
    while True:
        recv_data = udp_socket.recvfrom(1024)
        print()
        print('接收到的数据：%s\r' % recv_data[0].decode('gbk'))


def send_msg(udp_socket, dest_ip, dest_port):
    # 发送数据
    while True:
        send_data = input("输入要发送的数据：")
        udp_socket.sendto(send_data.encode('gbk'), (dest_ip, dest_port))
        # udp_socket.sendto(send_data.encode('gbk'), ('192.168.0.105', 7890))


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

    # 绑定
    udp_socket.bind(("", 7890))

    # 获取对方的 IP 和 端口
    dest_ip = input("请输入对方的IP：")
    dest_port = int(input("请输入对方的 port："))


    t_recv = threading.Thread(target=recv_msg, args=(udp_socket,))
    t_send = threading.Thread(target=send_msg, args=(udp_socket, dest_ip, dest_port))
    
    t_recv.start()
    t_send.start()


if __name__ == '__main__':
    main()