
\[\+)                 @   sq   d  Z  d d l Z d d l Z d d l Z d d g Z d Z e Z	 Gd d   d e j
  Z d d d	 d  Z d S)
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  Z d  Z e Z e Z d d   Z d d   Z d d   Z e 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 e Z d d   Z d d    Z e Z d! d"   Z d# d$   Z d% d&   Z d' S)(	_Databasec             C   s   | |  _  | d |  _ | d |  _ | d |  _ d  |  _ y t j |  j d d d } WnF t k
 r t j |  j d d d  } |  j |  j  Wd  QXYn X| j	   |  j
   d  S)Nz.dirz.datz.bakrencodingzLatin-1w)_mode_dirfile_datfile_bakfile_index_ior   OSError_chmodclose_update)selfZfilebasenamemodef r   /usr/lib/python3.4/dbm/dumb.py__init__/   s    		
z_Database.__init__c             C   s   i  |  _  y t j |  j d d d } Wn t k
 r< YnY X| P xH | D]@ } | j   } t |  \ } } | j d  } | |  j  | <qK WWd  QXd  S)Nr   r   zLatin-1)r   r   r   r	   r   rstripevalencode)r   r   linekeypos_and_siz_pairr   r   r   r   N   s    	z_Database._updatec             C   s   |  j  d  k r d  Sy |  j j |  j  Wn t k
 r> Yn Xy |  j j |  j |  j  Wn t k
 rp Yn X|  j j |  j d d d ] } |  j	 |  j  xC |  j  j
   D]2 \ } } d | j d  | f } | j |  q WWd  QXd  S)Nr   r   zLatin-1z%r, %r
)r   _osunlinkr   r   renamer	   r   r   r   itemsdecodewrite)r   r   r   r   entryr   r   r   _commit_   s    !z_Database._commitc             C   s"   |  j  d  k r t d   n  d  S)Nz"DBM object has already been closed)r   r   )r   r   r   r   _verify_openz   s    z_Database._verify_openc          	   C   s|   t  | t  r! | j d  } n  |  j   |  j | \ } } t j |  j d  # } | j |  | j	 |  } Wd  QX| S)Nzutf-8rb)

isinstancestrr   r&   r   r   r   r
   seekread)r   r   possizr   Zdatr   r   r   __getitem__~   s    
z_Database.__getitem__c             C   s   t  j |  j d  g } | j d d  t | j    } | t d t t } | j d | |  | } | j |  Wd  QX| t |  f S)Nzrb+r         s    )	r   r   r
   r*   inttell
_BLOCKSIZEr#   len)r   valr   r,   Znposr   r   r   _addval   s    z_Database._addvalc          	   C   sH   t  j |  j d  ! } | j |  | j |  Wd  QX| t |  f S)Nzrb+)r   r   r
   r*   r#   r4   )r   r,   r5   r   r   r   r   _setval   s    z_Database._setvalc             C   se   | |  j  | <t j |  j d d d 7 } |  j |  j  | j d | j d  | f  Wd  QXd  S)Nar   zLatin-1z%r, %r
)r   r   r   r	   r   r#   r"   )r   r   r   r   r   r   r   _addkey   s    z_Database._addkeyc             C   s>  t  | t  r! | j d  } n$ t  | t t f  sE t d   n  t  | t  rf | j d  } n$ t  | t t f  s t d   n  |  j   | |  j k r |  j | |  j	 |   n{ |  j | \ } } | t
 d t
 } t |  t
 d t
 } | | k r$|  j | |  |  j | <n |  j	 |  |  j | <d  S)Nzutf-8zkeys must be bytes or stringszvalues must be bytes or stringsr0   )r(   r)   r   bytes	bytearray	TypeErrorr&   r   r9   r6   r3   r4   r7   )r   r   r5   r,   r-   Z	oldblocksZ	newblocksr   r   r   __setitem__   s"    
z_Database.__setitem__c             C   sC   t  | t  r! | j d  } n  |  j   |  j | =|  j   d  S)Nzutf-8)r(   r)   r   r&   r   r%   )r   r   r   r   r   __delitem__   s
    

z_Database.__delitem__c             C   s9   y t  |  j  SWn! t k
 r4 t d  d   Yn Xd  S)Nz"DBM object has already been closed)listr   r<   r   )r   r   r   r   keys   s    z_Database.keysc                s*     j      f d d     j j   D S)Nc                s    g  |  ] } |   | f  q Sr   r   ).0r   )r   r   r   
<listcomp>   s   	 z#_Database.items.<locals>.<listcomp>)r&   r   r@   )r   r   )r   r   r!      s    
z_Database.itemsc             C   so   t  | t  r! | j d  } n  y | |  j k SWn6 t k
 rj |  j d  k rc t d  d   n   Yn Xd  S)Nzutf-8z"DBM object has already been closed)r(   r)   r   r   r<   r   )r   r   r   r   r   __contains__   s    z_Database.__contains__c             C   s9   y t  |  j  SWn! t k
 r4 t d  d   Yn Xd  S)Nz"DBM object has already been closed)iterr   r<   r   )r   r   r   r   iterkeys   s    z_Database.iterkeysc             C   s9   y t  |  j  SWn! t k
 r4 t d  d   Yn Xd  S)Nz"DBM object has already been closed)r4   r   r<   r   )r   r   r   r   __len__   s    z_Database.__len__c             C   s,   |  j    d  |  _ |  _ |  _ |  _ d  S)N)r%   r   r
   r	   r   )r   r   r   r   r      s    
z_Database.closec             C   s/   t  |  j d  r+ |  j j | |  j  n  d  S)Nchmod)hasattrr   rG   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)__name__
__module____qualname__r   r   r   r   r%   syncr&   r.   r6   r7   r9   r=   r>   r@   r!   rC   rE   __iter__rF   r   __del__r   rJ   rL   r   r   r   r   r   "   s0   
	"r   i  c             C   sM   y  t  j d  } t  j |  Wn t k
 r4 Yn X| | @} t |  |  S)a  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, is ignored in the dbm.dumb module; the
    database is 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   umaskAttributeErrorr   )rI   Zflagr   Zumr   r   r   r   
  s    )__doc__ior   osr   collections__all__r3   r   r   MutableMappingr   r   r   r   r   r   <module>   s   