python：利用socket打造udp聊天器

测试环境及工具

• python 3.7.5
• windows 10
• 测试工具： ScoketToolV4
• pycharm 2019.3.3

说明

在python中，我们使用的模块是socket，因为这个模块是python的标准库，所以可以不用安装，使用的测试工具:工具下载地址

UDP发送数据 - 客户端

首先先写一个upd发送信息的代码，由于改写的注释都在代码中提到了，所以这里我就不多BB了。

循环发送数据的udp代码

import socket  # 导入包


def udp_socket():
    # 创建udp套接字,socket.socket()中传入两个参数，第一个为ip类型协议（socket.AF_IENT指的是ipv4协议），第二个时传输协议（这里指的是udp协议）
    u_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    while True:
        # 发送数据
        send_data = input("请输入要发送的数据:")
        if send_data == "exit":  # 如果输入字符串是exit，程序退出
            break
        address = ("192.168.79.1", 9999)  # 目的IP地址和端口号, ip地址为字符串类型，而端口号为int类型
        u_socket.sendto(send_data.encode('utf-8'), address)  # 在发送时，要将发送的内容进行编码，以字节的方式传输

    u_socket.close()  # 关闭套接字


if __name__ == "__main__":
    udp_socket()

运行程序，使用网络测试工具进行调试
首先创建一个udp服务端，然后将端口设置为9999，这是因为我们在代码中绑定的端口号为9999

然后运行程序，并发送一些数据

可以看到，在网络调试已经接收到了我们所发送的消息，并且当输入exit时，程序退出

UDP接收数据 - 服务器

有了发送，那么也会有接收部分，关于相应的功能我也在代码中写好注释了，这里也不用赘述了。

import socket


def recv_data():
    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    local_adress = ("", 7788)  # 类型为元组，绑定本地信息,在绑定本地信息时,ip地址可以不用写
    udp_socket.bind(local_adress)  # 必须绑定自己电脑的ip以及port，其他的不行
    print("正在接收数据...")
    while True:  # 循环接收数据
        # 接收数据
        data = udp_socket.recvfrom(1024)  # 最大支持接收1024个字节
        content = data[0]  # 发送的内容
        if content.decode("gbk") == "exit":  # 当收到数据为exit时，程序退出
            break
        send_address = data[1]  # 发送方IP地址和端口号
        # 为什么解码方式为gbk，因为在windows中默认的解码方式为gbk，所以utf-8不能识别，但在其他操作系统中使用utf-8进行解码
        print("%s发送：%s" % (str(send_address), content.decode("gbk")))

    udp_socket.close()  # 关闭套接字


if __name__ == '__main__':
    recv_data()

启动程序，并在网络调试助手中创建一个udp客户端用来连接，并发送数据

运行程序时，程序处于堵塞状态，这是由于它正在等待客户端向他发送数据，当我们发送exit时，程序退出

UDP聊天器

有了收数据和发数据的功能，那么一个UDP聊天器的功能就不难实现了，代码如下

import socket


def send_msg(udp_socket):
    dest_ip = input("请输入要发送的IP地址：")
    dest_port = int(input("请输入对方端口号："))
    data = input("请输入你要发送的内容：")
    udp_socket.sendto(data.encode('gbk'), (dest_ip, dest_port))  # 发送的内容


def recv_msg(udp_socket):
    recv_data = udp_socket.recvfrom(1024)  # 最大接收1024个字节

    content = recv_data[0].decode('gbk')  # 返回的内容
    send_ip = str(recv_data[1])  # 发送方的ip地址以及端口号

    print("%s发送：%s" % (send_ip, content))


def main():
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)  # 创建套接字
    local_adress = ("", 7788)
    udp_socket.bind(local_adress)
    # 发送数据

    while True:
        print("-----ocean聊天器-----")
        print("1:发送消息")
        print("2:接收消息")
        print("0:退出系统")
        option = input("请选择功能：")
        if option == "1":
            # 发送数据
            send_msg(udp_socket)
        elif option == "2":
            # 接收数据
            recv_msg(udp_socket)
        elif option == "0":
            break
        else:
            print("输入错误，请重新输入")
    udp_socket.close()


if __name__ == '__main__':
    main()

这样一个简单的udp聊天器就完成了。

小结

这里我画一个脑图，做个小结