o i1)@sdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZe dddZGdd d ZGd d d eZdd d ZgdZdS)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@seZdZdZdFddZdGd d ZdGd d ZdHddZdIddZdJddZ dKddZ dGddZ dLddZ dMdd Z dMd!d"ZdNd$d%ZdOd(d)ZdPd+d,ZedQd-d.ZedRd2d.ZdSdTd6d.ZdOd7d8ZdUd;d<ZdVd=d>ZdWd?d@ZdWdAdBZdXdDdEZd3S)YDirDBMz 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. namebytesreturnNonecCstj||_t||_|js|jdSt|j djD]}t |q"t|j dj}|D]}|dd}tj |rJt |q6t ||q6dS)zJ @param name: Base path to use for the directory storage. z*.newz*.rplN) ospathabspathZdnamer _dnamePathisdirZcreateDirectoryglobchildremoveexistsrename)selfr f replacementsoldr^C:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/persisted/dirdbm.py__init__(s      zDirDBM.__init__kcCst|ddddS)z? Encode a key so it can be used as a filename.  _/-N)base64 encodebytesreplacerr"rrr _encodeCszDirDBM._encodecCst|ddddS)z3 Decode a filename to get the key. r$r#r&r%N)r' decodebytesr)r*rrr _decodeJzDirDBM._decoderFilePath[AnyStr]cCs6| }|}Wd|S1swY|S)z Read in the contents of a file. Override in subclasses to e.g. provide transparently encrypted dirdbm. N)openread)rrrsrrr _readFilePs   zDirDBM._readFiledatacCsB|d}|||WddS1swYdS)zw Write data to a file. Override in subclasses to e.g. provide transparently encrypted dirdbm. wN)r0writeflush)rrr4rrrr _writeFileZs   "zDirDBM._writeFileintcCst|jS)zF @return: The number of key/value pairs in this Shelf N)lenrlistdirrrrr __len__dszDirDBM.__len__vcCst|tks tdt|tkstd||}|j|}|r)|d}n|d}z|||Wn t yB| w|rK| | |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+rrrZsiblingExtensionr8 BaseExceptionrZmoveTo)rr"r>rnewrrr __setitem__js$       zDirDBM.__setitem__cCsJt|tks td|j||}z||WSty$t|w)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 rArrr+r3OSErrorKeyError)rr"rrrr __getitem__s   zDirDBM.__getitem__cCsRt|tks td||}z |j|WdSty(t| |w)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 rAr+rrrrErFr-r*rrr __delitem__s   zDirDBM.__delitem__Iterable[bytes]cCstt|j|jS)z9 @return: a L{list} of filenames (keys). N)listmapr-rZ asBytesModer;r<rrr keysr.z DirDBM.keyscCs(g}|}|D] }|||q|S)z? @return: a L{list} of file-contents (values). NrLappend)rvalsrLkeyrrr valuess z DirDBM.valuesIterable[tuple[bytes, bytes]]cCs,g}|}|D] }||||fq|S)zL @return: a L{list} of 2-tuples containing key/value pairs. NrM)ritemsrLrPrrr rSs z DirDBM.itemsrPboolcCs.t|tks td||}|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?N)r@r rAr+rrisfilerrPrrr has_keys  zDirDBM.has_keyvaluecCs||vr |||<|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. Nr)rrPrXrrr setdefaultszDirDBM.setdefaultcCdSNrrVrrr getz DirDBM.getdefaultr bytes | _TcCrZr[rrrPr^rrr r\r]N _T | Nonebytes | _T | NonecCs||vr||S|S)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)} Nrr`rrr r\s cCs ||S)z) @see: L{DirDBM.has_key} N)rWrVrrr __contains__s zDirDBM.__contains__otherMapping[bytes, bytes]cCs|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)rS)rrdrPvalrrr updates z DirDBM.updatecCsHt|}||jks J||j}||D]}||||<q|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. N)rr __class__rclearrL)rrZfpathdr"rrr copyTo s   z DirDBM.copyTocCs|D]}||=qdS)z< Delete all key/value pairs in this dirdbm. N)rLr*rrr ris z DirDBM.clearcCsdS)zL Close this dbm: no-op, for dbm-style interface compliance. Nrr<rrr close$sz DirDBM.closefloatcCs>t|tks td|j||}|r|St|)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 rArrr+rUgetModificationTimerF)rrPrrrr rn)s zDirDBM.getModificationTime)r r rr)r"r rr )rr/rr )rr/r4r rr)rr9)r"r r>r rr)r"r rr)rrI)rrR)rPr rrT)rPr rXr rr )rPr rr )rPr r^r rr_r[)rPr r^rarrb)rdrerr)rr rr )rr)rPr rrm)__name__ __module__ __qualname____doc__r!r+r-r3r8r=rDrGrHrLrQrSrWrYrr\rcrgrkrirlrnrrrr r s6      "           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 rD)rr"r>rrr rDAs 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 N)rtloadsr rGr*rrr rGNs zShelf.__getitem__N)rorprqrrrDrGrrrr rs9s rscCst|S)z This is for 'anydbm' compatibility. @param file: The parameter to pass to the DirDBM constructor. @param flag: ignored @param mode: ignored N)r )fileflagmoderrr r0\s r0)r0r rs)NN)rr __future__rr'rrrttypingrrrrrZtwisted.python.filepathrr r rsr0__all__rrrr s    #