U
SW @ s d dl Z d dlZz d dlZW n ek
r8 d dlZY nX d dlT d dlmZ d dlmZ zd dl m
Z
dd Z W n ek
r dd Z Y nX e d e
fi Ze d
d d ZG d
d deZG dd de
ZG dd deZG dd deZdS ) N)*)Node) PickledField APSWDatabasec K s t | f|S Nr Zdb_namekwargs r
/kv.pyKeyValueDatabase s r c K s t | fddi|S )NZcheck_same_threadF)SqliteDatabaser r
r
r r s Sentinelz:memory:F)Z threadlocalsc @ s e Zd Zdd Zdd ZdS ) JSONFieldc C s
t |S r )jsondumpsselfvaluer
r
r db_value s zJSONField.db_valuec C s |d k rt |S d S r )r loadsr r
r
r python_value s zJSONField.python_valueN)__name__
__module__ __qualname__r r r
r
r
r r s r c @ s e Zd ZdZd&ddZdd Zdd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z dd Z
dd Zdd Zd d Zd d Zd'd d!Zefd"d#Zd$d% ZdS )(KeyStorea#
Rich dictionary with support for storing a wide variety of data types.
:param peewee.Field value_type: Field type to use for values.
:param boolean ordered: Whether keys should be returned in sorted order.
:param peewee.Model model: Model class to use for Keys/Values.
FNc C sb || _ || _|pt| _| j | _| | _| jj| _| jj | _ | j
| jd t | jt | _
d S )NT) _value_field_ordered key_value_db _databaseZcompilerZ _compiler create_modelmodelkeyr Z create_table
isinstancer
_native_upsert)r Z value_fieldordereddatabaser
r
r __init__( s
zKeyStore.__init__c s G fdddt }|S )Nc s0 e Zd ZedddZ jZG fdddZdS )z&KeyStore.create_model.<locals>.KVModel T)Z
max_lengthZ primary_keyc s e Zd Z jZdS )z+KeyStore.create_model.<locals>.KVModel.MetaN)r r r r r& r
r r
r Meta; s r* N)r r r Z CharFieldr" r r r* r
r) r
r KVModel7 s r+ )ZModel)r r+ r
r) r r 6 s zKeyStore.create_modelc G s& | j j| }| jr"|| j}|S r )r! selectZtuplesr Zorder_byr" )r r, queryr
r
r r- @ s zKeyStore.queryc C s t |ts| j|kdfS |dfS )NTF)r# r r" )r noder
r
r convert_nodeF s
zKeyStore.convert_nodec C s" | |\}}| j | S r )r/ r! r, whereexists)r r" r. _r
r
r __contains__K s zKeyStore.__contains__c C s | j S r )r! r, countr) r
r
r __len__O s zKeyStore.__len__c sf | |\}}| | j|}td fdd|D }t|dkrV|rVt|n |rb|d S |S )Nr c s g | ] } |qS r
r
).0valZ item_getterr
r
<listcomp>V s z(KeyStore.__getitem__.<locals>.<listcomp>)r/ r- r r0 operator
itemgetterlenKeyError)r r. converted is_singleresultr
r8 r __getitem__R s
zKeyStore.__getitem__c C s* | j jf | jj|| jj|i d S r )r! insertr" namer Zupsertexecute)r r" r r
r
r _upsert] s zKeyStore._upsertc C s t |tr0| jj|i}| jjf || nf| jrD| || nRz| jj
||d W n< | j | jjf | jj|i| j
|k Y nX d S )N)r" r )r# r r rC r! updater0 rD r$ rE Zcreater Zrollbackr" )r r. r rF r
r
r __setitem__b s
zKeyStore.__setitem__c C s& | |\}}| j | d S r )r/ r! deleter0 rD )r r. r> r2 r
r
r __delitem__r s zKeyStore.__delitem__c C s | S r )r- rD r) r
r
r __iter__v s zKeyStore.__iter__c C s t td| | jS Nr )mapr: r; r- r" r) r
r
r keysy s z
KeyStore.keysc C s t td| | jS rK )rL r: r; r- r r) r
r
r values| s zKeyStore.valuesc C s t | S r )iterr) r
r
r items s zKeyStore.itemsc C s( z
| | W S t k
r" | Y S X d S r )r= )r kdefaultr
r
r get s
z KeyStore.getc
C sj | j V | |\}}z | | }W n. tk
rT |tkr@ | Y W 5 Q R S X | |= W 5 Q R X |S r )r Z transactionr/ r= r )r rQ rR r. r? resr
r
r pop s z KeyStore.popc C s | j d S r )r! rH rD r) r
r
r clear s zKeyStore.clear)FN)N)r r r __doc__r' r r- r/ r3 r5 rA rE rG rI rJ rM rN rP rS r rU rV r
r
r
r r s$
r c s e Zd Zd fdd Z ZS )PickledKeyStoreFNc s t t| t || d S r )superrX r' r )r r% r& __class__r
r r' s
zPickledKeyStore.__init__)FNr r r r'
__classcell__r
r
rZ r rX s rX c s e Zd Zd fdd Z ZS ) JSONKeyStoreFNc s" t dd}tt| ||| d S )NT)Znull)r rY r^ r' )r r% r& ZfieldrZ r
r r' s
zJSONKeyStore.__init__)FNr\ r
r
rZ r r^ s r^ )r: pickleZ
simplejsonr ImportErrorZpeeweer Zplayhouse.fieldsr Zplayhouse.apsw_extr r typeobjectr r Z TextFieldr r rX r^ r
r
r
r <module> s& x