3 \�. � @ sP d dl Zd dlZd dlZd dlZddgZdZe Z G dd� dej �Z d d d�Z dS ) � 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 krXdd l}|jdtdd� t| ||d �S ) Nr r r r r z)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 )Zastr' �ior �osr � collections�__all__rA r r �MutableMappingr r r r r r �<module> s