U .Wj1)@sdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZe dddZGdd d ZGd d d eZdd d Zd d d gZdS)aw A L{DirDBM} is a L{dbm}-style interface to a directory. Each key is stored as a single file. This is not expected to be very fast or efficient, but it's good for easy debugging. L{DirDBM}s are *not* thread-safe, they should only be accessed by one thread at a time. No files should be placed in the working directory of a DirDBM save those created by the DirDBM itself! ) annotationsN)AnyStrIterableMappingTypeVaroverload)FilePath_T)boundc@seZdZdZdddddZddddd Zdddd d Zd dd ddZd dddddZddddZ ddddddZ dddddZ dddddZ dddd Z ddd!d"Zd#dd$d%Zdd&d'd(d)Zdddd*d+d,Zeddd'd-d.Zedd/d0d1d2d.ZdFdd4d5d1d6d.Zdd&d'd7d8Zd9dd:d;d<Zddd d=d>Zddd?d@ZdddAdBZddCd'dDdEZd3S)GDirDBMz A directory with a DBM interface. This class presents a hash-like interface to a directory of small, flat files. It can only use strings as keys or values. bytesNone)namereturncCstj||_t||_|js.|jnrt|j djD]}t |qBt|j dj}|D]4}|dd}tj |rt |qjt ||qjdS)zJ @param name: Base path to use for the directory storage. z*.newz*.rplN) ospathabspathZdnamer _dnamePathisdirZcreateDirectoryglobchildremoveexistsrename)selfrf replacementsoldrcc:\users\psant\downloads\hotstarbackend\school\venv38\Lib\site-packages\twisted/persisted/dirdbm.py__init__(s       zDirDBM.__init__)krcCst|ddddS)z? Encode a key so it can be used as a filename.  _/-)base64 encodebytesreplacerr"rrr _encodeCszDirDBM._encodecCst|ddddS)z3 Decode a filename to get the key. r$r#r&r%)r' decodebytesr)r*rrr _decodeJszDirDBM._decodezFilePath[AnyStr])rrc Cs |}|}W5QRX|S)z Read in the contents of a file. Override in subclasses to e.g. provide transparently encrypted dirdbm. )openread)rrrsrrr _readFilePs zDirDBM._readFile)rdatarc Cs,|d}|||W5QRXdS)zw Write data to a file. Override in subclasses to e.g. provide transparently encrypted dirdbm. wN)r.writeflush)rrr2rrrr _writeFileZs  zDirDBM._writeFileint)rcCst|jS)zF @return: The number of key/value pairs in this Shelf )lenrlistdirrrrr __len__dszDirDBM.__len__)r"vrcCst|tkstdt|tks(td||}|j|}|rR|d}n |d}z|||Wnt k r| YnX|r| | |dS)z C{dirdbm[k] = v} Create or modify a textfile in this directory @type k: bytes @param k: key to set @type v: bytes @param v: value to associate with C{k} DirDBM key must be byteszDirDBM value must be bytesz.rplz.newN) typer TypeErrorr+rrrZsiblingExtensionr6 BaseExceptionrZmoveTo)rr"r<rnewrrr __setitem__js"      zDirDBM.__setitem__cCsTt|tkstd|j||}z ||WStk rNt|YnXdS)z C{dirdbm[k]} Get the contents of a file in this directory as a string. @param k: key to lookup @return: The value associated with C{k} @raise KeyError: Raised when there is no such key r=N) r>r r?rrr+r1OSErrorKeyError)rr"rrrr __getitem__s  zDirDBM.__getitem__cCsZt|tkstd||}z|j|Wn"tk rTt| |YnXdS)z C{del dirdbm[foo]} Delete a file in this directory. @type k: bytes @param k: key to delete @raise KeyError: Raised when there is no such key r=N) r>r r?r+rrrrCrDr-r*rrr __delitem__s  zDirDBM.__delitem__zIterable[bytes]cCstt|j|jS)z9 @return: a L{list} of filenames (keys). )listmapr-rZ asBytesModer9r:rrr keyssz DirDBM.keyscCs(g}|}|D]}|||q|S)z? @return: a L{list} of file-contents (values). rIappend)rvalsrIkeyrrr valuess z DirDBM.valueszIterable[tuple[bytes, bytes]]cCs,g}|}|D]}||||fq|S)zL @return: a L{list} of 2-tuples containing key/value pairs. rJ)ritemsrIrMrrr rOs z DirDBM.itemsbool)rMrcCs.t|tkstd||}|j|S)z @type key: bytes @param key: The key to test @return: A true value if this dirdbm has the specified key, a false value otherwise. r=)r>r r?r+rrisfilerrMrrr has_keys  zDirDBM.has_key)rMvaluercCs||kr|||<|S||S)z @type key: bytes @param key: The key to lookup @param value: The value to associate with key if key is not already associated with a value. r)rrMrTrrr setdefaultszDirDBM.setdefaultcCsdSNrrRrrr getsz DirDBM.getr z bytes | _T)rMdefaultrcCsdSrVrrrMrXrrr rWsNz _T | Nonezbytes | _T | NonecCs||kr||S|SdS)z @param key: The key to lookup @param default: The value to return if the given key does not exist @return: The value associated with C{key} or C{default} if not L{DirDBM.has_key(key)} NrrYrrr rWs cCs ||S)z) @see: L{DirDBM.has_key} )rSrRrrr __contains__szDirDBM.__contains__zMapping[bytes, bytes])otherrcCs|D]\}}|||<qdS)z Add all the key/value pairs in L{dict} to this dirdbm. Any conflicting keys will be overwritten with the values from L{dict}. @param dict: A mapping of key/value pairs to add to this dirdbm. N)rO)rr[rMvalrrr updatesz DirDBM.updatecCsHt|}||jkst||j}||D]}||||<q2|S)a! Copy the contents of this dirdbm to the dirdbm at C{path}. @param path: The path of the dirdbm to copy to. If a dirdbm exists at the destination path, it is cleared first. @rtype: C{DirDBM} @return: The dirdbm this dirdbm was copied to. )rrAssertionError __class__rclearrI)rrZfpathdr"rrr copyTo s   z DirDBM.copyTocCs|D] }||=qdS)z< Delete all key/value pairs in this dirdbm. N)rIr*rrr r`s z DirDBM.clearcCsdS)zL Close this dbm: no-op, for dbm-style interface compliance. Nrr:rrr close$sz DirDBM.closefloatcCsBt|tkstd|j||}|r6|St|dS)z Returns modification time of an entry. @return: Last modification date (seconds since epoch) of entry C{key} @raise KeyError: Raised when there is no such key r=N) r>r r?rrr+rQgetModificationTimerD)rrMrrrr re)s  zDirDBM.getModificationTime)N)__name__ __module__ __qualname____doc__r!r+r-r1r6r;rBrErFrIrNrOrSrUrrWrZr]rbr`rcrerrrr r s4  "     r c@s eZdZdZddZddZdS)Shelfz A directory with a DBM shelf interface. This class presents a hash-like interface to a directory of small, flat files. Keys must be strings, but values can be any given object. cCst|}t|||dS)z C{shelf[foo] = bar} Create or modify a textfile in this directory. @type k: str @param k: The key to set @param v: The value to associate with C{key} N)pickledumpsr rB)rr"r<rrr rBAs zShelf.__setitem__cCstt||S)a C{dirdbm[foo]} Get and unpickle the contents of a file in this directory. @type k: bytes @param k: The key to lookup @return: The value associated with the given key @raise KeyError: Raised if the given key does not exist )rkloadsr rEr*rrr rENs zShelf.__getitem__N)rfrgrhrirBrErrrr rj9s rjcCst|S)z This is for 'anydbm' compatibility. @param file: The parameter to pass to the DirDBM constructor. @param flag: ignored @param mode: ignored )r )fileflagmoderrr r.\s r.)NN)ri __future__rr'rrrktypingrrrrrZtwisted.python.filepathrr r rjr.__all__rrrr s   #