U
     T‚gv'  ã                   @   sˆ   d Z ddlZddlZddlZddlZddlmZ ze W n ek
rP   e	ZY nX G dd„ dƒZ
G dd„ de
ƒZd
dd	„Z	d	ddgZdS )a„  
DBM-style interface to a directory.

Each key is stored as a single file.  This is not expected to be very fast or
efficient, but it's good for easy debugging.

DirDBMs are *not* thread-safe, they should only be accessed by one thread at
a time.

No files should be placed in the working directory of a DirDBM save those
created by the DirDBM itself!

Maintainer: Itamar Shtull-Trauring
é    N)ÚFilePathc                   @   sº   e Zd Z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dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd-dd „Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ ZdS ).ÚDirDBMz®
    A directory with a DBM interface.

    This class presents a hash-like interface to a directory of small,
    flat files. It can only use strings as keys or values.
    c                 C   s¤   t j |¡| _t|ƒ| _| j ¡ s.| j ¡  nrt | j 	d¡j¡D ]}t  
|¡ qBt | j 	d¡j¡}|D ]4}|dd… }t j |¡r’t  
|¡ qjt  ||¡ qjdS )zb
        @type name: str
        @param name: Base path to use for the directory storage.
        z*.newz*.rplNéüÿÿÿ)ÚosÚpathÚabspathZdnamer   Ú
_dnamePathÚisdirZcreateDirectoryÚglobÚchildÚremoveÚexistsÚrename)ÚselfÚnameÚfÚreplacementsÚold© r   ú</tmp/pip-unpacked-wheel-1_7di8ic/twisted/persisted/dirdbm.pyÚ__init__,   s    


zDirDBM.__init__c                 C   s   t  |¡ dd¡ dd¡S )z?
        Encode a key so it can be used as a filename.
        ó   
ó   _ó   /ó   -)Úbase64ÚencodebytesÚreplace©r   Úkr   r   r   Ú_encodeH   s    zDirDBM._encodec                 C   s   t  | dd¡ dd¡¡S )z3
        Decode a filename to get the key.
        r   r   r   r   )r   Údecodebytesr   r   r   r   r   Ú_decodeO   s    zDirDBM._decodec              	   C   s$   t |jdƒ�}| ¡ }W 5 Q R X |S )z�
        Read in the contents of a file.

        Override in subclasses to e.g. provide transparently encrypted dirdbm.
        Úrb)Ú_openr   Úread)r   r   r   Úsr   r   r   Ú	_readFileU   s    zDirDBM._readFilec              	   C   s.   t |jdƒ�}| |¡ | ¡  W 5 Q R X dS )zw
        Write data to a file.

        Override in subclasses to e.g. provide transparently encrypted dirdbm.
        ÚwbN)r$   r   ÚwriteÚflush)r   r   Údatar   r   r   r   Ú
_writeFile_   s    
zDirDBM._writeFilec                 C   s   t | j ¡ ƒS )zF
        @return: The number of key/value pairs in this Shelf
        )Úlenr   Úlistdir©r   r   r   r   Ú__len__i   s    zDirDBM.__len__c                 C   sª   t |ƒtkstdƒ‚t |ƒtks(tdƒ‚|  |¡}| j |¡}| ¡ rR| d¡}n
| d¡}z|  ||¡ W n t	k
rŠ   | 
¡  ‚ Y nX | ¡ rœ| 
¡  | |¡ dS )zÔ
        C{dirdbm[k] = v}
        Create or modify a textfile in this directory

        @type k: bytes
        @param k: key to set

        @type v: bytes
        @param v: value to associate with C{k}
        úDirDBM key must be byteszDirDBM value must be bytesz.rplz.newN)ÚtypeÚbytesÚ	TypeErrorr    r   r   r   ZsiblingExtensionr,   ÚBaseExceptionr   ZmoveTo)r   r   Úvr   Únewr   r   r   Ú__setitem__o   s"    

zDirDBM.__setitem__c                 C   sT   t |ƒtkstdƒ‚| j |  |¡¡}z|  |¡W S  tk
rN   t|ƒ‚Y nX dS )a  
        C{dirdbm[k]}
        Get the contents of a file in this directory as a string.

        @type k: bytes
        @param k: key to lookup

        @return: The value associated with C{k}
        @raise KeyError: Raised when there is no such key
        r1   N)	r2   r3   r4   r   r   r    r'   ÚOSErrorÚKeyError)r   r   r   r   r   r   Ú__getitem__‘   s    zDirDBM.__getitem__c                 C   sZ   t |ƒtkstdƒ‚|  |¡}z| j |¡ ¡  W n" tk
rT   t|  	|¡ƒ‚Y nX dS )zÀ
        C{del dirdbm[foo]}
        Delete a file in this directory.

        @type k: bytes
        @param k: key to delete

        @raise KeyError: Raised when there is no such key
        r1   N)
r2   r3   r4   r    r   r   r   r9   r:   r"   r   r   r   r   Ú__delitem__¤   s    

zDirDBM.__delitem__c                 C   s   t t| j| j ¡  ¡ ƒƒS )z9
        @return: a L{list} of filenames (keys).
        )ÚlistÚmapr"   r   ZasBytesModer.   r/   r   r   r   Úkeys¶   s    zDirDBM.keysc                 C   s(   g }|   ¡ }|D ]}| | | ¡ q|S )z?
        @return: a L{list} of file-contents (values).
        ©r?   Úappend)r   Úvalsr?   Úkeyr   r   r   Úvalues¼   s
    zDirDBM.valuesc                 C   s,   g }|   ¡ }|D ]}| || | f¡ q|S )zL
        @return: a L{list} of 2-tuples containing key/value pairs.
        r@   )r   Úitemsr?   rC   r   r   r   rE   Æ   s
    zDirDBM.itemsc                 C   s.   t |ƒtkstdƒ‚|  |¡}| j |¡ ¡ S )z¬
        @type key: bytes
        @param key: The key to test

        @return: A true value if this dirdbm has the specified key, a false
        value otherwise.
        r1   )r2   r3   r4   r    r   r   Úisfile©r   rC   r   r   r   Úhas_keyÐ   s    
zDirDBM.has_keyc                 C   s   || kr|| |< |S | | S )z¶
        @type key: bytes
        @param key: The key to lookup

        @param value: The value to associate with key if key is not already
        associated with a value.
        r   )r   rC   Úvaluer   r   r   Ú
setdefaultÝ   s    zDirDBM.setdefaultNc                 C   s   || kr| | S |S dS )zü
        @type key: bytes
        @param key: The key to lookup

        @param default: The value to return if the given key does not exist

        @return: The value associated with C{key} or C{default} if not
        L{DirDBM.has_key(key)}
        Nr   )r   rC   Údefaultr   r   r   Úgetê   s    
z
DirDBM.getc                 C   s
   |   |¡S )z)
        @see: L{DirDBM.has_key}
        )rH   rG   r   r   r   Ú__contains__ù   s    zDirDBM.__contains__c                 C   s   |  ¡ D ]\}}|| |< qdS )zþ
        Add all the key/value pairs in L{dict} to this dirdbm.  Any conflicting
        keys will be overwritten with the values from L{dict}.

        @type dict: mapping
        @param dict: A mapping of key/value pairs to add to this dirdbm.
        N)rE   )r   ÚdictrC   Úvalr   r   r   Úupdateÿ   s    zDirDBM.updatec                 C   sH   t |ƒ}|| jkst‚|  |j¡}| ¡  |  ¡ D ]}| | ||< q2|S )a<  
        Copy the contents of this dirdbm to the dirdbm at C{path}.

        @type path: L{str}
        @param path: The path of the dirdbm to copy to.  If a dirdbm
        exists at the destination path, it is cleared first.

        @rtype: C{DirDBM}
        @return: The dirdbm this dirdbm was copied to.
        )r   r   ÚAssertionErrorÚ	__class__r   Úclearr?   )r   r   Údr   r   r   r   ÚcopyTo
  s    zDirDBM.copyToc                 C   s   |   ¡ D ]
}| |= qdS )z<
        Delete all key/value pairs in this dirdbm.
        N)r?   r   r   r   r   rS     s    zDirDBM.clearc                 C   s   dS )zL
        Close this dbm: no-op, for dbm-style interface compliance.
        Nr   r/   r   r   r   Úclose%  s    zDirDBM.closec                 C   sB   t |ƒtkstdƒ‚| j |  |¡¡}| ¡ r6| ¡ S t|ƒ‚dS )zÁ
        Returns modification time of an entry.

        @return: Last modification date (seconds since epoch) of entry C{key}
        @raise KeyError: Raised when there is no such key
        r1   N)	r2   r3   r4   r   r   r    rF   ÚgetModificationTimer:   )r   rC   r   r   r   r   rW   *  s    zDirDBM.getModificationTime)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r    r"   r'   r,   r0   r8   r;   r<   r?   rD   rE   rH   rJ   rL   rM   rP   rU   rS   rV   rW   r   r   r   r   r   $   s,   

"


r   c                   @   s    e Zd ZdZdd„ Zdd„ ZdS )ÚShelfzÃ
    A directory with a DBM shelf interface.

    This class presents a hash-like interface to a directory of small,
    flat files. Keys must be strings, but values can be any given object.
    c                 C   s   t  |¡}t | ||¡ dS )zÉ
        C{shelf[foo] = bar}
        Create or modify a textfile in this directory.

        @type k: str
        @param k: The key to set

        @param v: The value to associate with C{key}
        N)ÚpickleÚdumpsr   r8   )r   r   r6   r   r   r   r8   B  s    

zShelf.__setitem__c                 C   s   t  t | |¡¡S )a  
        C{dirdbm[foo]}
        Get and unpickle the contents of a file in this directory.

        @type k: bytes
        @param k: The key to lookup

        @return: The value associated with the given key
        @raise KeyError: Raised if the given key does not exist
        )r]   Úloadsr   r;   r   r   r   r   r;   O  s    zShelf.__getitem__N)rX   rY   rZ   r[   r8   r;   r   r   r   r   r\   :  s   r\   c                 C   s   t | ƒS )z£
    This is for 'anydbm' compatibility.

    @param file: The parameter to pass to the DirDBM constructor.

    @param flag: ignored
    @param mode: ignored
    )r   )ÚfileÚflagÚmoder   r   r   Úopen]  s    	rc   )NN)r[   r   r
   r   r]   Ztwisted.python.filepathr   r$   Ú	NameErrorrc   r   r\   Ú__all__r   r   r   r   Ú<module>   s   
  #
