
    Pi|:                     0   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 n # e$ r e                    d           Y nw xY w	 ddlmZ n # e$ r e                    d           Y nw xY w	 ddlmZ n # e$ r e                    d	           Y nw xY w	 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%j&        e"           n # e$ r e                    d           Y nw xY wd Z'dddZ(ddddZ)	 dddddZ* G d de          Z+dS )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                 `    | 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    ^/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pyarrow/fs.py__getattr__r$   I   sX    }  
 
 	

 8888      Fuse_mmapc          	         t          | t                    r| S t          | t                    r|rt          d          t          j        |           \  }}|                    |          }|ri|                    |g          d         }|j        t          j	        k    r(t          d|j        j
         d|j         d|  d          t          ||          }|S 	 dd l}t          | |j                  rDt          |           j        dk    rt!          |          S t#          t%          |                     S n# t&          $ r Y nw xY wt)          d	t          |            d
          )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   s   *j)) )
	J	$	$ '
  	8   &z22D""4(( 
	/**F844Q7K8#555 )5@5E5J) )%0%5) ) &) ) )   #62..B			?MMM *f&?@@ ?
##,0AAA*H====#M*$=$=>>>	?  	 	 	D	 BZ(8(8 B B B
 
 	
s   D6 6
EE)
memory_mapc                   t          |           s|t          d          || fS |st          ||          }t          |t                    rt          |           } n$t          | t                    st          d          |                    |           } || fS t          |           } t	          |          }	 |	                    |           }|j
        t          j        k    }n# t          $ r d}d}Y nw xY w|sJ	 t          j        |           \  }} nF# t          $ r$}t          |          }d|v sd|v rn|Y d}~nd}~ww xY w|                    |           } || fS )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HH
j/22 	"4((DDD#&& 	*   ((..44  D
 !*555J?,,T22	
 $.H,==	    	  /
	)2488J 	 	 	a&&C$$(:c(A(A  	 ((..ts*   :C% %C65C6<D 
ED==Ei   T)
chunk_sizeuse_threadsc                   t          | |          \  }}t          ||          \  }}	|                    |          }
|
j        t          j        k    r't          |d          }t          ||||	||           dS t          ||||	||           dS )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$ ''44I~+++!+>>>
Y
+-='	6 	6 	6 	6 	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dS )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                     || _         d S N)r9   )selfr9   s     r#   __init__zFSSpecHandler.__init__  s    r%   c                 Z    t          |t                    r| j        |j        k    S t          S rV   r*   r6   r9   NotImplementedrW   others     r#   __eq__zFSSpecHandler.__eq__  )    e]++ 	'7eh&&r%   c                 Z    t          |t                    r| j        |j        k    S t          S rV   rZ   r\   s     r#   __ne__zFSSpecHandler.__ne__   r_   r%   c                 ^    | 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%  s5    7#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	          | |||                    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.  sn    F|<6!!MEE&\[((&EDD$Ee$dhhw6M6MNNNNr%   c                    g }|D ]}	 | j                             |          }|                    |                     ||                     G# t          $ r0 |                    t          |t          j                             Y w xY w|S rV   )r9   rq   appendrs   FileNotFoundErrorr   r   r?   )rW   pathsinfosr2   rq   s        r#   r/   zFSSpecHandler.get_file_info;  s     	A 	ADAw||D)) T33D$??@@@@ % @ @ @XdH,=>>?????@ s   A7BBc                 6   | j                             |j                  sP| j                             |j                  rt	          |j                  |j        rg S t          |j                  |j        rd }nd}g }| j                             |j        |dd          }|	                                D ]c\  }}|
                    d          }|j        
                    d          }||k    r)|                    |                     ||                     d|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%   w}}X.// 	?w~~h/00 ?():;;;+ ?I+H,=>>> 	HHH4 & 
 
 )..00 	A 	AJD$JJsOOE(..s33H   T33D$??@@@r%   c                 b    	 | j                             ||           d S # t          $ r Y d S w xY w)N)create_parents)r9   mkdirFileExistsError)rW   r2   rI   s      r#   
create_dirzFSSpecHandler.create_dirc  sG    	GMM$yM99999 	 	 	DD	s     
..c                 >    | j                             |d           d S NTrH   )r9   rmrh   s     r#   
delete_dirzFSSpecHandler.delete_dirj  s     

44
(((((r%   c                 J   	 | j                             |d          }n# t          $ r |rY d S  w xY w|D ]m}| j                             |          r| j                             |d           9| j                             |          r| j                             |           nd S )NF)r}   TrH   )r9   listdirrv   r   r   isfile)rW   r2   missing_dir_oksubpathssubpaths        r#   _delete_dir_contentsz"FSSpecHandler._delete_dir_contentsm  s    	wtE::HH  	 	 	 	   	$ 	$Gw}}W%% $

7d
3333(( $

7###		$ 	$s    00c                     |                     d          dk    rt          d|d          |                     ||           d S )Nr~    z$delete_dir_contents called on path 'r   )r   r,   r   )rW   r2   r   s      r#   delete_dir_contentsz!FSSpecHandler.delete_dir_contentsz  sO    ::c??b  6cC C C!!$77777r%   c                 2    |                      dd           d S )Nr~   F)r   )rW   s    r#   delete_root_dir_contentsz&FSSpecHandler.delete_root_dir_contents  s    !!#u-----r%   c                     | j                             |          st          |          | j                             |           d S rV   )r9   r   rv   r   rh   s     r#   delete_filezFSSpecHandler.delete_file  sB     w~~d## 	*#D)))

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

3
-----r%   c                 <    | j                             ||           d S rV   )r9   copyr   s      r#   	copy_filezFSSpecHandler.copy_file  s      	S$r%   c                     ddl m} | j                            |          st	          |           || 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  _    &&&&&&w~~d## 	*#D)))z$',,t$,77cBBBBr%   c                     ddl m} | j                            |          st	          |           || j                            |d          d          S r   r   r   s      r#   open_input_filezFSSpecHandler.open_input_file  r   r%   c                 \    ddl m}  || j                            |d          d          S )Nr   r   wbr   wr   r   r9   r   rW   r2   metadatar   s       r#   open_output_streamz FSSpecHandler.open_output_stream  ;    &&&&&&z$',,t$,77cBBBBr%   c                 \    ddl m}  || 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   	  sg            
  
$ $ $   
O 
O \
O	 	 	  :  ) ) )$ $ $8 8 8. . .  . . .     C C CC C CC C C
C 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      s[  $  8 7 7 7 7 7 7 7                          	,0000000 , , ,*+++++,-....... - - -+,,,,,-*,,,,,,, * * *)))))*)L L L L L L L L L L L L L L L L L L L L L L L L MMMFO'((((  ) ) )((((()	 	 	 05 *
 *
 *
 *
 *
Z9e 9 9 9 9 9z ?CH-&DH- H- H- H- H-VcC cC cC cC cC% cC cC cC cC cCsE   1 AAA A65A6:B BB"C C,+C,