多任务版udp聊天器

多任务版udp聊天器

import socket
import threading


def recv_msg(udp_socket):
    # 3.接收数据
    while True:
        recv_data = udp_socket.recvfrom(1024)
        resv_ip = recv_data[1]
        recv_msg = recv_data[0].decode('gbk')
        print('来自{}发送的数据\n{}'.format(resv_ip, recv_msg))


def send_msg(udp_socket, dest_ip, dest_port):
    # 4.发送数据
    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(("", 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()