python web02-02-案例：udp聊天

python web02-02-案例：udp聊天

缺陷，该UDP代码只能发一次消息，接受一次消息。 后面会慢慢对该问题进行解决

import socket


def send_msg(udp_socket):
    """发送消息"""
    # 获取要发送的内容
    # dest_ip = input("请输入对方的ip:")
    # dest_port = int(input("请输入对方的port:"))
    
    # 测试端口，默认使用上面的方式
    dest_ip = "192.168.31.226"
    dest_port = 8080
    send_data = input("请输入要发送的消息: ")
    # udp_socket.sendto(send_data.encode("utf-8"), (dest_ip, dest_port))
    udp_socket.sendto(send_data.encode("gbk"), (dest_ip, dest_port))


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


def main():
    # 1\创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    # 2、绑定信息
    udp_socket.bind(("", 7788))

    # 3、循环循环来进行处理事情
    while True:
        # 发送
        send_msg(udp_socket)

        # 接收并显示
        recv_msg(udp_socket)
    # 4、关闭套接字
    udp_socket.close()
if __name__ == "__main__":
    main()