3 \C!ã@s†dZddlmZmZddlmZddlZddddgZGd d „d ejƒZ Gd d„dejƒZ Gd d„de ƒZ Gd d„de ƒZ ddd„Z dS)a˜ Manage shelves of pickled objects. A "shelf" is a persistent, dictionary-like object. The difference with dbm databases is that the values (not the keys!) in a shelf can be essentially arbitrary Python objects -- anything that the "pickle" module can handle. This includes most class instances, recursive data types, and objects containing lots of shared sub-objects. The keys are ordinary strings. To summarize the interface (key is a string, data is an arbitrary object): import shelve d = shelve.open(filename) # open, with (g)dbm filename -- no suffix d[key] = data # store data at key (overwrites old data if # using an existing key) data = d[key] # retrieve a COPY of the data at key (raise # KeyError if no such key) -- NOTE that this # access returns a *copy* of the entry! del d[key] # delete data stored at key (raises KeyError # if no such key) flag = key in d # true if the key exists list = d.keys() # a list of all existing keys (slow!) d.close() # close it Dependent on the implementation, closing a persistent dictionary may or may not be necessary to flush changes to disk. Normally, d[key] returns a COPY of the entry. This needs care when mutable entries are mutated: for example, if d[key] is a list, d[key].append(anitem) does NOT modify the entry d[key] itself, as stored in the persistent mapping -- it only modifies the copy, which is then immediately discarded, so that the append has NO effect whatsoever. To append an item to d[key] in a way that will affect the persistent mapping, use: data = d[key] data.append(anitem) d[key] = data To avoid the problem with mutable entries, you may pass the keyword argument writeback=True in the call to shelve.open. When you use: d = shelve.open(filename, writeback=True) then d keeps a cache of all entries you access, and writes them all back to the persistent mapping when you call d.close(). This ensures that such usage as d[key].append(anitem) works as intended. However, using keyword argument writeback=True may consume vast amount of memory for the cache, and it may make d.close() very slow, if you access many of d's entries after opening it in this way: d has no way to check which of the entries you access are mutable and/or which ones you actually mutate, so it must cache, and write back at close, all of the entries that you access. You can call d.sync() to write back all the entries in the cache, and empty the cache (d.sync() also synchronizes the persistent dictionary on disk, if feasible). é)ÚPicklerÚ Unpickler)ÚBytesIONÚShelfÚ BsdDbShelfÚDbfilenameShelfÚopenc@s8eZdZdZdd„ZeZZZZZ Z dd„Z dS)Ú _ClosedDictz>Marker for a closed dict. Access attempts raise a ValueError.cGs tdƒ‚dS)Nz!invalid operation on closed shelf)Ú ValueError)ÚselfÚargs©r ú/usr/lib64/python3.6/shelve.pyÚclosedEsz_ClosedDict.closedcCsdS)Nzr )r r r rÚ__repr__Isz_ClosedDict.__repr__N) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚ__iter__Ú__len__Ú __getitem__Ú __setitem__Ú __delitem__Úkeysrr r r rr Bsr c@s|eZdZdZddd„Zdd„Zd d „Zd d „Zd d d„Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„ZdS)!rz¨Base class for shelf implementations. This is initialized with a dictionary-like object. See the module's __doc__ string for an overview of the interface. NFúutf-8cCs.||_|dkrd}||_||_i|_||_dS)Né)ÚdictÚ _protocolÚ writebackÚcacheÚ keyencoding)r rÚprotocolrr!r r rÚ__init__TszShelf.__init__ccs&x |jjƒD]}|j|jƒVq WdS)N)rrÚdecoder!)r Úkr r rr^szShelf.__iter__cCs t|jƒS)N)Úlenr)r r r rrbsz Shelf.__len__cCs|j|jƒ|jkS)N)Úencoder!r)r Úkeyr r rÚ __contains__eszShelf.__contains__cCs|j|jƒ|jkr||S|S)N)r'r!r)r r(Údefaultr r rÚgethsz Shelf.getc CsZy|j|}WnFtk rTt|j|j|jƒƒ}t|ƒjƒ}|jrP||j|<YnX|S)N) r ÚKeyErrorrrr'r!rÚloadr)r r(ÚvalueÚfr r rrms zShelf.__getitem__cCsF|jr||j|<tƒ}t||jƒ}|j|ƒ|jƒ|j|j|j ƒ<dS)N) rr rrrÚdumpÚgetvaluerr'r!)r r(r.r/Úpr r rrws    zShelf.__setitem__c Cs6|j|j|jƒ=y |j|=Wntk r0YnXdS)N)rr'r!r r,)r r(r r rrs  zShelf.__delitem__cCs|S)Nr )r r r rÚ __enter__†szShelf.__enter__cCs |jƒdS)N)Úclose)r Útyper.Ú tracebackr r rÚ__exit__‰szShelf.__exit__cCsf|jdkrdSz0|jƒy|jjƒWntk r:YnXWdy tƒ|_Wnd|_YnXXdS)N)rÚsyncr4ÚAttributeErrorr )r r r rr4Œs   z Shelf.closecCst|dƒsdS|jƒdS)Nr)Úhasattrr4)r r r rÚ__del__s z Shelf.__del__cCsX|jr>|jr>d|_x|jjƒD]\}}|||<qWd|_i|_t|jdƒrT|jjƒdS)NFTr8)rr Úitemsr:rr8)r r(Úentryr r rr8¤s   z Shelf.sync)NFr)N)rrrrr#rrr)r+rrrr3r7r4r;r8r r r rrMs   c@sBeZdZdZddd„Zdd„Zd d „Zd d „Zd d„Zdd„Z dS)ra²Shelf implementation using the "BSD" db interface. This adds methods first(), next(), previous(), last() and set_location() that have no counterpart in [g]dbm databases. The actual database must be opened using one of the "bsddb" modules "open" routines (i.e. bsddb.hashopen, bsddb.btopen or bsddb.rnopen) and passed to the constructor. See the module's __doc__ string for an overview of the interface. NFúutf-8cCstj|||||ƒdS)N)rr#)r rr"rr!r r rr#¼szBsdDbShelf.__init__cCs0|jj|ƒ\}}t|ƒ}|j|jƒt|ƒjƒfS)N)rÚ set_locationrr$r!rr-)r r(r.r/r r rr?ÀszBsdDbShelf.set_locationcCs.t|jƒ\}}t|ƒ}|j|jƒt|ƒjƒfS)N)Únextrrr$r!rr-)r r(r.r/r r rr@ÅszBsdDbShelf.nextcCs.|jjƒ\}}t|ƒ}|j|jƒt|ƒjƒfS)N)rÚpreviousrr$r!rr-)r r(r.r/r r rrAÊszBsdDbShelf.previouscCs.|jjƒ\}}t|ƒ}|j|jƒt|ƒjƒfS)N)rÚfirstrr$r!rr-)r r(r.r/r r rrBÏszBsdDbShelf.firstcCs.|jjƒ\}}t|ƒ}|j|jƒt|ƒjƒfS)N)rÚlastrr$r!rr-)r r(r.r/r r rrCÔszBsdDbShelf.last)NFr>) rrrrr#r?r@rArBrCr r r rr¯s  c@seZdZdZddd„ZdS)rzÇShelf implementation using the "dbm" generic dbm interface. This is initialized with the filename for the dbm database. See the module's __doc__ string for an overview of the interface. ÚcNFcCs$ddl}tj||j||ƒ||ƒdS)Nr)Údbmrr#r)r ÚfilenameÚflagr"rrEr r rr#ászDbfilenameShelf.__init__)rDNF)rrrrr#r r r rrÚsrDFcCst||||ƒS)aóOpen a persistent dictionary for reading and writing. The filename parameter is the base filename for the underlying database. As a side-effect, an extension may be added to the filename and more than one file may be created. The optional flag parameter has the same interpretation as the flag parameter of dbm.open(). The optional protocol parameter specifies the version of the pickle protocol. See the module's __doc__ string for an overview of the interface. )r)rFrGr"rr r rræs )rDNF)rÚpicklerrÚiorÚ collectionsÚ__all__ÚMutableMappingr rrrrr r r rÚ9s   b+