单线程下udp聊天

效果展示

 网络调试助手下载链接:https://pan.baidu.com/s/1f9wTqRhAeyUN87XDcQUa7w提取码: kgur

import socket


def send_msg(udp_socket, send_data):
    """发送数据"""
    send_data = send_data
    udp_socket.sendto(send_data.encode('gbk'), ('192.168.2.101', 8080))


def recv_msg(udp_socket):
    """接收收据"""
    recv_data = udp_socket.recvfrom(1024)
    send_adress = recv_data[1]
    send_content = recv_data[0].decode('gbk')
    print('%s:%s' %(send_adress, send_content))


def main():
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    udp_socket.bind(('', 7788))
    # 循环收发数据
    while True:
        print("___UDP聊天器___")
        print("1.发送数据")
        print("2.接收数据")
        print("3.退出聊天")
        op = input("请输入功能：")
        if op == "1":
            send_data = input("请输入要发送的数据：")
            send_msg(udp_socket, send_data)
        elif op == "2":
            recv_msg(udp_socket)
        elif op == "3":
            break
        else:
            print("您的输入有误，请重新输入")


if __name__ == '__main__':
    main()