
    eiu:                        d Z ddlmZmZ ddlmZmZmZmZm	Z	m
Z
mZmZmZmZmZ eZg Z	 ddlmZ 	 ddlmZ 	 ddlmZ 	 dd
lmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$ ddl%Z% e%jL                  e"       d Z'dddZ(ddddZ)	 dddddZ* G d de      Z+y# e$ r ej+                  d       Y zw xY w# e$ r ej+                  d       Y w xY w# e$ r ej+                  d	       Y w xY w# e$ r ej+                  d       Y w xY w)zO
FileSystem abstraction to interact with various local and remote filesystems.
    )_is_path_like_stringify_path)FileSelectorFileTypeFileInfo
FileSystemLocalFileSystemSubTreeFileSystem_MockFileSystemFileSystemHandlerPyFileSystem_copy_files_copy_files_selector)AzureFileSystemr   )HadoopFileSystemr   )GcsFileSystemr   )
AwsDefaultS3RetryStrategyAwsStandardS3RetryStrategyS3FileSystem
S3LogLevelS3RetryStrategyensure_s3_initializedfinalize_s3ensure_s3_finalizedinitialize_s3resolve_s3_regionNr   c                 N    | t         v rt        d|  d      t        d|  d      )Nz8The pyarrow installation is not built with support for ''z&module 'pyarrow.fs' has no attribute ')_not_importedImportErrorAttributeError)names    L/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/pyarrow/fs.py__getattr__r$   I   sC    }vQ
 	

 
0a8     Fuse_mmapc          	         t        | t              r| S t        | t              r|rt        d      t        j                  |       \  }}|j                  |      }|rq|j                  |g      d   }|j                  t        j                  k7  r3t        d|j                  j                   d|j                   d|  d      t        ||      }|S 	 dd l}t        | |j                        r8t        |       j                  dk(  rt!        |      S t#        t%        |             S t)        d	t        |        d
      # t&        $ r Y #w xY w)NzXSpecifying to use memory mapping not supported for filesystem specified as an URI stringr   zWThe path component of the filesystem URI must point to a directory but it has a type: `z`. The path component is `z#` and the given filesystem URI is ``r	   r&   zUnrecognized filesystem: zP. `filesystem` argument must be a FileSystem instance or a valid file system URI)
isinstancer   str
ValueErrorfrom_urinormalize_pathget_file_infotyper   	Directoryr"   pathr
   fsspecAbstractFileSystem__name__r	   r   FSSpecHandlerr    	TypeError)
filesystemr'   fsr2   prefixprefix_infor3   s          r#   _ensure_filesystemr<   U   sd   *j)	J	$ 8  &&z2D""4(**F84Q7K8#5#55 55@5E5E5J5J4K L%%0%5%5$6 7%,a)  #62.B			? *f&?&?@
#,,0AA*H==#M*$=>>'Z(8'9 :B B
 	
  		s   D9 9	EE)
memory_mapc                h   t        |       s|t        d      || fS |Yt        ||      }t        |t              rt        |       } nt        | t              st        d      |j                  |       } || fS t        |       } t	        |      }	 |j                  |       }|j                  t        j                  k7  }|s	 t        j                  |       \  }} || fS |j                  |       } || fS # t        $ r d}d}Y Cw xY w# t        $ r#}t        |      }d|v sd|v rn|Y d}~|| fS d}~ww xY w)z
    Return filesystem/path from path which could be an URI or a plain
    filesystem path or a combination of fsspec protocol and URI.
    Nzg'filesystem' passed but the specified path is file-like, so there is nothing to open with 'filesystem'.r&   zPExpected string path; path-like objects are only allowed with a local filesystemFzempty schemezCannot parse URI)r   r,   r<   r*   r	   r   r+   r7   r.   r/   r0   r   NotFoundr   r-   )r2   r8   r=   	file_infoexists_locallyemsgs          r#   _resolve_filesystem_and_pathrD      s{   
 !?  4'
ZH
j/2"4(DD#&*  ((.44 D
 !*5J?,,T2	
 $..H,=,== 
	)2248J t ((.t/  	  	a&C$(:c(A   t	s*   C2 D 2DD	D1D,,D1i   T)
chunk_sizeuse_threadsc                    t        | |      \  }}t        ||      \  }}	|j                  |      }
|
j                  t        j                  k(  rt        |d      }t        ||||	||       yt        ||||	||       y)a  
    Copy files between FileSystems.

    This functions allows you to recursively copy directories of files from
    one file system to another, such as from S3 to your local machine.

    Parameters
    ----------
    source : string
        Source file path or URI to a single file or directory.
        If a directory, files will be copied recursively from this path.
    destination : string
        Destination file path or URI. If `source` is a file, `destination`
        is also interpreted as the destination file (not directory).
        Directories will be created as necessary.
    source_filesystem : FileSystem, optional
        Source filesystem, needs to be specified if `source` is not a URI,
        otherwise inferred.
    destination_filesystem : FileSystem, optional
        Destination filesystem, needs to be specified if `destination` is not
        a URI, otherwise inferred.
    chunk_size : int, default 1MB
        The maximum size of block to read before flushing to the
        destination file. A larger chunk_size will use more memory while
        copying but may help accommodate high latency FileSystems.
    use_threads : bool, default True
        Whether to use multiple threads to accelerate copying.

    Examples
    --------
    Inspect an S3 bucket's files:

    >>> s3, path = fs.FileSystem.from_uri(
    ...            "s3://registry.opendata.aws/roda/ndjson/")
    >>> selector = fs.FileSelector(path)
    >>> s3.get_file_info(selector)
    [<FileInfo for 'registry.opendata.aws/roda/ndjson/index.ndjson':...]

    Copy one file from S3 bucket to a local directory:

    >>> fs.copy_files("s3://registry.opendata.aws/roda/ndjson/index.ndjson",
    ...               f"file:///{local_path}/index_copy.ndjson")

    >>> fs.LocalFileSystem().get_file_info(str(local_path)+
    ...                                    '/index_copy.ndjson')
    <FileInfo for '.../index_copy.ndjson': type=FileType.File, size=...>

    Copy file using a FileSystem object:

    >>> fs.copy_files("registry.opendata.aws/roda/ndjson/index.ndjson",
    ...               f"file:///{local_path}/index_copy.ndjson",
    ...               source_filesystem=fs.S3FileSystem())
    T	recursiveN)rD   r/   r0   r   r1   r   r   r   )sourcedestinationsource_filesystemdestination_filesystemrE   rF   	source_fssource_pathdestination_fsdestination_pathr@   
source_sels               r#   
copy_filesrS      s    p :!I{ (D+($N$ ''4I~~+++!+>
Y
+-='	6 	I{"$4	-r%   c                       e Zd ZdZd Zd Zd Zd Zd Ze	d        Z
d Zd	 Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zy)r6   a  
    Handler for fsspec-based Python filesystems.

    https://filesystem-spec.readthedocs.io/en/latest/index.html

    Parameters
    ----------
    fs : FSSpec-compliant filesystem instance

    Examples
    --------
    >>> PyFileSystem(FSSpecHandler(fsspec_fs)) # doctest: +SKIP
    c                     || _         y N)r9   )selfr9   s     r#   __init__zFSSpecHandler.__init__  s	    r%   c                 `    t        |t              r| j                  |j                  k(  S t        S rV   r*   r6   r9   NotImplementedrW   others     r#   __eq__zFSSpecHandler.__eq__  %    e]+77ehh&&r%   c                 `    t        |t              r| j                  |j                  k7  S t        S rV   rZ   r\   s     r#   __ne__zFSSpecHandler.__ne__   r_   r%   c                 b    | j                   j                  }t        |t              r|d   }d| S )Nr   zfsspec+)r9   protocolr*   list)rW   rc   s     r#   get_type_namezFSSpecHandler.get_type_name%  s1    77##h%{H
##r%   c                     |S rV    rW   r2   s     r#   r.   zFSSpecHandler.normalize_path+  s    r%   c           	          |d   }|d   dk(  rt         j                  }n+|d   dk(  rt         j                  }d }nt         j                  }t	        | |||j                  dd             S )Nsizer0   file	directorymtime)rj   rm   )r   Filer1   Unknownr   get)r2   inforj   ftypes       r#   _create_file_infozFSSpecHandler._create_file_info.  sc    F|<6!MME&\[(&&ED$$Ee$dhhw6MNNr%   c                    g }|D ]?  }	 | j                   j                  |      }|j                  | j                  ||             A |S # t        $ r, |j                  t        |t        j                               Y xw xY wrV   )r9   rq   appendrs   FileNotFoundErrorr   r   r?   )rW   pathsinfosr2   rq   s        r#   r/   zFSSpecHandler.get_file_info;  s~     	ADAww||D) T33D$?@	A 	 % @XdH,=,=>?@s   A		2A>=A>c                 X   | j                   j                  |j                        s]| j                   j                  |j                        rt	        |j                        |j
                  rg S t        |j                        |j                  rd }nd}g }| j                   j                  |j                  |dd      }|j                         D ]X  \  }}|j                  d      }|j                  j                  d      }||k7  s8|j                  | j                  ||             Z |S )N   T)maxdepthwithdirsdetail/)r9   isdirbase_direxistsNotADirectoryErrorallow_not_foundrv   rI   finditemsstripru   rs   )	rW   selectorr{   rx   selected_filesr2   rq   _pathr   s	            r#   get_file_info_selectorz$FSSpecHandler.get_file_info_selectorF  s	   ww}}X../ww~~h//0():):;;++I+H,=,=>>HH4 & 
 )..0 	AJD$JJsOE((..s3H  T33D$?@	A r%   c                 ^    	 | j                   j                  ||       y # t        $ r Y y w xY w)N)create_parents)r9   mkdirFileExistsError)rW   r2   rI   s      r#   
create_dirzFSSpecHandler.create_dirc  s-    	GGMM$yM9 		s     	,,c                 >    | j                   j                  |d       y NTrH   )r9   rmrh   s     r#   
delete_dirzFSSpecHandler.delete_dirj  s    

44
(r%   c                 R   	 | j                   j                  |d      }|D ]r  }| j                   j                  |      r| j                   j	                  |d       <| j                   j                  |      sX| j                   j	                  |       t y # t        $ r |rY y  w xY w)NF)r}   TrH   )r9   listdirrv   r   r   isfile)rW   r2   missing_dir_oksubpathssubpaths        r#   _delete_dir_contentsz"FSSpecHandler._delete_dir_contentsm  s    	wwtE:H
   	$Gww}}W%

7d
3(

7#		$	 ! 		s   B B&$B&c                 j    |j                  d      dk(  rt        d|d      | j                  ||       y )Nr~    z$delete_dir_contents called on path 'r   )r   r,   r   )rW   r2   r   s      r#   delete_dir_contentsz!FSSpecHandler.delete_dir_contentsz  s8    ::c?b 6cC C!!$7r%   c                 &    | j                  d       y )Nr~   )r   )rW   s    r#   delete_root_dir_contentsz&FSSpecHandler.delete_root_dir_contents  s    !!#&r%   c                     | j                   j                  |      st        |      | j                   j                  |       y rV   )r9   r   rv   r   rh   s     r#   delete_filezFSSpecHandler.delete_file  s/     ww~~d##D))

4r%   c                 @    | j                   j                  ||d       y r   )r9   mvrW   srcdests      r#   movezFSSpecHandler.move  s    

3
-r%   c                 <    | j                   j                  ||       y rV   )r9   copyr   s      r#   	copy_filezFSSpecHandler.copy_file  s     	S$r%   c                     ddl m} | j                  j                  |      st	        |       || j                  j                  |d      d      S Nr   
PythonFilerbmoderpyarrowr   r9   r   rv   openrW   r2   r   s      r#   open_input_streamzFSSpecHandler.open_input_stream  >    &ww~~d##D))$'',,t$,7cBBr%   c                     ddl m} | j                  j                  |      st	        |       || j                  j                  |d      d      S r   r   r   s      r#   open_input_filezFSSpecHandler.open_input_file  r   r%   c                 X    ddl m}  || j                  j                  |d      d      S )Nr   r   wbr   wr   r   r9   r   rW   r2   metadatar   s       r#   open_output_streamz FSSpecHandler.open_output_stream  #    &$'',,t$,7cBBr%   c                 X    ddl m}  || j                  j                  |d      d      S )Nr   r   abr   r   r   r   s       r#   open_append_streamz FSSpecHandler.open_append_stream  r   r%   N)r5   
__module____qualname____doc__rX   r^   ra   re   r.   staticmethodrs   r/   r   r   r   r   r   r   r   r   r   r   r   r   r   rg   r%   r#   r6   r6   	  s    

$ 
O 
O	:)$8'. CCC
Cr%   r6   rV   )NN),r   pyarrow.utilr   r   pyarrow._fsr   r   r   r   r	   r
   r   r   r   r   r   	FileStatsr   pyarrow._azurefsr   r    ru   pyarrow._hdfsr   pyarrow._gcsfsr   pyarrow._s3fsr   r   r   r   r   r   r   r   r   r   atexitregisterr$   r<   rD   rS   r6   rg   r%   r#   <module>r      s4  $ 8    	,0-.*,)L L L FOO'(	 05 *
Z9e 9z ?CH-&DH-VcC% cC{  ,*+,
  -+,-
  *)*  )()sD   B B. C
 C& B+*B+.CC
C#"C#&C?>C?