a
XC?h- � @ sR d dl Zd dlZd dlZd dlZddgZdZ e
Z G dd� dej j
�Zd
d d�ZdS ) � 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 sL || _ |dk| _|d | _|d | _|d | _d | _| �|� | �|� d S )N�rz.dirz.datz.bak)�_mode� _readonly�_dirfile�_datfile�_bakfile�_index�_create�_update)�selfZ filebasename�mode�flag� r �-/opt/alt/python39/lib64/python3.9/dbm/dumb.py�__init__0 s
z_Database.__init__c C s� |dkr@| j | j| jfD ]&}zt�|� W q ty< Y q0 qztj| j ddd�}W nZ ty� |dvrn� tj| j ddd�� }| �| j � W d � n1 s�0 Y Y n
0 |� � d S )N�nr �Latin-1��encoding�r r �w)
r
r r �_os�remove�OSError�_ior �_chmod�close)r r �filename�fr r r r
I s 0z_Database._createc C s� d| _ i | _ztj| jddd�}W n" tyD |dvr:� d| _ Y nZ0 |�D |D ].}|�� }t�|�\}}|� d�}|| j|< qPW d � n1 s�0 Y d S )NFr r r r T)
� _modifiedr r r r r �rstrip�_astZ literal_eval�encode)r r r" �line�key�pos_and_siz_pairr r r r \ s
z_Database._updatec C s� | j d u s| jsd S z| j�| j� W n ty8 Y n0 z| j�| j| j� W n tyb Y n0 | jj | jddd��L}| �
| j� | j � � D ]$\}}d|� d�|f }|�
|� q�W d � n1 s�0 Y d S )Nr r r �%r, %r
)r r# r �unlinkr r �renamer r r r �items�decode�write)r r" r( r) �entryr r r �_commitp s z_Database._commitc C s | j d u rtd��d S �N�"DBM object has already been closed)r r �r r r r � _verify_open� s
z_Database._verify_openc C sp t |t�r|�d�}| �� | j| \}}t�| jd��$}|�|� |� |�}W d � n1 sb0 Y |S )N�utf-8�rb)
�
isinstance�strr& r5 r r r r
�seek�read)r r( �pos�sizr" Zdatr r r � __getitem__� s
(z_Database.__getitem__c C s� t �| jd��\}|�dd� t|�� �}|t d t t }|�d|| � |}|�|� W d � n1 sp0 Y |t|�fS )N�rb+r � � � ) r r r
r: �int�tell�
_BLOCKSIZEr/ �len)r �valr" r< Znposr r r �_addval� s (z_Database._addvalc C sN t �| jd��$}|�|� |�|� W d � n1 s80 Y |t|�fS )Nr? )r r r
r: r/ rF )r r< rG r" r r r �_setval� s
(z_Database._setvalc C sd || j |< tj| jddd��4}| �| j� |�d|�d�|f � W d � n1 sV0 Y d S )N�ar r r* )r r r r r r/ r. )r r( r) r" r r r �_addkey� s
z_Database._addkeyc C s� | j rtd��t|t�r$|�d�}nt|ttf�s:td��t|t�rP|�d�}nt|ttf�sftd��| �� d| _ || j
vr�| � || � |�� n^| j
| \}}|t
d t
}t|�t
d t
}||kr�| �||�| j
|< n| � |�| j
|< d S )N�'The database is opened for reading onlyr6 z keys must be bytes or stringsz values must be bytes or stringsTrA )r r r8 r9 r& �bytes� bytearray� TypeErrorr5 r# r rK rH rE rF rI )r r( rG r< r= Z oldblocksZ newblocksr r r � __setitem__� s(
z_Database.__setitem__c C sD | j rtd��t|t�r"|�d�}| �� d| _| j|= | �� d S )NrL r6 T) r r r8 r9 r& r5 r# r r1 �r r( r r r � __delitem__� s
z_Database.__delitem__c C s. z t | j�W S ty( td�d �Y n0 d S r2 )�listr rO r r4 r r r �keys� s z_Database.keysc s � � � � fdd�� j�� D �S )Nc s g | ]}|� | f�qS r r )�.0r( r4 r r �
� � z#_Database.items..)r5 r rT r4 r r4 r r- � s z_Database.itemsc C sP t |t�r|�d�}z || jv W S tyJ | jd u rDtd�d �n� Y n0 d S )Nr6 r3 )r8 r9 r& r rO r rQ r r r � __contains__� s
z_Database.__contains__c C s. z t | j�W S ty( td�d �Y n0 d S r2 )�iterr rO r r4 r r r �iterkeys s z_Database.iterkeysc C s. z t | j�W S ty( td�d �Y n0 d S r2 )rF r rO r r4 r r r �__len__
s z_Database.__len__c C sD z$| � � W d | _ | _ | _| _nd | _ | _ | _| _0 d S �N)r1 r r
r r r4 r r r r s
z_Database.closec C s | j �|| j� d S r\ )r �chmodr )r �filer r r r s z_Database._chmodc C s | S r\ r r4 r r r � __enter__ s z_Database.__enter__c G s | � � d S r\ )r )r �argsr r r �__exit__ s z_Database.__exit__N)r ) �__name__�
__module__� __qualname__r r r r
r r1 �syncr5 r> rH rI rK rP rR rT r- rX rZ �__iter__r[ r �__del__r r_ ra r r r r r # s2
% r r � c C sT zt �d�}t �|� W n ty* Y n 0 || @ }|dvrFtd��t| ||d�S )Nr )r r r r z)Flag must be one of 'r', 'w', 'c', or 'n')r )r �umask�AttributeError�
ValueErrorr )r^ r r Zumr r r r " s
)r rh )�astr% �ior �osr �collections.abc� collections�__all__rE r r �abc�MutableMappingr r r r r r � s