Edit File: dumb.cpython-36.pyc
3 \�. � @ sT d Z ddlZddlZddlZddlZddgZdZ e Z G dd� dej �Z d d d�ZdS ) a� A dumb and slow but simple dbm clone. For database spam, spam.dir contains the index (a text file), spam.bak *may* contain a backup of the index (also a text file), while spam.dat contains the data (a binary file). XXX TO DO: - seems to contain a bug when updating... - reclaim free space (currently, space once occupied by deleted or expanded items is never reused) - support concurrent access (currently, if two processes take turns making updates, they can mess up the index) - support efficient access to large databases (currently, the whole index is read when the database is opened, and some updates rewrite the whole index) - support opening for read-only (flag = 'm') � N�error�openi c @ s� e Zd ZeZeZd+dd�Zdd� Zdd� Zdd � ZeZ d d � Z d d � Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � ZeZd d!� Zd"d#� ZeZd$d%� Zd&d'� Zd(d)� Zd*S ),� _Database�cc C sJ || _ |dk| _|d | _|d | _|d | _d | _| j|� | j� d S )N�rz.dirz.datz.bak)�_mode� _readonly�_dirfile�_datfile�_bakfile�_index�_create�_update)�selfZ filebasename�mode�flag� r � /usr/lib64/python3.6/dbm/dumb.py�__init__0 s z_Database.__init__c C s� |dkrFx<| j | j| jfD ](}ytj|� W q tk r@ Y qX qW ytj| j ddd�}W n^ tk r� |d kr�dd l}|j dt dd � tj| j d dd��}| j | j � W d Q R X Y n X |j � d S ) N�nr zLatin-1)�encodingr r zIThe database file is missing, the semantics of the 'c' flag will be used.� )� stacklevel�w)r r ) r r r �_os�remove�OSError�_ior �warnings�warn�DeprecationWarning�_chmod�close)r r �filename�fr r r r r I s z_Database._createc C s� i | _ ytj| jddd�}W n tk r: | j | _Y nPX d| _|�> x6|D ].}|j� }tj |�\}}|j d�}|| j |< qNW W d Q R X d S )Nr zLatin-1)r F) r r r r r r � _modified�rstrip�_astZ literal_eval�encode)r r$ �line�key�pos_and_siz_pairr r r r _ s z_Database._updatec C s� | j d ks| j rd S y| jj| j� W n tk r< Y nX y| jj| j| j� W n tk rh Y nX | jj | jddd��F}| j | j� x2| j j � D ]$\}}d|j d�|f }|j |� q�W W d Q R X d S )Nr zLatin-1)r z%r, %r )r r% r �unlinkr r �renamer r r r! �items�decode�write)r r$ r* r+ �entryr r r �_commitq s z_Database._commitc C s | j d krtd��d S )Nz"DBM object has already been closed)r r )r r r r � _verify_open� s z_Database._verify_openc C s\ t |t�r|jd�}| j� | j| \}}tj| jd��}|j|� |j |�}W d Q R X |S )Nzutf-8�rb) � isinstance�strr( r3 r r r r �seek�read)r r* �pos�sizr$ Zdatr r r � __getitem__� s z_Database.__getitem__c C sr t j| jd��R}|jdd� t|j� �}|t d t t }|jd|| � |}|j|� W d Q R X |t|�fS )Nzrb+r � � � ) r r r r7 �int�tell� _BLOCKSIZEr0 �len)r �valr$ r9 Znposr r r �_addval� s z_Database._addvalc C s: t j| jd��}|j|� |j|� W d Q R X |t|�fS )Nzrb+)r r r r7 r0 rB )r r9 rC r$ r r r �_setval� s z_Database._setvalc C sP || j |< tj| jddd��*}| j| j� |jd|jd�|f � W d Q R X d S )N�azLatin-1)r z%r, %r )r r r r r! r0 r/ )r r* r+ r$ r r r �_addkey� s z_Database._addkeyc C s | j r dd l}|jdtdd� t|t�r4|jd�}nt|ttf�sJt d��t|t�r`|jd�}nt|ttf�svt d��| j � d| _ || j kr�| j || j|�� n^| j | \}}|t d t }t|�t d t }||kr�| j||�| j |< n| j|�| j |< d S ) Nr z'The database is opened for reading onlyr< )r zutf-8z keys must be bytes or stringsz values must be bytes or stringsTr= )r r r r r5 r6 r( �bytes� bytearray� TypeErrorr3 r% r rG rD rA rB rE )r r* rC r r9 r: Z oldblocksZ newblocksr r r � __setitem__� s, z_Database.__setitem__c C sT | j r dd l}|jdtdd� t|t�r2|jd�}| j� d| _| j |= | j � d S )Nr z'The database is opened for reading onlyr< )r zutf-8T) r r r r r5 r6 r( r3 r% r r2 )r r* r r r r � __delitem__� s z_Database.__delitem__c C s. y t | j�S tk r( td�d �Y nX d S )Nz"DBM object has already been closed)�listr rJ r )r r r r �keys� s z_Database.keysc s � j � � fdd�� jj� D �S )Nc s g | ]}|� | f�qS r r )�.0r* )r r r � <listcomp>� s z#_Database.items.<locals>.<listcomp>)r3 r rN )r r )r r r. � s z_Database.itemsc C sP t |t�r|jd�}y || jkS tk rJ | jd krDtd�d �n� Y nX d S )Nzutf-8z"DBM object has already been closed)r5 r6 r( r rJ r )r r* r r r � __contains__� s z_Database.__contains__c C s. y t | j�S tk r( td�d �Y nX d S )Nz"DBM object has already been closed)�iterr rJ r )r r r r �iterkeys s z_Database.iterkeysc C s. y t | j�S tk r( td�d �Y nX d S )Nz"DBM object has already been closed)rB r rJ r )r r r r �__len__ s z_Database.__len__c C s, z | j � W d d | _ | _ | _| _X d S )N)r2 r r r r )r r r r r" s z_Database.closec C s t | jd�r | jj|| j� d S )N�chmod)�hasattrr rU r )r �filer r r r! s z_Database._chmodc C s | S )Nr )r r r r � __enter__! s z_Database.__enter__c G s | j � d S )N)r" )r �argsr r r �__exit__$ s z_Database.__exit__N)r ) �__name__� __module__� __qualname__r r r r r r2 �syncr3 r; rD rE rG rK rL rN r. rQ rS �__iter__rT r" �__del__r! rX rZ r r r r r # s2 ' r r � c C sf yt jd�}t j|� W n tk r, Y n X || @ }|d krXddl}|jdtdd � t| ||d �S ) aE Open the database file, filename, and return corresponding object. The flag argument, used to control how the database is opened in the other DBM implementations, supports only the semantics of 'c' and 'n' values. Other values will default to the semantics of 'c' value: the database will always opened for update and will be created if it does not exist. The optional mode argument is the UNIX mode of the file, used only when the database has to be created. It defaults to octal code 0o666 (and will be modified by the prevailing umask). r r r r r Nz)Flag must be one of 'r', 'w', 'c', or 'n'r< )r )r )r r r r )r �umask�AttributeErrorr r r r )rW r r Zumr r r r r ( s )r ra )�__doc__Zastr' �ior �osr � collections�__all__rA r r �MutableMappingr r r r r r �<module> s