
    gz                     N    d dl Z d dlZd dlmZ d dlmZ d dlmZ  G d de      Zy)    N)Optional)unquote)AbstractFileSystemc                   f     e Zd ZdZdZ fdZd	dZd Z	 	 	 	 d
dZe	dde
dee   fd       Z xZS )DataFileSystemzA handy decoder for data-URLs

    Example
    -------
    >>> with fsspec.open("data:,Hello%2C%20World%21") as f:
    ...     print(f.read())
    b"Hello, World!"

    See https://developer.mozilla.org/en-US/docs/Web/HTTP/Basics_of_HTTP/Data_URLs
    datac                 $    t        |   di | y)z!No parameters for this filesystemN )super__init__)selfkwargs	__class__s     P/var/www/openai/venv/lib/python3.12/site-packages/fsspec/implementations/data.pyr   zDataFileSystem.__init__   s    "6"    c                     |j                  dd      \  }}|j                  d      rt        j                  |      || S t	        |      j                         || S )N,   base64)splitendswithr   	b64decoder   encode)r   pathstartendr   prefr   s          r   cat_filezDataFileSystem.cat_file   sT    ZZQ'
d=="##D)%44t}##%eC00r   c                     |j                  dd      \  }}| j                  |      }|j                  dd      d   j                  dd      d   }|t        |      d|dS )Nr   r   :;r   file)namesizetypemimetype)r   r   len)r   r   r   r   r#   r   mimes          r   infozDataFileSystem.info!   s^    ZZQ'
d}}T"zz#q!!$**3215c$iTRRr   c                 h    d|vrt        d      t        j                  | j                  |            S )NrzRead only filesystem)
ValueErrorioBytesIOr   )r   r   mode
block_size
autocommitcache_optionsr   s          r   _openzDataFileSystem._open'   s/     d?344zz$---..r   r(   c                 \    d|xs d dt        j                  |       j                          S )zFormat the given data into data-URL syntax

        This version always base64 encodes, even when the data is ascii/url-safe.
        zdata: z;base64,)r   	b64encodedecode)r   r(   s     r   r   zDataFileSystem.encode4   s0     tzrl(6+;+;D+A+H+H+J*KLLr   )NN)rbNTN)N)__name__
__module____qualname____doc__protocolr   r   r)   r3   staticmethodbytesr   strr   __classcell__)r   s   @r   r   r   	   s\    	 H#1S / MU M(3- M Mr   r   )	r   r-   typingr   urllib.parser   fsspecr   r   r
   r   r   <module>rE      s#     	    %1M' 1Mr   