← Back
Editing: dirdbm.cpython-38.pyc
U ��W[�' � @ s� d Z ddlZddlZddlZz ddlZW n ek rD ddlZY nX ddlmZ ze W n e k rr e Z Y nX G dd� d�Z G dd� de �Z d dd �Z d ddgZdS ) 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 Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd-d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd 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}|dd� }t j� |�r�t � |� qjt � ||� qjdS )zb @type name: str @param name: Base path to use for the directory storage. z*.newz*.rplN���) �os�path�abspathZdnamer � _dnamePath�isdirZcreateDirectory�glob�child�remove�exists�rename)�self�name�fZ replacements�old� r �:/usr/lib/python3/dist-packages/twisted/persisted/dirdbm.py�__init__2 s zDirDBM.__init__c C s t �|��dd��dd�S )z? Encode a key so it can be used as a filename. � � _� /� -)�base64Z encodestring�replace�r �kr r r �_encodeO s zDirDBM._encodec C s t �|�dd��dd��S )z3 Decode a filename to get the key. r r r r )r Z decodestringr r r r r �_decodeW s zDirDBM._decodec C s$ t |jd��}|�� }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 � _readFile^ s zDirDBM._readFilec C s. t |jd��}|�|� |�� W 5 Q R X dS )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 � _writeFilei s zDirDBM._writeFilec C s t | j�� �S )zF @return: The number of key/value pairs in this Shelf )�lenr �listdir�r r r r �__len__t s zDirDBM.__len__c C s� t |�tkstd��t |�tks(td��| �|�}| j�|�}|�� rR|�d�}n |�d�}z| �||� W n |� � � Y n X |�� r�|� � |� |� dS )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 byteszDirDBM value must be bytesz.rplz.newN) �type�bytes� TypeErrorr r r r ZsiblingExtensionr) r ZmoveTo)r r �vr �newr r r � __setitem__{ s" zDirDBM.__setitem__c C sT t |�tkstd��| j�| �|��}z | �|�W S tk rN t|��Y nX dS )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 r. N) r/ r0 r1 r r r r$ �EnvironmentError�KeyError)r r r r r r � __getitem__� s zDirDBM.__getitem__c C sZ t |�tkstd��| �|�}z| j�|��� W n" tk rT t| � |���Y nX dS )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 r. N) r/ r0 r1 r r r r r5 r6 r r r r r � __delitem__� s zDirDBM.__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 Zvalsr; �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; r> r r r r@ � s z DirDBM.itemsc C s. t |�tkstd��| �|�}| 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. r. )r/ r0 r1 r r r �isfile�r r> r r r �has_key� s zDirDBM.has_keyc C s || kr|| |< |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 r> �valuer r r � setdefault� s zDirDBM.setdefaultNc C s || kr| | S |S dS )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 r> �defaultr r r �get� s z DirDBM.getc C s | � |�S )z) @see: L{DirDBM.has_key} )rC rB r r r � __contains__ s zDirDBM.__contains__c C s |� � D ]\}}|| |< qdS )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)r@ )r �dictr> �valr r r �update s z DirDBM.updatec C sH t |�}|| jkst�| �|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 ] }| |= qdS )z< Delete all key/value pairs in this dirdbm. N)r; r r r r rN 5 s z DirDBM.clearc C s dS )zL Close this dbm: no-op, for dbm-style interface compliance. Nr r, r r r �close= s z DirDBM.closec C sB t |�tkstd��| j�| �|��}|�� r6|�� S t|��dS )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 r. N) r/ r0 r1 r r r rA �getModificationTimer6 )r r> r r r r rR C s zDirDBM.getModificationTime)N)�__name__� __module__� __qualname__�__doc__r r r r$ r) r- r4 r7 r8 r; r? r@ rC rE rG rH rK rP rN rQ rR r r r r r * s, " r c @ s e Zd ZdZdd� Zdd� ZdS )�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�| ||� dS )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 r4 )r r r2 r r r r4 \ s zShelf.__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 )rX �loadsr r7 r r r r r7 j s zShelf.__getitem__N)rS rT rU rV r4 r7 r r r r rW T s rW 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 �openy s r^ )NN)rV r r r ZcPicklerX � ImportErrorZtwisted.python.filepathr r! � NameErrorr^ r rW �__all__r r r r �<module> s$ ,%
Save File
Cancel