【socket】做一个最简单的udp聊天室

这是个最最最简单的版本，只能发送一句接收一句循环，简单的练练手
大概逻辑就是这样，以后有机会在优化吧

import socket


def main():
    # 创建一个udp套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    # 绑定本地信息
    local_addr = ('', 7788)  # ip一般不用写，表示本机任何一个ip
    udp_socket.bind(local_addr)

    # 获取对方ip/port
    dest_ip = input("请输入对方ip：")
    dest_port = int(input("请输入对方port："))
    # 接收方地址192.168.100.1
    dest_addr = (dest_ip, dest_port)

    while True:

        # 从键盘获取发送数据内容
        send_data = input("请输入要发送内容：")

        # 输入exit退出
        if send_data == "exit":
            break

        udp_socket.sendto(send_data.encode("gbk"), dest_addr)

        # 接收回送过来的数据
        recv_data = udp_socket.recvfrom(1024)
        recv_msg = recv_data[0]
        # dest_addr = recv_data[1]
        print(recv_msg.decode('gbk'))

    # 关闭套接字
    udp_socket.close()


if __name__ == "__main__":
    main()

放一张运行效果图

做了一点点优化（真的真的要继续学别的东西了，可是这个东西好好玩，九命）

import socket


def send_msg(udp_socket, dest_addr):
    """发送消息"""
    if dest_addr == []:
        # 获取对方ip/port
        dest_ip = input("请输入对方ip：")
        dest_port = int(input("请输入对方port："))
        dest_addr = (dest_ip, dest_port)
    # 从键盘获取发送数据内容
    send_data = input("请输入要发送内容：")
    udp_socket.sendto(send_data.encode("gbk"), dest_addr)


def recv_msg(udp_socket):
    """接收数据"""
    recv_data = udp_socket.recvfrom(1024)
    recv_msg = recv_data[0]
    dest_addr = recv_data[1]
    print("%s:%s" % (str(dest_addr), recv_msg.decode("gbk")))
    return dest_addr


def main():
    # 创建一个udp套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    # 绑定本地信息
    local_addr = ('', 7788)  # ip一般不用写，表示本机任何一个ip
    udp_socket.bind(local_addr)
    dest_addr = []

    print("-----聊天器-----")
    print("1:发送信息")
    print("2:接收信息")
    print("0:退出系统")

    while True:
        op = input("请输入功能对应的序号：")
        if op == '1':
            # send
            send_msg(udp_socket, dest_addr)
        elif op == '2':
            # receive
            dest_addr = recv_msg(udp_socket)
        elif op == '0':
            break
        else:
            print("输入有误请重新输入...")

    # 关闭套接字
    udp_socket.close()


if __name__ == "__main__":
    main()