-iY0dZddlZddlZddlmZddlZddlZddlZ ddlZddl Z dZ n #e $rdZ YnwxYwe Z defdZdZGd d Ze rGd d ZdSGd d ZdS)zz pickleDB - https://patx.github.io/pickledb Harrison Erd - https://harrisonerd.com/ Licensed - BSD 3 Clause (see LICENSE) N)AnyTFreturncP tjdS#t$rYdSwxYw)z.wrapper)sCtD*4***6** :: K{4   r r )rrs` r dualmethodr"s#!!!!! Nr ceZdZdZdefdZdZdZdZdZ e de fd Z e de fd Z e de fd Ze dd Ze de fdZe dZe de fdZd S)PickleDBz A unified async/sync key-value store using orjson + aiofiles. All data is kept in-memory in `self.db` and serialized to disk as a single orjson-encoded file at `self.location`. locationctj||_i|_t j|_dSr)ospath expanduserrdbrLock_lock)rrs r __init__zPickleDB.__init__9s1**844 "$\^^ r c.||Srloadrs r __enter__zPickleDB.__enter__>s  r c6||dSdSrsaverexc_typeexc_valexc_tbs r __exit__zPickleDB.__exit__Bs!   IIKKKKK  r c>K|d{V|Srr$r&s r __aenter__zPickleDB.__aenter__Fs+iikk r cFK||d{VdSdSrr)r+s r __aexit__zPickleDB.__aexit__Js:  ))++           r rcKtj|jrtj|jdkrst j|jd4d{V}|d{V}dddd{Vn#1d{VswxYwYtj |}ni}|j 4d{V||_ dddd{Vn#1d{VswxYwY|S)z Load JSON database from disk into memory. Returns `self` to allow chaining: db = PickleDB("Example.json").load() rrbN) rrexistsrgetsizeaiofilesopenreadorjsonloadsr!r)rfdatanew_dbs r r%z PickleDB.loadNs 7>>$- ( ( RW__T]-K-Ka-O-O}T]D99 & & & & & & &QVVXX~~~~~~ & & & & & & & & & & & & & & & & & & & & & & & & & & &\$''FFF:        DG                            s$/B B&)B&C++ C58C5cK|jd}|j4d{Vtj|d4d{V}|t j|jd{Vdddd{Vn#1d{VswxYwYtj tj ||jd{Vdddd{Vn#1d{VswxYwYdS)z Atomically save database to disk. Writes to `.tmp` and then os.replace() over the original file. Returns True on success. z.tmpNwbT) rr!r8r9writer;dumpsrr to_threadrreplace)rtempr=s r r*z PickleDB.saveas-%%%: E E E E E E E E}T400 5 5 5 5 5 5 5Aggfl4733444444444 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5#BJdmDD D D D D D D D E E E E E E E E E E E E E E E E E E E E E E E E E E Ets4C 3A;)C ; B C B .C  CCcK|j4d{V||jt|<dddd{Vn#1d{VswxYwYdS)z*Set a key-value pair. Always returns True.NT)r!rstr)rkeyvalues r setz PickleDB.setps: & & & & & & & & %DGCHH  & & & & & & & & & & & & & & & & & & & & & & & & & & &ts: AANcK|j4d{V|jt||cdddd{VS#1d{VswxYwYdS)z+Get a key's value, or `default` if missing.N)r!rgetrH)rrIdefaults r rMz PickleDB.getws: 2 2 2 2 2 2 2 27;;s3xx11 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s(A  AAcK|j4d{V|jt|dducdddd{VS#1d{VswxYwYdS)zERemove a key-value pair. Returns True if it existed, False otherwise.N)r!rpoprH)rrIs r removezPickleDB.remove}s: ; ; ; ; ; ; ; ;7;;s3xx..d: ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;s*A  AAcK|j4d{Vt|jcdddd{VS#1d{VswxYwYdS)zReturn a list of all keys.N)r!listrkeysr&s r allz PickleDB.alls: ( ( ( ( ( ( ( ( '' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s&A  AAcK|j4d{V|jdddd{Vn#1d{VswxYwYdS)zBRemove all key-value pairs from the database. Always returns True.NT)r!rclearr&s r purgezPickleDB.purges:         GMMOOO                           ts< A Ar)__name__ __module__ __qualname____doc__rHr"r'r/r1r3rboolr%r*rKrMrQrUrXr r r rr1ss$$$$$ DZ$ D   Z tZ 222Z2 ;4;;;Z; ((Z( TZr rceZdZdZ ddededdfdZd edefd Zd edefd Z d edzd edefdZ e fd ededefdZ d ede fdZdeefdZde fdZddZdS)PickleDBSQLitea$ A unified async/sync key-value store backed by SQLite. Each key is stored as a row: CREATE TABLE kv ( key TEXT PRIMARY KEY, value BLOB NOT NULL ) Values are stored as JSON-encoded bytes via orjson. pickledb.sqlite3kv sqlite_path table_namerNc||_||_tj|jd|_tj|j_|jd|jd|jdS)NF)check_same_threadz, CREATE TABLE IF NOT EXISTS zy ( key TEXT PRIMARY KEY, value BLOB NOT NULL ) ) rbrcsqlite3connect_connRow row_factoryexecutecommit)rrbrcs r r"zPickleDBSQLite.__init__s +D (DO )9USSSDJ%,[DJ " J  ,0O    J       r rJc*tj|S)z2Serialize a Python object to orjson-encoded bytes.)r;rC)rrJs r _dumpszPickleDBSQLite._dumpss<&& &r r>c*tj|S)z;Deserialize orjson-encoded bytes back into a Python object.)r;r<)rr>s r _loadszPickleDBSQLite._loadss<%% %r rIctrdtffd }|S}attj}jdjd||fj|Sjdjdt|fjtS)z Set a key-value pair. If key is None, generate a new random UUID key and return it. In async code, returns a coroutine you must `await`. In sync code, returns the key string directly. rcKtjj4d{V}tj|_}uttj }| dj d||fd{V| d{V|cdddd{VS| dj dt|fd{V| d{Vtcdddd{VS#1d{VswxYwYdS)N INSERT INTO  (key, value) VALUES (?, ?)z) INSERT INTO z (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value=excluded.value ) aiosqlitergrbrfrirjrnrHuuiduuid4rkrcrl)rpayloadnew_keyrIrrJs r _asetz!PickleDBSQLite.set.._asetsI(01ABB(((((((b)0"&++e"4"4;&)$*,,&7&7G"$** [t [ [ [!(' 2###%))++-------#*((((((((((((((!jj )-   !XXw/ !iikk)))))))"3xx-((((((((((((((((((((((((((((((sB D0AD00 D:=D:Nrsrtz INSERT INTO z (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value=excluded.value ) r rHrnrvrwrhrkrcrl)rrIrJrzrxrys``` r rKzPickleDBSQLite.sets)zz (S((((((((2uwwkk%((G{djll++ ""O4?OOOg& !!### J  !_ S7#     J     s88Or rNcPtrdtffd }|Sjdjdt f}|}|turtS |dS)a9 Get the value for a key. If the key does not exist: - If default is MISSING, raises KeyError. - Otherwise returns default. In async code, returns a coroutine you must `await`. In sync code, returns the value directly. rcKtjj4d{V}tj|_|djdtfd{V}| d{V}| d{V|,turtcdddd{VS |dcdddd{VS#1d{VswxYwYdS)NSELECT value FROM  WHERE key = ?rJ)rurgrbrfrirjrkrcrHfetchonecloseMISSINGKeyErrorrp)rcursorrowrNrIrs r _agetz!PickleDBSQLite.get.._agets(01ABB9999999b)0')zzPPPP XXK((""""""%+OO$5$5555555$llnn,,,,,,,;&'11&.smm 3#*99999999999999 ${{3w<88999999999999999999999999999999sBC7 C77 DDr}r~NrJ) r rrhrkrcrHrrrrp)rrIrNrrrs``` r rMzPickleDBSQLite.getszz 9S99999999"uwwZ''DT_DDDS F//##C{g%%"3--';;s7|,, ,r ctrdtffd }|Sjdjdt f}j|jdkS)z Remove a key-value pair. Returns True if a row was deleted, False otherwise. In async code, returns a coroutine you must `await`. In sync code, returns a bool directly. rcBKtjj4d{V}|djdt fd{V}|d{V|jdkcdddd{VS#1d{VswxYwYdS)N DELETE FROM r~r)rurgrbrkrcrHrlrowcount)rrrIrs r _aremovez'PickleDBSQLite.remove.._aremove*sN(01ABB3333333b')zzJ4?JJJ XXK((""""""!iikk)))))))%2 333333333333333333333333333333sAB BBrr~r)r r]rhrkrcrHrlr)rrIrrs`` r rQzPickleDBSQLite.remove szz "33333333 xzz!Z''>t>>>S F J     ?Q& &r ctr"dttffd }|Sjdjd}d|DS)z Return a list of all keys in the database. In async code, returns a coroutine you must `await`. In sync code, returns the list directly. rc|Ktjj4d{V}tj|_|djdd{V}|d{V}| d{Vd|Dcdddd{VS#1d{VswxYwYdS)NSELECT key FROM ORDER BY keycg|] }|d SrIr .0rs r z5PickleDBSQLite.all.._aall..Ls;;;sE ;;;r ) rurgrbrfrirjrkrcfetchallr)rrrowsrs r _aallz!PickleDBSQLite.all.._aallDst(01ABB<<<<<<.Ss<<<3CJ<<._apurge_s((01ABB$$$$$$$b jj)I)I)IJJJJJJJJJ iikk)))))))#$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$s?A55 A?A?rT)r r]rhrkrcrl)rrs` r rXzPickleDBSQLite.purgeUs~zz !$t$$$$$$ wyy J  ?do?? @ @ @ J     4r cvtrdfd }|SjdS)z Close the underlying sync SQLite connection. In async code, returns a coroutine you must `await`. In sync code, closes immediately. rNc>KjdSr)rhrr&sr _aclosez%PickleDBSQLite.close.._aclosess!J$$&&&&&r rN)r rhr)rrs` r rzPickleDBSQLite.closeksUzz !''''''wyy J       r )r`rar)rYrZr[r\rHr"rbytesrnrprKrrMr]rQrSrUrXrr r r r_r_sq  2"         * ' ' ' ' ' ' &u & & & & &8 3:8 c8 c8 8 8 8 t07( -( -3( -( -3( -( -( -( -T 'c 'd ' ' ' '8 =c = = = =2 4    ,      r r_c&eZdZdZdededdfdZdS)r_zy This class is only usable if `aiosqlite` is installed, e.g.: pip install "pickledb[sqlite]" rrrNc td)NzUPickleDBSQLite requires `aiosqlite`. Install it via `pip install "pickledb[sqlite]"`.)r)rrrs r r"zPickleDBSQLite.__init__sE r )rYrZr[r\rr"r r r r_r_{sG   #        r )r\rrtypingrrvr;r8rfru sqlite_enable ImportErrorobjectrr]r rrr_r r r rsw   NNNMMMMM &(($   ]]]]]]]]~uffffffffffT          s )33