o
    g;                     @   s   d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlZd dl	Z	d dl
mZ d dlmZ d dlmZ d dlmZmZ edZG dd deZd	d
 Zdd ZG dd dejZdS )    N)AbstractFileSystem)compr)get_compression)
isfilelikestringify_pathzfsspec.localc                       s(  e Zd ZdZdZdZdZdA fdd	Zedd	 Z	dBd
dZ
dAddZdd ZdAddZdd Zdd Zdd Zdd Zdd ZdCddZdCdd Zd!d" Zd#d$ Zd%d& Zd'efd(d)Zd*d+ ZdDd,d-Zd.d/ ZdEd1d2ZdBd3d4Zd5d6 Zd7d8 Z e!d9d: Z"e!d;d< Z#d=d> Z$d?d@ Z%  Z&S )FLocalFileSystema   Interface to files on local storage

    Parameters
    ----------
    auto_mkdir: bool
        Whether, when opening a file, the directory containing it should
        be created (if it doesn't already exist). This is assumed by pyarrow
        code.
    /)filelocalTFc                    s   t  jdi | || _d S )N )super__init__
auto_mkdir)selfr   kwargs	__class__r   S/var/www/visachat/venv/lib/python3.10/site-packages/fsspec/implementations/local.pyr   !   s   
zLocalFileSystem.__init__c                 C      dS )Nr
   r   r   r   r   r   fsid%   s   zLocalFileSystem.fsidc                 K   sH   |  |}| |rt||r| j|dd d S tj|fi | d S NTexist_ok)_strip_protocolexistsFileExistsErrormakedirsosmkdir)r   pathcreate_parentsr   r   r   r   r   )   s   

zLocalFileSystem.mkdirc                 C   s   |  |}tj||d d S )Nr   )r   r   r   )r   r    r   r   r   r   r   2   s   
zLocalFileSystem.makedirsc                 C   s   |  |}t| d S N)r   r   rmdirr   r    r   r   r   r#   6   s   
zLocalFileSystem.rmdirc              
   K   s   |  |}| |}|d dkr@t|"}g }|D ]}z
|| | W q ty/   Y qw W d    n1 s:w   Y  n|g}|sLdd |D S |S )Ntype	directoryc                 S   s   g | ]}|d  qS )namer   ).0ir   r   r   
<listcomp>I   s    z&LocalFileSystem.ls.<locals>.<listcomp>)r   infor   scandirappendFileNotFoundError)r   r    detailr   r+   itinfosfr   r   r   ls:   s$   

zLocalFileSystem.lsc           
      K   s8  t |tjrG|jdd}| }|jddrd}n|jddr"d}nd}|j}|r@z|jdd}|j}W n ty?   d}Y nw | 	|j
}n2| 	|}tj|dd}t|j}|rbtj|dd}|j}t|jrnd}nt|jrwd}nd}||||j|d}d	D ]}	t|d
|	 ||	< q|rt||d< |S )NF)follow_symlinksr&   r	   otherTr   )r'   sizer%   createdislink)modeuidgidmtimeinonlinkst_destination)
isinstancer   DirEntrystat
is_symlinkis_diris_filest_sizeOSErrorr   r    S_ISLNKst_modeS_ISDIRS_ISREGst_ctimegetattrreadlink)
r   r    r   outlinktr6   out2resultfieldr   r   r   r+   L   sN   

zLocalFileSystem.infoc                 K   s
   t |S r"   )osplexists)r   r    r   r   r   r   rW   {      
zLocalFileSystem.lexistsc                 K   sl   |  |}|  |}| jr| j| |dd | |r$t|| d S | |r2| j|dd d S t	|r   )
r   r   r   _parentisfileshutilcopyfileisdirmkdirsr.   r   path1path2r   r   r   r   cp_file~   s   



zLocalFileSystem.cp_filec                 C      |  |}tj|S r"   )r   r   r    rZ   r$   r   r   r   rZ         
zLocalFileSystem.isfilec                 C   rc   r"   )r   r   r    r]   r$   r   r   r   r]      rd   zLocalFileSystem.isdirNc                 K   sX   t |r"t|d}t|| W d    d S 1 sw   Y  d S | j||fi |S )Nrb)r   openr[   copyfileobjrb   )r   r`   ra   callbackr   r2   r   r   r   get_file   s
   "zLocalFileSystem.get_filec                 K   s   | j ||fi |S r"   )rb   )r   r`   ra   rh   r   r   r   r   put_file      zLocalFileSystem.put_filec                 K   s$   |  |}|  |}t|| d S r"   )r   r[   mover_   r   r   r   mv   s   

zLocalFileSystem.mvc                 K   ,   |  |}|  |}tj||fi | d S r"   )r   r   rQ   r   srcdstr   r   r   r   rQ         

zLocalFileSystem.linkc                 K   rn   r"   )r   r   symlinkro   r   r   r   rs      rr   zLocalFileSystem.symlinkreturnc                 C   s   t j| |S r"   )r   r    r8   r   r$   r   r   r   r8         zLocalFileSystem.islinkc                 C   s   t | | d S r"   )r   remover   r$   r   r   r   rm_file   rk   zLocalFileSystem.rm_filec                 C   sn   t |ts|g}|D ]*}| |}| |r/|stdt|t kr)tdt	
| q
t| q
d S )Nz+Cannot delete directory, set recursive=Truez'Cannot delete current working directory)rA   listr   r]   
ValueErrorrV   abspathr   getcwdr[   rmtreerv   )r   r    	recursivemaxdepthpr   r   r   rm   s   


zLocalFileSystem.rmc                 C   s   |  |}d| S )Nfile://)r   )r   r'   r   r   r   unstrip_protocol   s   

z LocalFileSystem.unstrip_protocolre   c                 K   sB   |  |}| jrd|v r| j| |dd t||fd| i|S )NwTr   fs)r   r   r   rY   LocalFileOpener)r   r    r9   
block_sizer   r   r   r   _open   s   
zLocalFileSystem._openc                 K   sb   |  |}| jr| j| |dd | |rt|d  nt|d  |r/t	|d d S d S )NTr   ar   )
r   r   r   rY   r   r   utimerf   closetruncate)r   r    r   r   r   r   r   touch   s   

zLocalFileSystem.touchc                 C   $   | j |d}tjj|d tjjdS )Nr    r7   tzr+   datetimefromtimestamptimezoneutcr   r    r+   r   r   r   r7      s   zLocalFileSystem.createdc                 C   r   )Nr   r<   r   r   r   r   r   r   modified   s   zLocalFileSystem.modifiedc                 C   sd   |  |}tjdkr|ddd pdS |ddd }t|dkr0|dd dkr0|d d S |S )Nr      r         ::/)r   r   seprsplitlen)clsr    path_r   r   r   rY      s   

zLocalFileSystem._parentc                 C   s4  t |}|dr|dd  }n#|dr|dd  }n|dr(|dd  }n|dr3|dd  }t|}tjd	kr|d
d dkrR|d d |dd  }}n4|d d dkr|d	d }dksp|d	|d
  }dkrv|d}}n|d | ||d  }}nd}|d	p| j}|| S |d	p| jS )Nr      zfile:   zlocal://   zlocal:   r   r   r   r   // )r   
startswithmake_path_posixr   r   findrstriproot_marker)r   r    driveindex1index2r   r   r   r      s0   




zLocalFileSystem._strip_protocolc                 C   r   NTr   r   r   r   r   _isfilestore  s   zLocalFileSystem._isfilestorec                 C   s   t |}t||S r"   )r   r   chmod)r   r    r9   r   r   r   r     s   zLocalFileSystem.chmod)F)Tr"   )FN)re   N)'__name__
__module____qualname____doc__r   protocol
local_filer   propertyr   r   r   r#   r3   r+   rW   rb   rZ   r]   ri   rj   rm   rQ   rs   boolr8   rw   r   r   r   r   r7   r   classmethodrY   r   r   r   __classcell__r   r   r   r   r      sH    



	
/






&r   c                 C   s  t | ts)t | tttfrt| dd | D S t| } t | ts)td| dtj	dkrZ| 
dr5| S | 
dr?t| S | 
drK| dd	 } n| d
krQd} t  d|  S | dd dkrp| dd dkrp| dd	 } | dd dkrt| dkr| d d S | dd} | S | dd dkrtt| S | 
drd| dd	 dd S | 
dr| dd} tt d  |  S | dd} | 
dr| dd	 } n| d
krd} tt  d|  S )z-Make path generic and absolute for current OSc                 s   s    | ]}t |V  qd S r"   )r   )r(   r   r   r   r   	<genexpr>%  s    z"make_path_posix.<locals>.<genexpr>zcould not convert z
 to stringr   ~z./r   N.r   r   r   r   r   r   \)z\\r   r   )r   r   )rA   strrx   settupler%   r   	TypeErrorr   r   r   rV   
expanduserr{   r   replacer   
splitdriver   r   r   r   r   !  sH   






 


r   c                 C   s"   |  tjptjduo|  tjS )zReturn True if the path ends with a path separator.

    A forward slash is always considered a path separator, even on Operating
    Systems that normally use a backslash.
    N)endswithr   r   altsepr   r   r   r   trailing_sepT  s   "	r   c                   @   s   e Zd Z	d6ddZdd Zdd Zd	d
 Zdd Zdd Zdd Z	de
fddZde
fddZdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd7defd&d'Zed(d) Zd*d+ Zd8d,d-Zd.d/ Zd0d1 Zd2d3 Zd4d5 ZdS )9r   TNc                 K   sJ   t d| || _|| _|| _d | _|| _t||| _t	j
| _|   d S )Nzopen file: %s)loggerdebugr    r9   r   r2   
autocommitr   compressionioDEFAULT_BUFFER_SIZE	blocksizer   )r   r    r9   r   r   r   r   r   r   r   r   a  s   zLocalFileOpener.__init__c                 C   s   | j d u s	| j jr\| jsd| jvr,t| j| jd| _ | jr+t| j }|| j | jd| _ nt	 \}}t
| || _t|| jd| _ d| jvr^| j dd| _| j d | j| j _d S d S d S )Nr   r9   r   r   )r2   closedr   r9   rf   r    r   r   tempfilemkstempr   r   tempseekr6   )r   compressr)   r'   r   r   r   r   n  s"   


zLocalFileOpener._openc                 C   s2   d| j vrt|   | j| | j|| S Nr)r9   ry   r   r2   r   read)r   startendr   r   r   _fetch_range  s
   
zLocalFileOpener._fetch_rangec                 C   sL   d | _ |dd }| j| d|d v r$d | _ |   | j | d S d S )Nlocr   r9   )r2   pop__dict__updater   r   )r   stater   r   r   r   __setstate__  s   zLocalFileOpener.__setstate__c                 C   sD   | j  }|d d| jv r| j |d< |S | jjs td|S )Nr2   r   r   z+Cannot serialise open write-mode local file)r   copyr   r9   r2   tellr   ry   )r   dr   r   r   __getstate__  s   


zLocalFileOpener.__getstate__c                 C   s"   | j rtdt| j| j d S )Nz0Can only commit if not already set to autocommit)r   RuntimeErrorr[   rl   r   r    r   r   r   r   commit  s   zLocalFileOpener.commitc                 C   s   | j rtdt| j d S )Nz#Cannot discard if set to autocommit)r   r   r   rv   r   r   r   r   r   discard  s   zLocalFileOpener.discardrt   c                 C   r   r   r   r   r   r   r   readable  s   zLocalFileOpener.readablec                 C   s
   d| j vS r   r   r   r   r   r   writable  rX   zLocalFileOpener.writablec                 O      | j j|i |S r"   )r2   r   r   argsr   r   r   r   r     ru   zLocalFileOpener.readc                 O   r   r"   )r2   writer   r   r   r   r     ru   zLocalFileOpener.writec                 O   r   r"   )r2   r   r   r   r   r   r     ru   zLocalFileOpener.tellc                 O   r   r"   )r2   r   r   r   r   r   r     ru   zLocalFileOpener.seekc                 O   r   r"   )r2   seekabler   r   r   r   r     ru   zLocalFileOpener.seekablec                 O   r   r"   )r2   readliner   r   r   r   r     ru   zLocalFileOpener.readlinec                 O   r   r"   )r2   	readlinesr   r   r   r   r     ru   zLocalFileOpener.readlinesc                 C   
   | j  S r"   )r2   r   r   r   r   r   r     rX   zLocalFileOpener.closec                 C   s   | j |S r"   )r2   r   )r   r6   r   r   r   r        zLocalFileOpener.truncatec                 C   s   | j jS r"   )r2   r   r   r   r   r   r     s   zLocalFileOpener.closedc                 C   r   r"   )rawfilenor   r   r   r   r     rX   zLocalFileOpener.filenoc                 C   s   | j   d S r"   )r2   flushr   r   r   r   r     s   zLocalFileOpener.flushc                 C   r   r"   )r2   __iter__r   r   r   r   r     rX   zLocalFileOpener.__iter__c                 C   s   t | j|S r"   )rN   r2   )r   itemr   r   r   __getattr__  r   zLocalFileOpener.__getattr__c                 C   s
   d| _ | S r   )
_incontextr   r   r   r   	__enter__  s   zLocalFileOpener.__enter__c                 C   s   d| _ | j||| d S )NF)r   r2   __exit__)r   exc_type	exc_value	tracebackr   r   r   r     s   zLocalFileOpener.__exit__)TNNr"   )rt   N)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   intr   r   r   r   r   r   r   r   r   r   r   r   r   r   `  s8    
	


r   )r   r   loggingr   os.pathr    rV   r[   rC   r   fsspecr   fsspec.compressionr   fsspec.corer   fsspec.utilsr   r   	getLoggerr   r   r   r   IOBaser   r   r   r   r   <module>   s&    
  3