博客

import socket
import threading


def recv_msg(udp_socket):
    """接收数据并显示"""
    # 接收数据
    while True:
        recv_data = udp_socket.recvfrom(1024)
        # 解析出收到的数据
        # recv_data这个变量接收的是一个元组（接收到的数据（发送方ip, port））
        recv_msg = recv_data[0]  # 存储接收到的数据
        send_addr = recv_data[1]  # 存储发送方ip和端口

        # 3.打印接收到的数据
        print("收到的信息是:%s \n收到的地址:%s" % (recv_msg.decode("gbk"), str(send_addr)))


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))


def main():
    """完成udp聊天器的整体控制"""
    # 1.创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    
    # 2.绑定本地信息
    udp_socket.bind(("", 7788))

    # 3.获取对方ip
    dest_ip = input("请输入对方ip:")
    dest_port = int(input("请输入对方端口："))

    # 4.创建2个线程，去执行相应的功能
    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()