多线程版udp聊天器

多线程版udp聊天器

import socket
import threading


def recv_msg(udp_socket):
    while True:
        # 4. 接收数据
        recv_data = udp_socket.recvfrom(1024)
        print('%s:%s' % (str(recv_data[1]), recv_data[0].decode('gb2312')))


def send_msg(udp_socket, dest_addr):
    while True:
        # 5. 发送数据
        send_data = input('请输入要发送的消息:')
        udp_socket.sendto(send_data.encode('gb2312'), dest_addr)


def main():
    # 1. 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 2. 绑定本地信息
    local_addr = udp_socket.bind(('', 7890))
    # 3. 获取对方ip和port
    dest_ip = input('请输入对方ip地址:')
    dest_port = int(input('请输入对方端口:'))
    dest_addr = (dest_ip, dest_port)
    # 6. 创建两个2线程，去执行相应的功能
    recv = threading.Thread(target=recv_msg, name='接收消息', args=(udp_socket,))
    send = threading.Thread(target=send_msg, name='发送消息', args=(udp_socket, dest_addr))
    recv.start()
    send.start()


if __name__ == '__main__':
    main()