U
SW 3 @ s
d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
mZ d dl
mZ d dlmZ d dlmZ zd dlmZ W n ek
r eZY nX d dlmZ d dlmZ d d lmZ d
Z d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dlm Z W n( ek
r G d?d@ d@ej Z Y nX e dAZ!e!"e dBZ#e$fdCdDZ%ej&d dEkZ'ej&d dFkZ(ej&ddE dGkZ)e(r@d dl*Z*d dHlm+Z+ d dIlm,Z, dJdK Z-e.Z/e0Z1e.Z2e3e*dLZ4dMdK Z5e6Z7d]dNdOZ8n,e'rde9Z/e2Z1e:Z5dPdQ Z4e;dR nendVdK Z>zd dWl?m@ZA W n ek
r dZAY nX z d dlBZBW n ek
r eAZBY nX eAreAjCeBjCkreAZBzd dXlDmEZE eEF W n ek
r" Y nX zd dlGZGd dYlGmHZI W n ek
rV dZGY nX z d dlJZKW n> ek
r z d dlLZKW n ek
r dZKY nX Y nX z(d dZlMmNZN d d[lMmOZO d d\lMmPZP W n2 ek
r d^d]d^ZNd_d` ZOdadb ZPY nX z4d dclMmQZQ d ddlMmRZR d delMmSZS d dflMmTZT W n& ek
rZ d ZQ ZR ZSZTY nX eBreBUejVe. eBUejWe. eBUej
e. dgdhdidjdkdlgZXeYeXZZdmZ[dndo Z\dpdqdrdsdtdudvZ]e] Z^dwe^dk< dxe^dl< dydz Z_d{d| Z`G d}d~ d~eZaeaddddddddddddddddddddddddddZbeadddddZcecjdZeecjfZgecjhZidZjdEZkdFZldZmdZnebjoebjpebjqebjrebjsebjtebjuebjvebjwebjxebjyd Zzdd Z{dd Z|dd Z}G dd3 d3e$Z~G dd de$ZG dd de$ZG dd de$ZG dd7 d7eZeZG dd deZG dd deZedZG dd deZG dd. d.eZG dd deZG dd deZG dd deZG dd deZG dd= d=eZdd ZG dd deZG dd deZeddÃZG ddń deddǃZG ddɄ de$ZG dd d eZG dd d eZG dd" d"eZG dd d eZG dd5 d5eZG dd1 d1eZG ddф deZG dd d eZG dd deZG dd deZd_ddׄZG dd deZG dd d eZG dd8 d8eZG dd
d
eZG dd< d d>e܃ZG d?d@ d@e܃ZG dAdB dBeZG dCdD dDeZG dEdF dFe$ZeSdk rG dGdH dHe$ZSG dId deɃZeB reِdJZndZG dKdL dLe$ZG dMdN dNeZG dOd* d*e%eZdPdQ ZdRdS ZedTdUZG dVdW dWeZdXd0 ZdYdZ Zd[d\ ZdS (` N) bisect_left) bisect_right)deque)
namedtuple) OrderedDict)deepcopywraps)isclassz2.8.4 BareFieldBigIntegerField BlobField BooleanField CharFieldCheckClause CompositeKey
DatabaseError DataError DateField
DateTimeField DecimalFieldDeferredRelation DoesNotExist DoubleFieldDQFieldFixedCharField
FloatFieldfnForeignKeyFieldImproperlyConfigured IntegerFieldIntegrityErrorInterfaceError
InternalErrorJOIN JOIN_FULL
JOIN_INNERJOIN_LEFT_OUTERModel
MySQLDatabaseNotSupportedErrorOperationalErrorParamPostgresqlDatabaseprefetchPrimaryKeyFieldProgrammingErrorProxyRSmallIntegerFieldSqliteDatabaseSQL TextField TimeFieldTimestampFieldUsing UUIDFieldWindow) NullHandlerc @ s e Zd Zdd ZdS )r> c C s d S N )selfrecordr@ r@
/peewee.pyemiti s zNullHandler.emitN)__name__
__module__ __qualname__rD r@ r@ r@ rC r> h s r> ZpeeweeZ_metaclass_helper_c C s | t |fi S r? ) _METACLASS_)metabaser@ r@ rC with_metaclasss s rK )rL )Callable)reducec C s
t | tS r? )
isinstancerO )cr@ r@ rC } rS printc C s t | dS NZraw_unicode_escape)bytesencodesr@ r@ rC rS rT c C s |j |k r|||d S r? )
__traceback__with_traceback)tpvaluetbr@ r@ rC reraise s
r` c C s t j| t jd d S )N
)sysstdoutwriterY r@ r@ rC print_ s re z4def reraise(tp, value, tb=None): raise tp, value, tbzUnsupported python version.@B c C s& | j d | j| jd d t t S )Ng i ) microsecondsZsecondsZdays_Mtr@ r@ rC rS rT c C s | S r? )
total_secondsrj r@ r@ rC rS rT )dbapi2)compat)
extensions)format_date_time)sort_models_topologically) strip_parensc C sH |p
dd }|D ]2}z|t j | |W S tk
r@ Y qX q| S )Nc S s | S r? r@ xr@ r@ rC rS rT z"format_date_time..)datetimestrptime
ValueError)r^ formatsZ post_processZfmtr@ r@ rC rp s rp c sT t t g fdd dd }t|ddD ] } | q:ttS )z@Sort models topologically so that parents will precede children.c sD | kr@| kr@ | | jj D ]} |j q&| d S r? )add_meta reverse_relvalues model_classappend)model foreign_keydfsmodelsZorderingseenr@ rC r s
z&sort_models_topologically..dfsc S s | j j| j jfS r? )rz namedb_table)mr@ r@ rC rS rT z+sort_models_topologically..T)keyreverse)setsortedlistreversed)r namesr r@ r rC rq s
rq c C s | r| d dkr| S d }}t | }||k rf| | dkrf| |d dkrf|d7 }|d7 }|d8 }q$qfq$|rd}d}t||| D ]N}| | dkr|d7 }n| | dkr|d8 }|dk r|d7 }d}||kr qq||8 }|dkr| || S | S )Nr ( ))lenrange)rZ ZctilZ
unbalanced_ctZrequiredr@ r@ rC rr s6
rr )_DictQueryResultWrapper)_ModelQueryResultWrapper)_SortedFieldList)_TuplesQueryResultWrapperyearmonthdayhourminutesecond)%Y-%m-%d %H:%M:%S%Y-%m-%d %H:%M:%S.%f%Y-%m-%d%H:%M:%S %H:%M:%S.%f%H:%Mc C s( | t ks t|sd S t|t}t|| S r? )DATETIME_LOOKUPSAssertionErrorrp SQLITE_DATETIME_FORMATSgetattrZ lookup_typeZdatetime_stringdtr@ r@ rC _sqlite_date_part s
r z%Yz%Y-%mr z %Y-%m-%d %Hz%Y-%m-%d %H:%Mr )r r r r r r z%Y-%m-%d %H:%iz%Y-%m-%d %H:%i:%Sc C s, | t ks t|sd S t|t}|t | S r? )SQLITE_DATE_TRUNC_MAPPINGr rp r strftimer r@ r@ rC _sqlite_date_trunc& s
r c C s t | |t jd k S r? )researchI)Zregexr^ r@ r@ rC _sqlite_regexp- s r c @ s e Zd Zdd ZdS )attrdictc C s | | S r? r@ rA attrr@ r@ rC __getattr__1 s zattrdict.__getattr__N)rE rF rG r r@ r@ r@ rC r 0 s r andor+-*/&|^%=<<=>>=!=inznot iniszis notlikeilikebetweenregexp||)ANDORADDSUBMULDIVBIN_ANDBIN_ORXORMODEQLTLTEGTGTENEINNOT_INISIS_NOTLIKEILIKEBETWEENREGEXPCONCATINNERz
LEFT OUTERz RIGHT OUTERFULL)r
LEFT_OUTER RIGHT_OUTERr r ) eqltZltegtZgtener r r r r c C s | }|| |S r? )copyupdate)sourceZ overridesZmergedr@ r@ rC
merge_dictr s
r c s fdd} |_ |S )z
Method decorator that will "clone" the object before applying the given
method. This ensures that state is mutated in a more predictable fashion,
and promotes the use of method-chaining.
c s | } |f|| |S r? clone)rA argskwargsr funcr@ rC inner} s z returns_clone..inner)Z
call_localr r r@ r rC
returns_clonew s r c s fdd}|S )zu
Method decorator to indicate a method is not allowed to be called. Will
raise a `NotImplementedError`.
c s t d t| jf d S )Nz!%s is not allowed on %s instances)NotImplementedErrortyperE rA r r r r@ rC r s znot_allowed..innerr@ r r@ r rC not_allowed s r c sH e Zd ZdZddgZdd Zdd Zdd Zd
d Z fd d
Z Z
S )r3 zg
Proxy class useful for situations when you wish to defer the initialization
of an object.
obj
_callbacksc C s g | _ | d d S r? )r
initializerA r@ r@ rC __init__ s zProxy.__init__c C s || _ | jD ] }|| q d S r? )r r )rA r callbackr@ r@ rC r s
zProxy.initializec C s | j | |S r? )r r~ )rA r r@ r@ rC attach_callback s zProxy.attach_callbackc C s | j d krtdt| j |S )Nz Cannot use uninitialized Proxy.)r AttributeErrorr r r@ r@ rC r s
zProxy.__getattr__c s$ || j krtdtt| ||S )Nz Cannot set attribute on proxy.) __slots__r
superr3 __setattr__rA r r^ __class__r@ rC r
s
zProxy.__setattr__) rE rF rG __doc__r r r r r r
__classcell__r@ r@ r rC r3 s c @ s8 e Zd Ze Zd
ddZdd Zdd Zedd Z dS ) r Nc C s( g | _ |d k r$| | _| j| d S r? )fieldslower_rel_model_name _unresolvedry )rA Zrel_model_namer@ r@ rC r s
zDeferredRelation.__init__c C s | j |||f d S r? )r r~ )rA r} fieldr r@ r@ rC set_field s zDeferredRelation.set_fieldc C s( | j D ] \}}}||_||| qd S r? )r rel_model add_to_class)rA r r r r r@ r@ rC set_model s zDeferredRelation.set_modelc C s> t tj}|D ]*}|j| j kr|| tj| qd S r? )r r r r rE r r discard)Z model_clsZ
unresolvedZdrr@ r@ rC resolve s
zDeferredRelation.resolve)N)
rE rF rG r r r r r staticmethodr r@ r@ r@ rC r s
c @ s e Zd ZdddZdS ) _CDescriptorNc C s |d k rt |jS | S r? )Entity_aliasrA instanceZ
instance_typer@ r@ rC __get__ s
z_CDescriptor.__get__)N)rE rF rG r$ r@ r@ r@ rC r s r c @ s e Zd ZdZe ZdZdd Zed5ddZ d d
Z
d d Z e d
d Z
e d6ddZe dd Ze dd Ze dd Zdd Zdd Zd7dd ZeejZeejZeejZeej Z eej Z eej Z!Z"eej#Z$eejd d Z%eej d d Z&eej d d Z'eej d d Z(Z)eejd d Z*eejd d Z+eej#d d Z,d d Z-d!d" Z.eej/Z0eej1Z2eej3Z4eej5Z6eej7Z8eej9Z:eej;Zeej?Z@eejAZBd#d$ ZCd%d& ZDd8d'd(ZEd)d* ZFd+d, ZGd-d. ZHd/d0 ZId1d2 ZJd3d4 ZKdS )9Nodez=Base-class for any part of a query which shall be composable.nodec C s d| _ d | _d | _d | _d S NF)_negatedr! _bind_to _orderingr r@ r@ rC r s z
Node.__init__NFc s fdd}|S )Nc s& p| j } rt| } t|| | S r? )rE r setattr)methodZ method_namer clsr r@ rC decorator s
z Node.extend..decoratorr@ )r. r r r/ r@ r- rC extend s z Node.extendc C s
t | S r? )r r r@ r@ rC
clone_base s zNode.clone_basec C s, | }| j|_| j|_| j|_| j|_|S r? )r1 r( r! r* r) )rA instr@ r@ rC r s z
Node.clonec C s | j | _ d S r? )r( r r@ r@ rC
__invert__ s zNode.__invert__c C s
|| _ d S r? r! )rA ar@ r@ rC alias s z
Node.aliasc C s
|| _ dS )z
Bind the results of an expression to a specific model type. Useful
when adding expressions to a select, where the result of the expression
should be placed on a joined instance.
N)r) )rA Zbtr@ r@ rC bind_to s z Node.bind_toc C s
d| _ d S )NZASCr* r r@ r@ rC asc s zNode.ascc C s
d| _ d S )NDESCr8 r r@ r@ rC desc s z Node.descc C s | S r? )r9 r r@ r@ rC __pos__ s z Node.__pos__c C s | S r? )r; r r@ r@ rC __neg__
s z Node.__neg__c s fdd}|S )z
Lightweight factory which returns a method that builds an Expression
consisting of the left-hand and right-hand operands, using `op`.
c s rt || S t | |S r? )
ExpressionrA rhsinvopr@ rC r s zNode._e..innerr@ )rC rB r r@ rA rC _e
s zNode._eT)rB c C s$ |d krt | tjd S t | tj|S r? )r> OPr r r? r@ r@ rC __eq__' s z Node.__eq__c C s$ |d krt | tjd S t | tj|S r? )r> rE r r r? r@ r@ rC __ne__+ s z Node.__ne__c C s t | tj|S r? )r> rE r r? r@ r@ rC in_= s zNode.in_c C s t | tj|S r? )r> rE r r? r@ r@ rC not_in? s z Node.not_inc C s |rt | tjd S t | tjd S r? )r> rE r r )rA is_nullr@ r@ rC rJ A s z Node.is_nullc C s t | tjd| S )Nz%%%s%%r> rE r r? r@ r@ rC containsE s z
Node.containsc C s t | tjd| S )Nz%s%%rK r? r@ r@ rC
startswithG s zNode.startswithc C s t | tjd| S )Nz%%%srK r? r@ r@ rC endswithI s z
Node.endswithc C s t | tjt|td|S )Nr )r> rE r r r4 )rA ZlowZhighr@ r@ rC r K s z Node.betweenc C s t | tj|S r? )r> rE r )rA
expressionr@ r@ rC r M s z Node.regexpc C s t | tj|S r? )r> rE r r? r@ r@ rC concatO s z Node.concat)NF)N)F)T)LrE rF rG r r rR
_node_typer classmethodr0 r1 r r r3 r6 r7 r9 r; r< r= rD rE r __and__r __or__r __add__r __sub__r __mul__r Z__div__ __truediv__r __xor____radd____rsub____rmul__Z__rdiv__ __rtruediv____rand____ror____rxor__rF rG r __lt__r __le__r __gt__r __ge__r
__lshift__r
__rshift__r __mod__r __pow__r Zbin_andr Zbin_orrH rI rJ rL rM rN r r rP r@ r@ r@ rC r% sp
r% c s, e Zd ZdZdZ fddZdd Z ZS )r7 z2An unescaped SQL string, with optional parameters.sqlc s || _ || _tt| d S r? )r^ paramsr r7 r )rA r^ rj r r@ rC r V s z SQL.__init__c C s t | jf| j S r? )r7 r^ rj r r@ r@ rC r1 [ s zSQL.clone_baserE rF rG r rQ r r1 r r@ r@ r rC r7 R s c s4 e Zd ZdZdZ fddZdd Zdd Z ZS ) r z/A quoted-name or entity, e.g. "table"."column".entityc s t t| || _d S r? )r r r path)rA rm r r@ rC r c s zEntity.__init__c C s
t | j S r? )r rm r r@ r@ rC r1 g s zEntity.clone_basec C s t td | j|f S r? )r filterrm r r@ r@ rC r j s zEntity.__getattr__) rE rF rG r rQ r r1 r r r@ r@ r rC r _ s
r c sT e Zd ZdZdZedZ fddZedddZ d d
Z
dd d
Z dd Z Z
S )Funcz An arbitrary SQL function call.r )countsumc s6 || _ || _|r | | jknd| _tt| d S r' )r argumentsr
_no_coerce_coercer ro r )rA r rr r r@ rC r r s z
Func.__init__Tc C s
|| _ d S r? )rt )rA coercer@ r@ rC ru x s z Func.coercec C s t | jf| j }| j|_|S r? )ro r rr rt )rA resr@ r@ rC r1 | s zFunc.clone_baseNc C sJ t |tr|d kr|}|d kr0t||d }n
t|j}t| td|S )N) partition_byorder_byZOVER)rQ r= __sql__r7 r! r )rA rw rx windowri r@ r@ rC over s
z Func.overc s fdd}|S )Nc s t f| |S r? )ro r r r r@ rC dec s z Func.__getattr__..decr@ )rA r r~ r@ r} rC r s zFunc.__getattr__)T)NNN)rE rF rG r rQ r rs r r ru r1 r{ r r r@ r@ r rC ro m s
ro c s. e Zd ZdZdZd fdd Zdd Z ZS ) r> z0A binary expression, e.g `foo + 1` or `bar < 7`.rO Fc s* t t| || _|| _|| _|| _d S r? )r r> r lhsrC r@ flat)rA r rC r@ r r r@ rC r s
zExpression.__init__c C s t | j| j| j| jS r? )r> r rC r@ r r r@ r@ rC r1 s zExpression.clone_base)Frk r@ r@ r rC r> s r> c s. e Zd ZdZdZd fdd Zdd Z ZS ) r. z
Arbitrary parameter passed into a query. Instructs the query compiler to
specifically treat this value as a parameter, useful for `list` which is
special-cased for `IN` lookups.
paramNc s || _ || _tt| d S r? )r^ adaptr r. r )rA r^ r r r@ rC r s zParam.__init__c C s t | j| jS r? )r. r^ r r r@ r@ rC r1 s zParam.clone_base)Nrk r@ r@ r rC r. s c @ s e Zd ZdZdS ) Passthrough passthroughN)rE rF rG rQ r@ r@ r@ rC r s r c s4 e Zd ZdZdZdZdZ fddZdd Z Z S ) r z8A SQL clause, one or more Node objects joined by spaces.clause Fc s@ d|kr|d | _ d|kr$|d | _tt| t|| _d S )Nglueparens)r r r r r r nodes)rA r r r r@ rC r s
zClause.__init__c C s t | j }| j|_| j|_|S r? )r r r r rA r r@ r@ rC r1 s
zClause.clone_base)
rE rF rG r rQ r r r r1 r r@ r@ r rC r s c @ s e Zd ZdZdZdS ) CommaClausez5One or more Node objects joined by commas, no parens., N)rE rF rG r r r@ r@ r@ rC r s r c @ s e Zd ZdZdZdS )EnclosedClausezAOne or more Node objects joined by commas and enclosed in parens.TN)rE rF rG r r r@ r@ r@ rC r s r c s. e Zd Zd fdd Zdd Zdd Z ZS ) r= Nc s* t t| || _|| _| jp"d| _d S )Nw)r r= r rw rx r! )rA rw rx r r@ rC r s zWindow.__init__c C sP g }| j r$|ttdt| j | jrD|ttdt| j tt| S )Nz PARTITION BYORDER BY)rw r~ r r7 r rx r )rA Z over_clausesr@ r@ rC ry s zWindow.__sql__c C s t | j| jS r? )r= rw rx r r@ r@ rC r1 s zWindow.clone_base)NN)rE rF rG r ry r1 r r@ r@ r rC r= s c C s t d| S )Nz
CHECK (%s)r7 )r^ r@ r@ rC r s c s( e Zd ZdZ fddZdd Z ZS )r z:A "django-style" filter expression, e.g. {'foo__eq': 'x'}.c s t t| || _d S r? )r r r queryrA r r r@ rC r s z DQ.__init__c C s t f | jS r? )r r r r@ r@ rC r1 s z
DQ.clone_base)rE rF rG r r r1 r r@ r@ r rC r s c s e Zd ZdZ fddZ ZS ) _StripParensrr c s t t| || _d S r? )r r r r& )rA r& r r@ rC r s z_StripParens.__init__)rE rF rG rQ r r r@ r@ r rC r s r JoinMetadata src_model
dest_modelsrcdestr primary_keyr
is_backrefr6 is_self_join
is_expressionc @ s: e Zd Zd ddZdd Zdd Zdd Zed
d ZdS )
JoinNc C sX t |tst |trdS |j||}|d k r6|dfS |j||}|d k rT|dfS dS )NNNFT)rQ SelectQueryrz
rel_for_modelreverse_rel_for_model)rA r r r fk_fieldr{ r@ r@ rC get_foreign_key
s zJoin.get_foreign_keyc C s | j p
tjS r? ) join_typer& r r r@ r@ rC
get_join_type s zJoin.get_join_typec C s$ t |tr|jS t |tr |jS |S r? )rQ
ModelAliasr} r )rA model_or_aliasr@ r@ rC model_from_alias s
zJoin.model_from_aliasc
C sB | | j}| | j}t| jtr,| jjp.d }t| jttt f}t| jt
t frX| jpZd }|rt|}|j |j
jk}n6| ||| j\}}|d kr| jd k r| ||\}}|d k r|j}nd }|s|d k r|r|j
j} n|j } n.z| jjj } W n tk
r |j
j} Y nX nd } t||| j| j|p,| ||||||k|d S )Nr )r r r rQ onr% r! r> ro r7 r
FieldProxyr rz r r to_fieldr r r
r )
rA r r Z
join_aliasr Zon_fieldr r r Z target_attrr@ r@ rC _join_metadata sJ
zJoin._join_metadatac C s t | ds| | _| jS )N_cached_metadata)hasattrr r r r@ r@ rC metadataP s
z
Join.metadata)N) rE rF rG r r r r propertyr r@ r@ r@ rC r s
1r Z_Join)r r r r c @ s& e Zd Zdd ZdddZdd ZdS ) FieldDescriptorc C s || _ | j j| _d S r? )r r att_namerA r r@ r@ rC r Y s zFieldDescriptor.__init__Nc C s |d k r|j | jS | jS r? )_datagetr r r" r@ r@ rC r$ ] s zFieldDescriptor.__get__c C s ||j | j< |j| j d S r? )r r _dirtyry rA r# r^ r@ r@ rC __set__b s zFieldDescriptor.__set__)N)rE rF rG r r$ r r@ r@ r@ rC r V s
r c s e Zd ZdZdZdZdZdZd# fdd Zd d
Z d d Z
d
d Z dd Z dd Z
dd Zdd Zdd Zdd Zd$dd Zd d Zd d Zd!d" Z ZS )%r zA column on a table.r r unknownFNc
s || _ || _|| _|| _|| _|| _|| _|| _| | _|
| _ | | _
| | _ t j
d7 _
t j
| _| jrhdpjd| jf| _d| _tt | d S )Nr rL F)nullindexunique verbose_name help_text db_columndefaultchoicesr sequence constraintsschemar _field_counter_order _sort_key _is_boundr r )
rA r r r r r r r r r r r r r r@ rC r m s" zField.__init__c K sf t | f | j| j| j| j| j| j| j| j| j | j
| j | j d |}| j
rZ| j|_| j|_| j
|_
|S )N) r r r r r r r r r r r r )r r r r r r r r r r r r r r r r} )rA r r2 r@ r@ rC r1 s*
zField.clone_basec C sZ || _ || _| jp| j | _| js4tdd| | _|j| t ||t
| d| _ dS )z
Hook that replaces the `Field` attribute on a class with a named
`FieldDescriptor`. Called by the metaclass during construction of the
`Model`.
_+r TN) r r} r r r subtitlerz add_fieldr+ r r rA r} r r@ r@ rC r s zField.add_to_classc C s
| j jjS r? )r} rz databaser r@ r@ rC get_database s zField.get_databasec C s | }| |S r? ) get_db_fieldr compilerget_column_type)rA Z
field_typer@ r@ rC r s zField.get_column_typec C s | j S r? )db_fieldr r@ r@ rC r s zField.get_db_fieldc C s d S r? r@ r r@ r@ rC
get_modifiers s zField.get_modifiersc C s |S r? r@ rA r^ r@ r@ rC ru s z Field.coercec C s |dkr |S | |S )z5Convert the python value for storage in the database.Nru r r@ r@ rC db_value s zField.db_valuec C s |dkr |S | |S )z/Convert the database value to a pythonic value.Nr r r@ r@ rC python_value s zField.python_valuec C s |rt | jjj| jS t | jS r? )r r} rz r r )rA
with_tabler@ r@ rC as_entity s zField.as_entityc C s0 | }|r(td|dtt|f S t|S )z2Return the column type, e.g. VARCHAR(255) or REAL.%s(%s)r )r r7 joinmapstr)rA column_typeZ modifiersr@ r@ rC __ddl_column__ s zField.__ddl_column__c C sj | | |g}| js&|td | jr:|td | jrT|td| j | jrf|| j |S )z8Return a list of Node instances that defines the column.zNOT NULL PRIMARY KEYzDEFAULT NEXTVAL('%s')) r r r r~ r7 r r r r0 )rA r ddlr@ r@ rC __ddl__ s z
Field.__ddl__c C s t | jd | jj S N.)hashr r} rE r r@ r@ rC __hash__ s zField.__hash__) FFFNNNNNFNNN)F)rE rF rG r r r rQ r r r1 r r r r r ru r r r r r r r r@ r@ r rC r f s:
c s. e Zd ZdZd fdd Z fddZ ZS )r bareNc s$ t t| j|| |d k r || _d S r? )r r r ru )rA ru r r r r@ rC r s zBareField.__init__c s t t| jf d| ji|S )Nru )r r r1 ru rA r r r@ rC r1 s zBareField.clone_base)N)rE rF rG r r r1 r r@ r@ r rC r s c @ s e Zd ZdZeZdS )r" intN)rE rF rG r r ru r@ r@ r@ rC r" s c @ s e Zd ZdZdS )r bigintNrE rF rG r r@ r@ r@ rC r s c @ s e Zd ZdZdS )r5 smallintNr r@ r@ r@ rC r5 s c s e Zd ZdZ fddZ ZS )r1 r c s d|d< t t| j|| d S )NTr )r r1 r r r r@ rC r s zPrimaryKeyField.__init__)rE rF rG r r r r@ r@ r rC r1 s c s e Zd ZdZ fddZ ZS )_AutoPrimaryKeyFieldNc s4 || j kr tdt| |f tt| || d S )Nz%s must be named `%s`.) _column_namerw r r r r r r r@ rC r s
z!_AutoPrimaryKeyField.add_to_class)rE rF rG r r r r@ r@ r rC r s r c @ s e Zd ZdZeZdS )r floatN)rE rF rG r r ru r@ r@ r@ rC r s c @ s e Zd ZdZdS )r doubleNr r@ r@ r@ rC r s c sF e Zd ZdZd fdd Z fdd Zd
d Zd d
Zdd Z Z S )r decimal
r FNc s6 || _ || _|| _|p tjj| _tt| j|| d S r? )
max_digitsdecimal_places
auto_roundr ZDefaultContextroundingr r r )rA r r r r r r r r@ rC r s
zDecimalField.__init__c s( t t| jf | j| j| j| jd|S )N)r r r r )r r r1 r r r r r r r@ rC r1 s zDecimalField.clone_basec C s | j | jgS r? )r r r r@ r@ rC r s zDecimalField.get_modifiersc C sT t j}|s |d kr|S |dS | jrP|d| j }| j}|t|j||dS |S )Nr r )r )r Decimalr r r r Zquantize)rA r^ DZexpr r@ r@ rC r " s zDecimalField.db_valuec C s* |d k r&t |tjr|S tt|S d S r? )rQ r r r r r@ r@ rC r , s zDecimalField.python_value)r r FN)
rE rF rG r r r1 r r r r r@ r@ r rC r s
utf-8c C s* t | tr| S t | tr"| |S t| S r? )rQ unicode_type string_typedecode)rZ encodingr@ r@ rC coerce_to_unicode2 s
r c s> e Zd ZdZd fdd Z fddZdd Zd d
Z ZS ) r string c s || _ tt| j|| d S r? )
max_lengthr r r )rA r r r r r@ rC r < s zCharField.__init__c s t t| jf d| ji|S )Nr )r r r1 r r r r@ rC r1 @ s
zCharField.clone_basec C s | j r| j gpd S r? )r r r@ r@ rC r E s zCharField.get_modifiersc C s t |pdS N r r r@ r@ rC ru H s zCharField.coerce)r ) rE rF rG r r r1 r ru r r@ r@ r rC r 9 s
c s e Zd ZdZ fddZ ZS )r
fixed_charc s t t| |}|r | }|S r? )r r r stripr r r@ rC r N s zFixedCharField.python_value)rE rF rG r r r r@ r@ r rC r K s c @ s e Zd ZdZdd ZdS )r8 textc C s t |pdS r r r r@ r@ rC ru W s zTextField.coerceN)rE rF rG r ru r@ r@ r@ rC r8 T s c s4 e Zd ZdZeZ fddZdd Zdd Z Z S )r
blobc s0 t |jjtr |jj| j tt| ||S r? ) rQ rz r r3 r _set_constructorr r
r r r r@ rC r ^ s zBlobField.add_to_classc C s | | _d S r? )get_binary_type _constructor)rA r r@ r@ rC r c s zBlobField._set_constructorc C s, t |tr|d}t |tr(| |S |S rV )rQ r rX
basestringr
r r@ r@ rC r f s
zBlobField.db_value)
rE rF rG r binary_constructr
r r r r r@ r@ r rC r
Z s
c @ s e Zd ZdZdd Zdd ZdS )r< uuidc C s6 t |tjr|jS zt|jW S | Y S X d S r? )rQ r UUIDhexr r@ r@ rC r p s zUUIDField.db_valuec C s& t |tjr|S |d kr d S t|S r? )rQ r r r r@ r@ rC r x s zUUIDField.python_valueN)rE rF rG r r r r@ r@ r@ rC r< m s c s fdd}|S )Nc s | j jj | S r? )r} rz r extract_dater date_partr@ rC r~ ~ s z_date_part..decr@ )r r~ r@ r rC
_date_part} s r c s. e Zd ZdZd fdd Z fddZ ZS )_BaseFormattedFieldNc s$ |d k r|| _ tt| j|| d S r? )rx r r r )rA rx r r r r@ rC r s z _BaseFormattedField.__init__c s t t| jf d| ji|S )Nrx )r r r1 rx r r r@ rC r1 s
z _BaseFormattedField.clone_base)N)rE rF rG rx r r1 r r@ r@ r rC r s r c @ sj e Zd ZdZdddgZdd ZeedZeedZ eed Z
eed
Z eed Z eed Z
d
S )r ru r r r c C s |rt |trt|| jS |S r? )rQ r rp rx r r@ r@ rC r s zDateTimeField.python_valuer r r r r r N)rE rF rG r rx r r r r r r r r r r@ r@ r@ rC r s c @ sF e Zd ZdZdddgZdd ZeedZeedZ eed Z
d
S ) r dater r r c C s@ |r$t |tr$dd }t|| j|S |r.)rQ r rp rx ru r rA r^ Zppr@ r@ rC r s zDateField.python_valuer r r N) rE rF rG r rx r r r r r r r@ r@ r@ rC r s c @ sJ e Zd ZdZdddddgZdd Zeed Zeed
Z eed Z
d S )
r9 timer r r r r c C s` |r8t |tr$dd }t|| j|S t |tjr8| S |d k r\t |tjr\tjj| S |S )Nc S s | S r? )r rs r@ r@ rC rS rT z(TimeField.python_value..)rQ r rp rx ru r Z timedeltaminr r@ r@ rC r s
zTimeField.python_valuer r r N) rE rF rG r rx r r r r r r r@ r@ r@ rC r9 s c sF e Zd Zdd edD Z fddZdd Zdd Zd
d Z Z S ) r: c C s g | ] }d | qS )r r@ .0r r@ r@ rC
s zTimestampField. c s | ddpd| _| j| jkr s z*TimestampField.__init__..utcFr )popr valid_resolutionsrw r r" ru ZutcfromtimestampZ
fromtimestamp_convZutcnowZnow
setdefaultr r: r )rA r r Z_dtZ_defaultr r@ rC r s zTimestampField.__init__c C s | j dkr| jS tjS Nr )r r r r r@ r@ rC r s zTimestampField.get_db_fieldc C s |d kr d S t |tjrn4t |tjrr )rQ ru r r r r r roundr r" calendarZtimegmZ utctimetupler mktimeZ timetuple microsecond)rA r^ Z timestampr@ r@ rC r s
zTimestampField.db_valuec C sp |d k rlt |tttfrl|dkr$d S | jdkrbd| j }t|| j\}}|| }| |j|dS | |S |S )Nr r rf )r+ )rQ r r longr divmodr% replace)rA r^ Zticks_to_microsecondZticksrh r@ r@ rC r s
zTimestampField.python_value)
rE rF rG r r$ r r r r r r@ r@ r rC r: s
c @ s e Zd ZdZeZdS )r boolN)rE rF rG r r/ ru r@ r@ r@ rC r s c s: e Zd ZdZ fddZdd Zd ddZd d
Z ZS ) RelationDescriptorzEForeign-key abstraction to replace a related PK with a related model.c s || _ tt| | d S r? )r r r0 r )rA r r r r@ rC r s zRelationDescriptor.__init__c C sn |j | j}|d k s"| j|jkrZ| j|jkrN| j| jj|k}||j| j< |j| j S | jjsj| jj|S r? ) r r r
_obj_cacher r r r r )rA r# Zrel_idr r@ r@ rC get_object_or_id s z#RelationDescriptor.get_object_or_idNc C s |d k r| |S | jS r? )r2 r r" r@ r@ rC r$ s
zRelationDescriptor.__get__c C s| t || jr2t|| jjj|j| j< ||j| j< n8|j | j}||j| j< ||krj| j|jkrj|j| j= |j
| j d S r? ) rQ r r r r r r r r1 r r ry )rA r# r^ Z
orig_valuer@ r@ rC r " s
zRelationDescriptor.__set__)N) rE rF rG r r r2 r$ r r r@ r@ r rC r0 s
r0 c @ s" e Zd ZdZdd ZdddZdS )ReverseRelationDescriptorz r? r@ r r r r r@ rC r I s zForeignKeyField.__init__c s2 t t| jf | j| | j| j| j| jd|S )N)r rA r> r? r@ r ) r r r1 r _get_related_namer> r? r@ r r r r@ rC r1 Y s zForeignKeyField.clone_basec C s t | | jS r? )r0 r r r@ r@ rC _get_descriptorc s z ForeignKeyField._get_descriptorc C s t | S r? )r6 r r@ r@ rC _get_id_descriptorf s z"ForeignKeyField._get_id_descriptorc C s t | S r? )r3 r r@ r@ rC _get_backref_descriptori s z'ForeignKeyField._get_backref_descriptorc C s. | j rt| j r| | S | j p,d| jjj S )Nz%s_set)r< callabler} rz r r r@ r@ rC rB l s
z!ForeignKeyField._get_related_namec s t jtr, fdd}j| d S t jtrLj d S _ _jpfdj _jkrd7 j st
dd _ j
_jdkrj_jd k rt jtstjj_n jj
j_ j
jrt fdd }jjj
jkr.|d
njjj
jkrH|d j
jkr`|d n jkrt|d
t t tjj d_ j
j j< jj
jj< d S )Nc s | _ d S r? )r r r )r} r rA r@ rC r s s z.ForeignKeyField.add_to_class..callbackz%s_idZ_idr r rA c s. |j d jjf jd t| | d S )Nz%s.%s)r backref obj_id_name)r rz r rA r
)msgcontextr} r rI rA r@ rC invalid s z-ForeignKeyField.add_to_class..invalidzVThe related_name of %(field)s ("%(backref)s") conflicts with a field of the same name.zWThe related_name of %(field)s ("%(backref)s") is already in use by another foreign key.zRThe object id descriptor of %(field)s conflicts with a field named %(obj_id_name)sz]Model attribute "%(obj_id_name)s" would be shadowed by the object id descriptor of %(field)s.T) rQ r r3 r r r r r} r r r r r rz r rB rA r r r r validate_backrefsr r{ __dict__r+ rC rD rE r rel)rA r} r r rM r@ rL rC r q sR
z ForeignKeyField.add_to_classc s$ t | jts| j S tt| S )zr
Overridden to ensure Foreign Keys use same column type as the primary
key they point to.
)rQ r r1 r r r r r r@ rC r s
z ForeignKeyField.get_db_fieldc s$ t | jts| j S tt| S r? )rQ r r1 r r r r r r@ rC r s
z ForeignKeyField.get_modifiersc C s | j |S r? )r ru r r@ r@ rC ru s zForeignKeyField.coercec C s t || jr| }| j|S r? )rQ r
_get_pk_valuer r r r@ r@ rC r s zForeignKeyField.db_valuec C s t || jr|S | j|S r? )rQ r r r r r@ r@ rC r s z ForeignKeyField.python_value)NNNNN)rE rF rG r r1 rC rD rE rB r r r ru r r r r@ r@ r rC r H s
@ c @ sF e Zd ZdZdZdd Zdd ZdddZd d
Zd d Z d
d Z
dS )r z+A primary key composed of multiple columns.Nc G s
|| _ d S r? ) field_names)rA rR r@ r@ rC r s zCompositeKey.__init__c C s || _ || _t||| d S r? )r r} r+ r r@ r@ rC r s zCompositeKey.add_to_classc s$ d k r t fdd| jD S | S )Nc s g | ]}t |qS r@ )r r
field_namer# r@ rC r s z(CompositeKey.__get__..)tuplerR r" r@ rU rC r$ s
zCompositeKey.__get__c C s d S r? r@ r r@ r@ rC r s zCompositeKey.__set__c s& fddt j|D }ttj|S )Nc s" g | ]\}} j jj| |kqS r@ )r} rz r )r r r^ r r@ rC r s z'CompositeKey.__eq__..)ziprR rP operatorand_)rA other expressionsr@ r rC rF s
zCompositeKey.__eq__c C s t | jj| jfS r? )r r} rE rR r r@ r@ rC r s zCompositeKey.__hash__)N) rE rF rG r r r r r$ r rF r r@ r@ r@ rC r s
c @ sD e Zd ZdZdddZdd Zddd Zd
d Zd d
Zdd Z dS )AliasMaprk r c C s i | _ || _d S r? )
_alias_map_counter)rA startr@ r@ rC r s zAliasMap.__init__c C s
d| j S )Nzr] r r@ r@ rC __repr__ s zAliasMap.__repr__Nc C s: || j krd S | jd7 _|p.d| j| jf | j |< d S )Nr z%s%s)r] r^ prefix)rA r r6 r@ r@ rC ry s
z AliasMap.addc C s || j kr| | | j | S r? )r] ry rA r r@ r@ rC __getitem__ s
zAliasMap.__getitem__c C s
|| j kS r? r` rc r@ r@ rC __contains__ s zAliasMap.__contains__c C s. |r*|j D ]\}}|| kr|| j |< q| S r? )r] items)rA alias_mapr r6 r@ r@ rC r s
zAliasMap.update)r )N)
rE rF rG rb r ra ry rd re r r@ r@ r@ rC r\ s
r\ c 2 @ s e Zd Zddddddddd dd
d
dd d d
dZejdejdejdejdej dej
dej dej dej
dejdejdejdejdejd ejd ejd ejd ejd ejd!ejd"ejd#ejd$ejd%ej d&ej d'iZ e j d(e j!d)e j"d*e j#d+iZ$e%Z&dd/d0Z'd1d2 Z(d3d4 Z)d5d6 Z*d7d8 Z+d9d: Z,d;d< Z-d=d> Z.d?d@ Z/dAdB Z0dCdD Z1dEdF Z2dGdH Z3dIdJ Z4dKdL Z5dMdN Z6dOdP Z7dQdR Z8dSdT Z9ddUdVZ:ddXdYZ;ddZd[Zdd`daZ?dbdc Z@eAfdddeZBdfdg ZCdhdi ZDdjdk ZEdldm ZFdndo ZGddpdqZHeGdrZIddtduZJeGdvZKddwdxZLeGdyZMddzd{ZNeGd|ZOd}d~ ZPdd ZQeGdZRdddZSeGdZTdd ZUeGdZVdd ZWeGdZXd.S )
QueryCompilerr ZBIGINTZBLOBZSMALLINTZDATEZDATETIMEZDECIMALZREALZCHARINTEGERZVARCHARTEXTZTIME)r r r
r/ r ru r r r r r r r r r r r r r r r r r zNOT INr zIS NOTr r r r r r r r r #r r r r r z
INNER JOINzLEFT OUTER JOINzRIGHT OUTER JOINz FULL JOIN"?Nc C sJ || _ || _t| j|pi | _t| j|p*i | _| | _t dg| _
d S )Nr )
quote_char
interpolationr field_map
_field_mapop_map_op_map
get_parse_map
_parse_mapr _unknown_types)rA rn ro field_overrides op_overridesr@ r@ rC r H s
zQueryCompiler.__init__c C s2 | j | j| j| j| j| j| j| j| j| j | j
d S )N) rO r r r r rl r ri select_querycompound_select_queryrr ) _parse_expression _parse_param_parse_passthrough _parse_func
_parse_clause
_parse_entity _parse_field
_parse_sql_parse_select_query _parse_compound_select_query_parse_strip_parensr r@ r@ rC rt Q s zQueryCompiler.get_parse_mapc C s d| j || j f S )Nz%s%s%s)rn )rA rZ r@ r@ rC quoteb s zQueryCompiler.quotec C s || j kr| j | S | S r? )rq upper)rA fr@ r@ rC r e s z QueryCompiler.get_column_typec C s
| j | S r? )rs )rA qr@ r@ rC get_oph s zQueryCompiler.get_opc C s t | dd dS )Nc S s
| d j S Nr )r )r r@ r@ rC rS l rT z.QueryCompiler._sorted_fields..r )r rf )rA
field_dictr@ r@ rC _sorted_fieldsk s z QueryCompiler._sorted_fieldsc C s | j |gfS r? )ro rA r& rg convr@ r@ rC _parse_defaultn s z QueryCompiler._parse_defaultc
C s t |jtr|j}| |j||\}}| |j||\}}|jtjkrd|dkrd|sd|jr\dndg fS |jrndnd}||| |j|f } | || fS )Nz()z0 = 1z(0 = 1)%s %s %s
(%s %s %s))
rQ r r
parse_noder@ rC rE r r r )
rA r& rg r r Zlparamsr@ Zrparamstemplateri r@ r@ rC r{ q s z QueryCompiler._parse_expressionc C s* |j r | | |j|d S | j|jgfS r? )r r r^ ro r r@ r@ rC r} | s z QueryCompiler._parse_passthroughc C s^ |j r0|r|j|j krd }| | |j||S |d k rL| ||j|S | j|jgfS d S r? )r r r r^ ro r r@ r@ rC r| s zQueryCompiler._parse_paramc C s8 |j r
|p d }| |j||\}}d|jt|f |fS )Nr )rt parse_node_listrr r rr rA r& rg r ri rj r@ r@ rC r~ s zQueryCompiler._parse_funcc C s2 | |j|||j\}}|jr*dt| }||fS N(%s))r r r r rr r r@ r@ rC r s zQueryCompiler._parse_clausec C s d t| j|jg fS r )r r r rm r r@ r@ rC r s zQueryCompiler._parse_entityc C s |j t|jfS r? )r^ r rj r r@ r@ rC r s zQueryCompiler._parse_sqlc C s< |r(d | ||j | |jf}n | |j}|g fS r )r r r} r )rA r& rg r ri r@ r@ rC r s
zQueryCompiler._parse_fieldc C s d}|j j|kr0|jj|kr0|j |j }}d}n|j|j }}d}| }|j|kr\|j|_| ||\} }
| || ||\} } |r| | | |
f\}
}}}n| |
| | f\}
}}}|jjrd|
|j |f }nd|
|j |f }||| fS )Nrz TFz((%s) %s (%s))r )
r@ rQ r alias_map_classr^ generate_selectcalculate_alias_mapr compound_select_parenthesesrX )rA r& rg r ZcsqZfirst_qZsecond_qrB new_mapfirstZfirst_pr Zsecond_pr ZlprZrpri r@ r@ rC r s*
z*QueryCompiler._parse_compound_select_queryc C sV | }|js6|r$t|tr$|j}n
|jjj}|f|_| ||\}}dt
| |fS r ) r _explicit_selectionrQ r r r} rz r _selectr rr )rA r& rg r r Z select_fieldr rj r@ r@ rC r s
z!QueryCompiler._parse_select_queryc C s | |j||\}}t||fS r? )r r& rr r r@ r@ rC r s z!QueryCompiler._parse_strip_parensc C sh t |dd }d}|| jkrP| j| |||\}}|| jkoJ|jd koJ|d k}nt|tttfr|| |||\}}d| }nt|t r| j
}|rt|t r|j }t|t r|
|} q|
t ||j} n| } | g}nt|rt|t st|tr | || }
| |
||\}}n@|d k rH|
|} | | |d \}}} n| ||d \}}d}|||fS )NrQ Fr T)r ru rv r rQ r rV r r r* ro r r r r rQ r
r: r r r6 r _parser ) rA r& rg r Z node_typer ri rj r r^ rl _r@ r@ rC r s@
zQueryCompiler._parsec
s | || \}}}|r6 d k r6|r6 fdd|D }t|tr||jrNd| }|jrfd|d|jf}|jr|d||jf}|rtdd |D rg }g }t|D ] \} }
t|
tr| |
\} } q||fS )Nc s g | ]} |qS r@ )r r r r@ rC r s z,QueryCompiler.parse_node..zNOT %sr ASc s s | ]}t |tV qd S r? )rQ r% )r pr@ r@ rC r! s z+QueryCompiler.parse_node..)
r rQ r% r( r! r r* any enumerater )
rA r& rg r ri rj r Z clean_paramsZ clean_sqlidxr ZcsqlZcparamsr@ r rC r s"
zQueryCompiler.parse_noder c
C sF g }g }|D ]*}| |||\}} || || q |||fS r? )r r~ r0 r )
rA r rg r r ri rj r& Znode_sqlZ node_paramsr@ r@ rC r s
z QueryCompiler.parse_node_listc C s | }|d k r|j|_||j|jjj |j D ]V\}}|||jj |D ]8}t|j t
rv||j |j j qR||j |j jj qRq6| |S r? )
r r^ ry r} rz table_alias_joinsrf rQ r r% r6 r )rA r rg r r Z
joined_modelsZjoin_objr@ r@ rC r s z!QueryCompiler.calculate_alias_mapc C s | t| |S r? )r r )rA clausesrg r@ r@ rC build_query* s zQueryCompiler.build_queryc C sZ g }t }|g}|rV| }||ks||kr0q|| || D ]}|} |j}
t|jtttt frv|j
} nl|j } | j
r|j}
|j}n |j}
|j}| j}|rt|
|j}t||jj}| j
r|| }}||k} ntdt|
tr|
}n ||
|
||
}| }|| jkr2t| j| }nt|}|t||td| qBq|S )Nz Missing required join predicate.ON)r r# ry r rQ r r> ro r r r r6 r r r r r r r rw r% r~ r r join_mapr7 )rA joinsr} rg r r r currr r r
constraintr Zfk_modelZpk_modelfkr r@ Zdest_nr Zjoin_sqlr@ r@ rC generate_joins- sN
z QueryCompiler.generate_joinsc C s. |j }|jj}| ||}t|tr0t|g}n|jsBtdg}n*tdg}|jdkrl|tdt |j g7 }t
|j }d|_ |
|tdf |jd kr|| || n|t|j |jd k r|td |tdd |jD | |j||}|r|
| |jd k r4|
td
|jg |jrT|
td t|j g |jrp|
td |jg |jr|
td
t|j g |jd k s|jr|j r|jd k r|jn|j }|td| |jd k r|td|j |j \} }
| r"|
rdnd} |t| | ||S )NZSELECTzSELECT DISTINCT)TFr r FROMZWINDOWc S s& g | ] }t t|jtd | qS )r )r r7 r! ry )r rz r@ r@ rC r } s z1QueryCompiler.generate_select..WHEREzGROUP BYZHAVINGr zLIMIT %sz OFFSET %szFOR UPDATE NOWAITz
FOR UPDATE) r} rz r r rQ CompoundSelectr _distinctr7 r r r r r0 _fromr~ r r6 r _windowsr r _where _group_by_having _order_by_limit_offset limit_max _for_updater ) rA r rg r dbr Z
select_clauseZ join_clauseslimit
for_updateZno_waitZstmtr@ r@ rC r b sT
z QueryCompiler.generate_selectc
C s |j }| }|||jj |jr0d|j }nd}t|| tdg}g }| |j D ]D\}}t
|t t fs~t
||jd}|t|jddtj|dd qZ|t| |jr|td |jg |jd k rt|j } d
| _|td | g | ||S ) Nz UPDATE OR %sZUPDATEZSETr Fr T)r r r RETURNING)r} r ry rz r _on_conflictr7 r r _updaterQ r% r* r. r r~ r> rE r r r r0
_returningr r r )
rA r r rg statementr r r r^ returning_clauser@ r@ rC generate_update s2
z QueryCompiler.generate_updatec C s |dd |D S )Nc S s g | ]}|j d dqS )Fr r r r r@ r@ rC r s z3QueryCompiler._get_field_clause..r@ )rA r clause_typer@ r@ rC _get_field_clause s z QueryCompiler._get_field_clausec C s |j }|j}| }|||jj |jr4|jj}n|jrFd|j }nd}t ||
g}|j d k r|j r||
| |j |
t|j n|jd k rfg g }}d} | D ]n}
| st|
tdd}d} g } |D ]4} |
| }
t|
ttfst|
| jd}
|
|
q|
t| q|rF| | |t dt | g n |j jj rf|
|j |j |j r| t d | j|! t d
g n.|j"d k rt#|j" }d |_$| t d |g | %||S ) NzINSERT OR %s INTOz INSERT INTOFr r Tr ZVALUESr )r r )&r} rz r ry r _upsertr
upsert_sqlr r7 r _query_fieldsr~ r r _rows
_iter_rowsr keysrX
attrgetterrQ r% r* r. r r r0 r auto_incrementdefault_insert_clauseis_insert_returningget_primary_key_fieldsr r r r )rA r r rI rg r r r Z
value_clausesZ have_fieldsrow_dictr| r r^ r r@ r@ rC generate_insert sl
z QueryCompiler.generate_insertc C sf |j }td| g}|jr0|td|jg |jd k r\t|j }d|_|td|g | |S )Nz DELETE FROMr r r ) r} r7 r r r0 r r r r )rA r r r r r@ r@ rC generate_delete s
z QueryCompiler.generate_deletec C s | | }||}t| S r? )r r r r )rA r r r r@ r@ rC field_definition
s
z QueryCompiler.field_definitionc C sj t dt| t d|j t|j g}|jrH|t d|j |jrb|t d|j t| S )Nz FOREIGN KEYZ
REFERENCESz ON DELETE %sz ON UPDATE %s) r7 r r r r r> r~ r? r )rA r r r@ r@ rC foreign_key_constraint s
z$QueryCompiler.foreign_key_constraintc s fdd}|S )Nc s t | }| |||S r? )r r )rA r r r
function_namer@ rC r " s
z/QueryCompiler.return_parsed_node..innerr@ )r r r@ r rC return_parsed_node s z QueryCompiler.return_parsed_nodec C sN |p d|j j|j|jj jf }| |}ttd| tdt|f|j S )Nzfk_%s_%s_refs_%sz ALTER TABLEzADD CONSTRAINT)
rz r r r r r r7 r r r )rA r} r r Z fk_clauser@ r@ rC _create_foreign_key' s
z!QueryCompiler._create_foreign_keyr Fc s |rdnd}|j g g }} jrP fdd jjD }|ttdt| jD ]4}|| | t
|t rV|j sV||
| qV|j jr|j jD ] }t
|tst|}|| qtt|| t|| S )NzCREATE TABLE IF NOT EXISTSz CREATE TABLEc s g | ]} j | qS r@ )r r r r rI r@ rC r ; s z/QueryCompiler._create_table..r )rz
composite_keyr rR r~ r r7 r declared_fieldsr rQ r r= r r r% r ) rA r} safer columnsr Zpk_colsr r r@ r rC
_create_table5 s2
zQueryCompiler._create_tabler c C s6 |rdnd}t || g}|r.|t d t| S )NzDROP TABLE IF EXISTSz
DROP TABLECASCADEr7 r r~ r )rA r}
fail_silentlycascader r r@ r@ rC _drop_tableP s
zQueryCompiler._drop_tabler c C s< t d| g}|r"|t d |r4|t d t| S )NzTRUNCATE TABLEzRESTART IDENTITYr r )rA r} restart_identityr r r@ r@ rC _truncate_tableX s z QueryCompiler._truncate_tabler c C sR d|d |f }t|dkrNt|d }d|d d |d d f }|S )Nz%s_%sr @ r 7 )r r hashlibZmd5rX Z hexdigest)rA tabler r Z
index_hashr@ r@ rC
index_nameb s
zQueryCompiler.index_namec G s^ |j j}|rdnd}| |dd |D }tt|t|td| tdd |D f| S )NzCREATE UNIQUE INDEXz CREATE INDEXc S s g | ]
}|j qS r@ r r r@ r@ rC r l s z/QueryCompiler._create_index..r c S s g | ] }| qS r@ r r r@ r@ rC r r s )rz r r r r7 r r r )rA r} r r r@ tbl_namer r r@ r@ rC
_create_indexi s zQueryCompiler._create_indexr c C s< |j j}|rdnd}| |dd |D }tt|t|S )NzDROP INDEX IF EXISTSz
DROP INDEXc S s g | ]
}|j qS r@ r r r@ r@ rC r y s z-QueryCompiler._drop_index..)rz r r r r7 r )rA r} r r r r r r@ r@ rC _drop_indexv s zQueryCompiler._drop_indexr c C s t tdt|S )NzCREATE SEQUENCEr r7 r rA Z
sequence_namer@ r@ rC _create_sequence} s z QueryCompiler._create_sequencer c C s t tdt|S )Nz
DROP SEQUENCEr r r@ r@ rC _drop_sequence s z QueryCompiler._drop_sequencer )rl rm NN)NN)Nr )N)N)N)N)F)FF)FF)F)YrE rF rG rp rE r r r r r r r r r r r r r r r r r r r r r r r r r rr r& r r r r r r\ r r rt r r r r r r{ r} r| r~ r r r r r r r r r r r r r r r r r r r r r r r create_foreign_keyr create_tabler
drop_tabler truncate_tabler r create_indexr
drop_indexr create_sequencer
drop_sequencer@ r@ r@ rC rh s
'
5
? >
rh c @ s e Zd ZdddZdS )SqliteQueryCompilerFc C s | S r? )deleteri )rA r} r r r@ r@ rC r s z"SqliteQueryCompiler.truncate_tableN)FF)rE rF rG r r@ r@ r@ rC r
s r
c @ s e Zd Zdd Zdd ZeZdS )ResultIteratorc C s || _ d| _d S r )qrw_idx)rA r r@ r@ rC r s zResultIterator.__init__c C sf | j | jjk r | jj| j }n6| jjsP| j }| jj| | j jd7 _nt| j d7 _ |S r' )r r _ct
_result_cache
_populatediterater~
StopIterationrc r@ r@ rC next s
zResultIterator.nextN)rE rF rG r r __next__r@ r@ r@ rC r s r c @ sd e Zd ZdZdddZdd Zedd Zd d
Zd d Z d
d Z
dd Z dd Z e Z
dddZdS )QueryResultWrapperz
Provides an iterator over the results of a raw Query, additionally doing
two things:
- converts rows from the database into python representations
- ensures that multiple iterations do not result in multiple queries
Nc C sP || _ || _d| _d| _g | _d| _d| _|d k r@|\| _| _n d | _| _d S )Nr F) r cursorr r r r _initialized column_meta join_meta)rA r r rI r@ r@ rC r s zQueryResultWrapper.__init__c C s | j rt| jS t| S d S r? )r iterr r r r@ r@ rC __iter__ s
zQueryResultWrapper.__iter__c C s | | jS r? )
fill_cacher r r@ r@ rC rp s zQueryResultWrapper.countc C s | j S r? )rp r r@ r@ rC __len__ s zQueryResultWrapper.__len__c C s |S r? r@ rA rowr@ r@ rC process_row s z QueryResultWrapper.process_rowc C sV | j }|s2d| _t| j dd s,| j tn| jsL| | j j d| _| |S )NTr )
r fetchoner r closer r r descriptionr$ r" r@ r@ rC r s
zQueryResultWrapper.iteratec c s | V q d S r? )r r r@ r@ rC iterator s zQueryResultWrapper.iteratorc C sh | j | jk r*| j| j }| j d7 _ |S | jr4t| }| j| | jd7 _| j d7 _ |S r' )r r r r r r r~ )rA r2 r r@ r@ rC r s zQueryResultWrapper.nextc C s` |p
t d}|dk r td| j| _| js\|| jkr\z t| W q$ tk
rX Y q\Y q$X q$d S )NZInfr z"Negative values are not supported.)r rw r r r r r rA nr@ r@ rC r s z QueryResultWrapper.fill_cache)N)N)rE rF rG r r r r rp r! r$ r r( r r r r@ r@ r@ rC r s
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ExtQueryResultWrapperc C s t |}g | _}| jd k rht | j}t| jD ](\}}| ||s0| || d | q0||krld S nd}t||D ]}| || d | qvd S r )r r r r _initialize_node_initialize_by_namer )rA r' Zn_colsr Zn_metar r& r@ r@ rC r s
z ExtQueryResultWrapper.initializec C sH | j jj}||kr2|| }| j||j|jf n| j||d f d S r? )r rz r r r~ r r )rA r r Z
model_colsr r@ r@ rC r- s
z)ExtQueryResultWrapper._initialize_by_namec C s t |tr*| j||jp |j|jf dS t |trt|j r|j d }t |tr|jpb|jpb|j}|j
rp|jprd }| j|||f dS dS )NTr F) rQ r r r~ r! r r ro r rr rt )rA r& r argr r r@ r@ rC r,
s
z&ExtQueryResultWrapper._initialize_nodeN)rE rF rG r r- r, r@ r@ r@ rC r+ s r+ c @ s e Zd Zdd ZdS )TuplesQueryResultWrapperc s t fddt|D S )Nc s8 g | ]0\}} j | d dkr"|n j | d |qS )rL Nr )r r colr r@ rC r s z8TuplesQueryResultWrapper.process_row..)rV r r" r@ r rC r$ s z$TuplesQueryResultWrapper.process_rowNrE rF rG r$ r@ r@ r@ rC r/ s r/ c @ s e Zd Zdd ZdS )NaiveQueryResultWrapperc C sJ | }| jD ].\}}}t|||d k r2||| n|| q| |S r? )r r r+ _prepare_instance)rA r# r# r columnr r@ r@ rC r$ ! s
&z#NaiveQueryResultWrapper.process_rowNr1 r@ r@ r@ rC r2 s r2 c @ s e Zd Zdd ZdS )DictQueryResultWrapperc C s: i }| j D ]*\}}}|d k r(||| n|| ||< q
|S r? r )rA r# rv r r4 r r@ r@ rC r$ , s "z"DictQueryResultWrapper.process_rowNr1 r@ r@ r@ rC r5 + s r5 c @ s> e Zd Zdd Zdd Zdd Zdd Zdd
d Zd d
Zd S )ModelQueryResultWrapperc C s4 | \| _}tdd | jD | _| || _d S )Nc s s | ]}t |tr|V qd S r? )rQ r )r r0 r@ r@ rC r! 8 s
z5ModelQueryResultWrapper.initialize..)generate_column_map
column_mapr r _col_setgenerate_join_list join_list)rA r' Z model_setr@ r@ rC r 6 s z"ModelQueryResultWrapper.initializec C s g }t | jg}t| jD ]\}}d }}t|trrt|trT|j}|j}|jj }n|j
}}|j }|j pn|j }nL|j
d kr| j }}n
|j
}}t|tr|j r|j }nt|tr|jd }|||||f || q||fS N)r r r r rQ r r _model_aliasfield_instancer r} r! r r) r% r rm r~ ry ) rA r8 r r r& r r r constructorr@ r@ rC r7 < s,
z+ModelQueryResultWrapper.generate_column_mapc C s g }| j }| jg}|r| }||kr(q|| D ]|}|j}|j|ksN|j|kr0|jd k r|j| jk}|j| jk} |jj o~|p~| }
n d }
}} |
||
|| f |
|j q0q|S r' ) r r r# r r r r r9 r r r~ ) rA r r; r stackcurrentr r
fk_present
pk_presentZcheckr@ r@ rC r: X s2
z*ModelQueryResultWrapper.generate_join_listc C s. | |}| |}|D ] }| q|d S r )construct_instances follow_joinsr3 )rA r# collected instancesr r@ r@ rC r$ u s
z#ModelQueryResultWrapper.process_rowNc C s i }t | jD ]x\}\}}}}|d k r0||kr0q|| } ||krJ| ||< || }
|d krj| jj| d }t|
||d kr|| n|| q|S r )r r8 r r' r+ ) rA r# r Zcollected_modelsr r r@ r r r^ r# r@ r@ rC rE | s
z+ModelQueryResultWrapper.construct_instancesc
C sN || j g}| jD ]4\}}}}||j }z||j }W n tk
rV ||j }Y nX d} |r|rv|j|jj } n|r|j|j
j } | sq|j
d k }
|
o|j |jkot ||j
j d k} | rt
||j
j |j|j |jr0|
o|jd k ot ||j
j d k o|j|jj d k} | r0t
||jj | t
||j | || q|S NT)r r; r r KeyErrorr r r r r r r r r+ r r~ )
rA rG preparedr Z
check_nullrC rD r2 joined_instZhas_fkZmpkZcan_populate_joined_pkZcan_populate_joined_fkr@ r@ rC rF sX
z$ModelQueryResultWrapper.follow_joins)N) rE rF rG r r7 r: r$ rE rF r@ r@ r@ rC r6 5 s
r6 JoinCacher r c s8 e Zd Z fddZ fddZdd Zdd Z ZS ) AggregateQueryResultWrapperc s g | _ tt| j|| d S r? )_rowr rO r r r r@ rC r s z$AggregateQueryResultWrapper.__init__c s t t| | t | _| jD ]\}}}}| j| q t | _i | _i | _ i | _
| j D ]\}}}}|j rt|j
j}n|j}|j p|j}|r| j|j n&| j
|jt | j
|j |j | j |ji t||jp|d| j |j |j< qXi | _i }t| jD ]^\}\}} }
}|| jkrD| j|g | j| ||
f ||g || ||
f q| jD ]D} | | jkrql| j
| d} | D ]}
| j| ||
qqld S )NrN r@ )r rO r r
all_modelsr8 ry Zmodels_with_aggregateZback_referencessource_to_destZdest_to_sourcer; r r rA r r r r& r rM r6 columns_to_comparer r~ r r0 )rA r' r r r r r Zkey_to_columnsr r} col_namer ZsourcesZ joined_modelr r@ rC r sL
z&AggregateQueryResultWrapper.initializec C sD i }| j D ]0\}}g ||< |D ]\}}|| || q"q|S r? )rS rf r~ )rA r# r r} Z column_datar rT r@ r@ rC read_model_data s z+AggregateQueryResultWrapper.read_model_datac
C s8 | j r| j }n
| j }|sDd| _t| jdd s>| j tn| js^| | jj
d| _dd }i }| |}|| j }|
D ]"\}}t ||< ||| ||< q| |}| j } | d krʐq~t }
| | } |
D ] \} }
|| |
kr|
| q|
s| j | q~| j|
}| | |}|
D ]D\}}d}|j D ]}|d k rLd}qL|s6||| ||< q6q| j g}|g}|r | }|| jkrq| j| D ]d}z| j| |j \}}W n tk
r Y qY nX |js |jr|| D ]}t ||g q |j|kr0q||j
D ]j\}}|d krTq>z|| |j|j j }W n tk
r Y q>Y nX t||| || q>n`|r |j|krq||
D ]<\}}||j |j|j j }t ||j j | || q||j qq|D ]}| q$|S )NTr c s, j jr$t fdd j jjD S S )Nc s g | ]} j | qS r@ )r rS rU r@ rC r
s zHAggregateQueryResultWrapper.iterate.._get_pk..)rz r rV r rR rQ rU r@ rU rC _get_pk
s
z4AggregateQueryResultWrapper.iterate.._get_pkF) rP r# r r% r r r& r r r r' rE r rf r rU r ry r~ rQ r r| r rR r rJ r r r+ r r r3 )rA r# rV Z identity_mapZ _constructedZprimary_instancer r# Z
model_dataZcur_rowZduplicate_modelsZ cur_row_datar} dataZdifferent_modelsZ
new_instancesZall_noner^ rA rH rB r r r pkr2 rL r@ r@ rC r
s
z#AggregateQueryResultWrapper.iterate)rE rF rG r r rU r r r@ r@ r rC rO s 6rO c s e Zd ZdZdZ fddZdd Zdd Zd d
Zd d Z d+ddZ
dd Z e dd Z
e dd Ze d,ddZe d-ddZd.ddZd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd/d)d*Z ZS )0Queryz?Base class representing a database query on one or more tables.Tc s@ t t| || _|jj| _d| _|| _| jg i| _d | _ d S rI )
r rY r r} rz r r
_query_ctxr r rA r} r r@ rC r v
s
zQuery.__init__c C s | \}}d| j||f S )Nr )ri r} rA ri rj r@ r@ rC ra
s zQuery.__repr__c C s t | | j}| j|_| |S r? )r r} r _clone_attributesr r@ r@ rC r
s z Query.clonec C s, | j d k r| j |_ | |_| j|_|S r? )r r _clone_joinsr rZ r r@ r@ rC r]
s
zQuery._clone_attributesc C s t dd | j D S )Nc s s | ]\}}|t |fV qd S r? )r )r Zmcjr@ r@ rC r!
s z%Query._clone_joins..)dictr rf r r@ r@ rC r^
s zQuery._clone_joinsNc C s, t tj|}|d kr|S |p tj}|||S r? )rP rX rY )rA initialr[ Z conjunctionZreducedr@ r@ rC _add_query_clauses
s
zQuery._add_query_clausesc C sx g }|D ]j}t |tr"|| qt |tr8|| qt |trR|| qt|rt|t r||j
j q|S r? ) rQ r% r~ rY r r0 get_proxy_fieldsr
r: r* rz r )rA r accumr. r@ r@ rC _model_shorthand
s
zQuery._model_shorthandc G s | | j|| _d S r? )rb r rA r[ r@ r@ rC r5
s z Query.wherec G s | | j|tj| _d S r? )rb r rX or_rf r@ r@ rC orwhere
s
z
Query.orwherec C s | j }|s8t|tp(t|o(|j| }|rNtdnt|trN|jj| }| j
|g | j | t |||| t|ts|| _ d S )Nz#A join condition must be specified.)
rZ rQ r r
rz
rel_existsrw r r r r& r~ r )rA r r r r Zrequire_join_conditionr@ r@ rC r
s
z
Query.joinc C s |p| j | _dS )z"Change or reset the query context.N)r} rZ r[ r@ r@ rC switch
s z Query.switchc K sN | j }| j|g D ]}|j|kr| S q| |j|fd|i||S )Nr )rZ r r r rj r )rA lmrmr Z join_kwargsZctxr r@ r@ rC ensure_join
s
zQuery.ensure_joinc C s g }g }t tf}t| D ]\}}| j}d|kr`|ddd tkr`|dd\}}t| }ntj}| dD ](} t
|| }
t |
|rp|
j }|
|
qp|
t|
|| q ||fS )N__r )r r3 r rf r} rsplit
DJANGO_MAPrE r splitr rQ r r~ r> ) rA Zqdictrd r Z relationshipr r^ r rC piece
model_attrr@ r@ rC convert_dict_to_node
s
zQuery.convert_dict_to_nodec O sH t }|r$|ttjdd |D M }|r6|tf |M }t|g}t }|r| }t|t s^qFd|j
fd|j ffD ]`\}}t|tr| |j
\} }
||
ttj| } |j| _|j| _t||| qr|| qrqF|j }| } |D ]T} t| tr | j| j }
}| }n"t| tr.| jj| j }
}| j}| |
||} q| |S )Nc S s g | ] }| qS r@ r )r r5 r@ r@ rC r
s z Query.filter..r r@ )r% rP rX rY r r r popleftrQ r> r r@ rt r r r( r! r+ r~ r r r} r r3 r rm r5 )rA r r Zdq_noder Zdq_joinsr Zsiderr r r rO r rk rl field_objr@ r@ rC rn
s>
z Query.filterc C s
| j S r? )r r r r@ r@ rC r s zQuery.compilerc C s t d S r? r r r@ r@ rC ri s z Query.sqlc C s | \}}| j||| jS r? )ri r execute_sqlrequire_commitr\ r@ r@ rC _execute s zQuery._executec C s t d S r? rw r r@ r@ rC execute s z
Query.executeFc C s6 |r| }n | }|r.|s.|d S |S d S r )tuplesr rz r% )rA Zas_tupleconvertr# r@ r@ rC scalar s z Query.scalar)N)NN)N)N)FF)rE rF rG r ry r ra r r] r^ rb re r r5 rh r rj rm rt rn r ri rz r{ r~ r r@ r@ r rC rY r
s2
(rY c st e Zd ZdZ fddZdd ZedZedZedZ e
dd
d Z e
dd d
Z dd Z
dd Zdd Z ZS )RawQueryzh
Execute a SQL query, returning a standard iterable interface that returns
model instances.
c s6 || _ t|| _d | _d| _d| _tt| | d S r' ) _sqlr _params_qr_tuples_dictsr r r )rA r r rj r r@ rC r % s
zRawQuery.__init__c C s* t | j| jf| j }| j|_| j|_|S r? )r r} r r r r r r@ r@ rC r - s zRawQuery.clonejoiningr5 rj Tc C s
|| _ d S r? r rA r| r@ r@ rC r| 7 s zRawQuery.tuplesc C s
|| _ d S r? r rA dictsr@ r@ rC r ; s zRawQuery.dictsc C s | j | jfS r? )r r r r@ r@ rC ri ? s z RawQuery.sqlc C sX | j d krR| jr | jt}n | jr2| jt}n | jt}|| j| d | _ | j S r? )
r r r get_result_wrapperRESULTS_TUPLESr
RESULTS_DICTS
RESULTS_NAIVEr} rz )rA ZQRWr@ r@ rC r{ B s
zRawQuery.executec C s t | S r? r r{ r r@ r@ rC r M s zRawQuery.__iter__)T)T)rE rF rG r r r r r r5 rj r r| r ri r{ r r r@ r@ r rC r s r c K s8 | dd}|rtd|r,| p"g | p*d S |r4|S d S )Nr0 Fz,"extend" is the only valid keyword argument.)r# rw )ZorigZnew_valr r0 r@ r@ rC allow_extendP s r c s e Zd ZdZ fddZ fddZdd ZeeZedZ ed Z
ed
Z d d Z d
d Z
dd ZeeZedd Zedd Zedd Zedd Zedd Zedd Zed d ZedZd d!Zed[d#d$Zed\d&d'Zed]d(d)Zed^d*d+Z ed_d,d-Z ed`d.d/Z edad1d2Z dbd3d4Z dcd5d6Z!ddd7d8Z"ded9d:Z#dfd;d<Z$d=d> Z%d?d@ Z&dgdBdCZ'dhdDdEZ(dFdG Z)dHdI Z*dJdK Z+dLdM Z,dNdO Z-dPdQ Z.dRdS Z/dTdU Z0dVdW Z1e2rdXdY Z3 Z4S )ir ry c s t t| | | jj| _| j| d | _d | _d | _ d | _
d | _ d | _ d | _
d| _d| _d| _d| _d| _d| _d | _d | _d S )NF)FF)r r r r
commit_selectry _SelectQuery__selectr r r r r r r r r _naiver r _aggregate_rowsr! r )rA r} selectionr r@ rC r \ s$
zSelectQuery.__init__c s t t| |}| j|_t| j|_| jd k rhg |_| jD ],}t|trZ|j |
q:|j | q:| j d k r~t| j |_ | j r| j
|_ | j
d k rt| j
|_
| jd k rt| j|_| j|_| j|_| j|_| j|_| j|_| j|_| j|_| j|_| j|_|S r? )r r r] r r r r rQ r% r~ r r r r r r r r r r r r r r! )rA r r r r@ rC r] p s6
z SelectQuery._clone_attributesc s fdd}|S )Nc s0 | j jjj} |kr td t| j | |S )Nz!Your database does not support %s)r} rz r compound_operationsrw r )rA rZ Z
supported_opsrX r@ rC r s z&SelectQuery.compound_op..innerr@ )rX r r@ r rC compound_op s zSelectQuery.compound_opUNION INTERSECTEXCEPTc C s4 | j tdt| |@ d }| |B | S )Nr r )r} r4 r7 from_r r6 rx )rA r@ Z wrapped_rhsr@ r@ rC rY s zSelectQuery.__xor__c C s t d| |S )N UNION ALL)r _compound_op_staticr? r@ r@ rC union_all s zSelectQuery.union_allc G s, t |dk| _|p| jjj}| || _d S r )r r r} rz r re r rA r r@ r@ rC Z__select s zSelectQuery.__selectc G s |r t |nd | _d S r? )r r )rA r r@ r@ rC r s zSelectQuery.from_c O s |r| |nd | _d S r? )re r r r@ r@ rC group_by s zSelectQuery.group_byc G s | | j|| _d S r? )rb r rf r@ r@ rC having s zSelectQuery.havingc O s t | jt|f|| _d S r? )r r r r r@ r@ rC rx s zSelectQuery.order_byc O s t | jt|f|| _d S r? )r r r )rA Zwindowsr r@ r@ rC rz s zSelectQuery.windowc C s
|| _ d S r? )r )rA Zlimr@ r@ rC r s zSelectQuery.limitc C s
|| _ d S r? )r )rA Zoffr@ r@ rC offset s zSelectQuery.offset c C s$ |dkr|d8 }|| _ || | _d S Nr r )r r )rA ZpageZ paginate_byr@ r@ rC paginate s zSelectQuery.paginateTc C s
|| _ d S r? )r )rA Z is_distinctr@ r@ rC distinct s zSelectQuery.distinctFc C s ||f| _ d S r? )r )rA r Znowaitr@ r@ rC r s zSelectQuery.for_updatec C s
|| _ d S r? )r )rA naiver@ r@ rC r s zSelectQuery.naivec C s
|| _ d S r? r r r@ r@ rC r| s zSelectQuery.tuplesc C s
|| _ d S r? r r r@ r@ rC r s zSelectQuery.dictsc C s
|| _ d S r? )r )rA aggregate_rowsr@ r@ rC r s zSelectQuery.aggregate_rowsNc C s
|| _ d S r? r4 )rA r6 r@ r@ rC r6 s zSelectQuery.aliasc C sx |d kr t |jjd}| j|kr4| | j}n| }| |j|}|j
sbdd |j D |_
t |j |f |_ |S )Nrp c S s g | ] }| qS r@ )r6 )r rt r@ r@ rC r s z(SelectQuery.annotate..)
r Countrz r r6 rZ rj r} r rm r r rV )rA r Z
annotationr r@ r@ rC annotate s
zSelectQuery.annotatec C s* |d krt td}| }|g|_|S )Nr )r r r7 rx r )rA aggregationr r@ r@ rC
_aggregate s
zSelectQuery._aggregatec C s | |j|dS )Nr} )r r~ )rA r r} r@ r@ rC aggregate s zSelectQuery.aggregatec C s4 | j s| js| js| jr$| j|dS | jddp2dS )N clear_limitFr r )r r r r
wrapped_countr rA r r@ r@ rC rp s zSelectQuery.countc C sJ | }|rd |_|_| \}}d| }| jj|f| }| pHdS )Nz+SELECT COUNT(1) FROM (%s) AS wrapped_selectr )rx r r ri r} rawr~ )rA r r ri rj wrappedZrqr@ r@ rC r s zSelectQuery.wrapped_countc C s$ | dd}tdg|_t| S )Nr 1)r r7 r r/ r~ r r@ r@ rC exists s zSelectQuery.existsc C sH | dd}zt| W S tk
rB | jd| Y nX d S )Nr z:Instance matching query does not exist:
SQL: %s
PARAMS: %s)r r r{ r r} r ri r r@ r@ rC r s zSelectQuery.getr c C s< | }|| |jd | }|r8|dkr4|d S |S d S Nr r )r{ r r )rA r* rv r r@ r@ rC peek s
zSelectQuery.peekc C s" | j |kr|| _ d| _| j|dS )NT)r* )r r r r) r@ r@ rC r & s
zSelectQuery.firstc C s | | S r? )r r r r@ r@ rC ri , s zSelectQuery.sqlc C sT | j }| jD ]B}t|tr*|j |kr* dS t|tr |jd k r |j|kr dS q dS )NFT)r} r rQ r r% r) )rA r} r& r@ r@ rC verify_naive/ s
zSelectQuery.verify_naivec C s | j | jfS r? )r r r r@ r@ rC get_query_meta9 s zSelectQuery.get_query_metac C sf | j r| jtS | jr$| jtS | js8| jr8| rD| jt S | j
rV| jt S | jt S d S r? )
r r r r r r r r r r r RESULTS_AGGREGATE_MODELSRESULTS_MODELSr r@ r@ rC _get_result_wrapper< s z SelectQuery._get_result_wrapperc C sN | j s| jd krD| j}| }| }||| || _d| _ | jS | jS d S r' )r r r} r r rz )rA r} Z
query_meta
ResultWrapperr@ r@ rC r{ H s zSelectQuery.executec C s t | S r? r r r@ r@ rC r S s zSelectQuery.__iter__c C s t | S r? r r{ r( r r@ r@ rC r( V s zSelectQuery.iteratorc C sN | }t|tr|j}n|}|d k r:|dkr6|d nd }|| |j| S r )r{ rQ slicestopr r )rA r^ rv r r@ r@ rC rd Y s
zSelectQuery.__getitem__c C s t | S r? )r r{ r r@ r@ rC r! d s zSelectQuery.__len__c C s t | S r? )idr r@ r@ rC r h s zSelectQuery.__hash__)r )T)TF)T)T)T)T)N)N)N)NT)F)F)r )r )5rE rF rG rQ r r] r r r rT rS rV rY r r r r4 r r r rx rz r r r r r r r| r r r6 r r r rp r r r r r ri r r r r{ r r( rd r! PY3r r r@ r@ r rC r Y s|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )NoopSelectQueryc C s | j dfS Nr@ )r get_noop_sqlr r@ r@ rC ri l s zNoopSelectQuery.sqlc C s dS )Nr r@ r r@ r@ rC r o s z NoopSelectQuery.get_query_metac C s | j tS r? )r r r r r@ r@ rC r r s z#NoopSelectQuery._get_result_wrapperN)rE rF rG ri r r r@ r@ r@ rC r k s r c sP e Zd ZdZd fdd Z fddZddd Zd
d Zd d
Zdd Z Z
S )r rz Nc s( || _ || _|| _tt| |g d S r? )r rX r@ r r r )rA r} r rX r@ r r@ rC r x s zCompoundSelect.__init__c s, t t| |}| j|_| j|_| j|_|S r? )r r r] r rX r@ r r r@ rC r] ~ s
z CompoundSelect._clone_attributesFc C s | j |dS )Nr )r r r@ r@ rC rp s zCompoundSelect.countc C s
| j S r? )r r r r@ r@ rC r s z CompoundSelect.get_query_metac C s | j o| j S r? )r r r@ r r@ r@ rC r s zCompoundSelect.verify_naivec C s | j r| jtS | jr$| jtS | jr6| jtS | jj pD| j
j }| jj pZ| j
j pZ| j }|sl|rl| rx| jt
S | jtS d S r? )r r r r r r r r r r r@ r r r r )rA Z has_joinsZis_naiver@ r@ rC r s z"CompoundSelect._get_result_wrapper)NNN)F) rE rF rG rQ r r] rp r r r r r@ r@ r rC r u s
r c st e Zd Z fddZ fddZdd Zeedd Zeedd
d Zeedd d
Z dd Z
dd Z Z S ) _WriteQueryc s, d | _ d| _d| _d | _tt| | d S r' )r r r r r r r r[ r r@ rC r s
z_WriteQuery.__init__c s6 t t| |}| jr2t| j|_| j|_| j|_|S r? )r r r] r r r r r r r@ rC r] s z _WriteQuery._clone_attributesc s fdd}|S )Nc s0 | j jj}|js tdt| | f||S )Nz/RETURNING is not supported by your database: %s)r} rz r r rw r )rA r r r r, r@ rC r s
z-_WriteQuery.requires_returning..innerr@ )r, r r@ r rC requires_returning s z _WriteQuery.requires_returningc G s> t |dkr |d d kr d | _n|s.| jjj}| || _d S r )r r r} rz r re r r@ r@ rC returning s
z_WriteQuery.returningTc C s
|| _ d S r? r r r@ r@ rC r| s z_WriteQuery.tuplesc C s
|| _ d S r? r r r@ r@ rC r s z_WriteQuery.dictsc C s: | j d k r.| jr | jtS | jr.| jtS | jtS r? )r r r r r r r r r r@ r@ rC r s
z _WriteQuery.get_result_wrapperc C s2 | }| j| jg if}|| j| || _| jS r? )r r r} rz r )rA r rI r@ r@ rC _execute_with_result_wrapper s z(_WriteQuery._execute_with_result_wrapper)T)T)
rE rF rG r r] r r r r| r r r r r@ r@ r rC r s r c s` e Zd Zd fdd Z fddZedddZedZd d
Z d d Z
d
d Z dd Z Z
S ) UpdateQueryNc s || _ d | _tt| | d S r? )r r r r r )rA r} r r r@ rC r s zUpdateQuery.__init__c s( t t| |}t| j|_| j|_|S r? )r r r] r` r r r r r@ rC r] s z UpdateQuery._clone_attributesc C s
|| _ d S r? r rA actionr@ r@ rC on_conflict s zUpdateQuery.on_conflictr c C s | | S r? )r r r r@ r@ rC ri s zUpdateQuery.sqlc C s@ | j d k r | jd kr | S | jd k r,| jS | j| S d S r? r r r r
rows_affectedrz r r@ r@ rC r{ s
zUpdateQuery.executec C s | j jjjstdt| S )NzwUPDATE queries cannot be iterated over unless they specify a RETURNING clause, which is not supported by your database.)r} rz r r rw r r{ r r@ r@ rC r s zUpdateQuery.__iter__c C s t | S r? r r r@ r@ rC r( s zUpdateQuery.iterator)N)N)rE rF rG r r] r r r r ri r{ r r( r r@ r@ r rC r s r c s e Zd Zd fdd Zdd Z fddZed Zed
Ze dd d
Z
e d ddZ e d ddZ e
dd Zdd Zdd Zdd Z ZS ) InsertQueryNFc sf t t| | d| _|d k p$|d k | _d| _|d k r>|| _n |pDi g| _|| _|| _|| _ d | _
d S r' ) r r r r _is_multi_row_insert_return_id_listr r r _validate_fieldsr )rA r} r rowsr r validate_fieldsr r@ rC r
s zInsertQuery.__init__c
# s | j j}| jr |j fdd}|j}|j}| jD ]}| }t }|D ]B}| jrZ|| ||j krp|j | } n|} || || < |
| qH|r|D ]} | |kr|| || < q|V q2d S )Nc s | krt d| d S )Nz "%s" is not a recognized field.)rJ )r valid_fieldsr@ rC validate_field
s z.InsertQuery._iter_rows..validate_field) r} rz r r
_default_dict_default_callablesr r r r ry )
rA Z
model_metar defaultsZ callablesr Z field_rowr r r r@ r rC r
s,
zInsertQuery._iter_rowsc sT t t| |}| j|_| j|_| j|_| j|_| j|_| j|_| j |_ | j
|_
|S r? ) r r r] r r r r r r r r r r r@ rC r] /
s z InsertQuery._clone_attributesr z where clauseTc C s
|| _ d S r? )r )rA upsertr@ r@ rC r >
s zInsertQuery.upsertc C s
|| _ d S r? r r r@ r@ rC r B
s zInsertQuery.on_conflictc C s
|| _ d S r? )r )rA return_id_listr@ r@ rC r F
s zInsertQuery.return_id_listc C s | j jr| jr| jrdS dS NTF)r insert_returningr r r r@ r@ rC r J
s z InsertQuery.is_insert_returningc C s | | S r? )r r r r@ r@ rC ri Q
s zInsertQuery.sqlc C sP g }d }| j }| jD ]*}t| j|| j }|r|| q|rH|S |S d S r? )r r r r} r r r{ r~ )rA Zid_listZlast_idr r# r@ r@ rC _insert_with_loopT
s
z InsertQuery._insert_with_loopc C s | j o"| jd ko"| jd ko"| jj }|r0| S | jd k rL| jd krL| S | jd k r\| jS | }| j s| jj r|
}| j j }dd t
| |D }| j j jr|S |d S | j|| j S | jrttd| S dS d S )Nc S s g | ]\}}| |qS r@ )r )r r r4 r@ r@ rC r w
s z'InsertQuery.execute..r T)r r r r insert_manyr r r rz r r% r} rz rW r r last_insert_idr r rX
itemgetterfetchall)rA Zinsert_with_loopr Zpk_rowrI Z
clean_datar@ r@ rC r{ d
s8
zInsertQuery.execute)NNNNF)T)N)T)rE rF rG r r r] r r r5 r r r r r r ri r r{ r r@ r@ r rC r
s(
r c @ s$ e Zd ZedZdd Zdd ZdS ) DeleteQueryr c C s | | S r? )r r r r@ r@ rC ri
s zDeleteQuery.sqlc C s@ | j d k r | jd kr | S | jd k r,| jS | j| S d S r? r r r@ r@ rC r{
s
zDeleteQuery.executeN)rE rF rG r r ri r{ r@ r@ r@ rC r
s r
IndexMetadata)r ri r r r ColumnMetadata)r Z data_typer r r ForeignKeyMetadata)r4
dest_table dest_columnr c @ s e Zd ZdS )PeeweeExceptionNrE rF rG r@ r@ r@ rC r
s r c @ s e Zd ZdS )r! Nr r@ r@ r@ rC r!
s c @ s e Zd ZdS )r Nr r@ r@ r@ rC r
s c @ s e Zd ZdS )r Nr r@ r@ r@ rC r
s c @ s e Zd ZdS )r# Nr r@ r@ r@ rC r#
s c @ s e Zd ZdS )r$ Nr r@ r@ r@ rC r$
s c @ s e Zd ZdS )r% Nr r@ r@ r@ rC r%
s c @ s e Zd ZdS )r, Nr r@ r@ r@ rC r,
s c @ s e Zd ZdS )r- Nr r@ r@ r@ rC r-
s c @ s e Zd ZdS )r2 Nr r@ r@ r@ rC r2
s c @ s* e Zd ZdgZdd Zdd Zdd ZdS ) ExceptionWrapper
exceptionsc C s
|| _ d S r? )r )rA r r@ r@ rC r
s zExceptionWrapper.__init__c C s d S r? r@ r r@ r@ rC __enter__
rT zExceptionWrapper.__enter__c C sH |d kr d S |j | jkrD| j|j }tr.|}n|j}t||| | d S r? )rE r PY26r r` )rA exc_type exc_value tracebackZnew_typeZexc_argsr@ r@ rC __exit__
s zExceptionWrapper.__exit__N)rE rF rG r r r r r@ r@ r@ rC r
s r c s e Zd Z fddZ ZS )_BaseConnectionLocalc s4 t t| jf | d | _d| _d | _g | _g | _d S rI )r r r
autocommitclosedconn
context_stack transactionsr r r@ rC r
s z _BaseConnectionLocal.__init__)rE rF rG r r r@ r@ r rC r
s r c @ s e Zd ZdS )_ConnectionLocalNr r@ r@ r@ rC r
s r c
@ s| e Zd ZdZeZddddgZdZdZdZ i Z
dZ dZ dZ
dZdZdZdZi Zd Zg ZdZdZdZdZdZdZee e ee e e e!e"d
Z#d}d d Z$d
d Z%dd Z&dd Z'dd Z(dd Z)dd Z*dd Z+dd Z,d d Z-d d Z.e/d!d" Z0e/d#d$ Z1d%d& Z2d'd( Z3d)d* Z4d+d, Z5d-d. Z6d~d/d0Z7d1d2 Z8d3d4 Z9d5d6 Z:d7d8 Z;d9d: Z Z>d?d@ Z?ddAdBZ@dCdD ZAdEdF ZBdGdH ZCdIdJ ZDdKdL ZEddMdNZFdOdP ZGddQdRZHddSdTZIddUdVZJddWdXZKddYdZZLd[d\ ZMdd]d^ZNdd_d`ZOddadbZPddcddZQddedfZRdgdh ZSddidjZTddkdlZUddmdnZVddodpZWdqdr ZXdsdt ZYdudv ZZdwdx Z[dydz Z\d{d| Z]dS )DatabaseFr r r r Trm Nrl ) ZConstraintErrorr r r# r$ r% r, r- r2 c K sn i | _ |rt | _nt | _| j|f| t | _|| _|| _ || _
t | j |pRi | _ t | j
|pdi | _
d S r? )connect_kwargsr _localr init threadingZLock
_conn_lockr autorollback use_speedupsr rw rx ) rA r Z threadlocalsr r opsr r r r@ r@ rC r
s
zDatabase.__init__c K s0 | s| |d k| _|| _| j| d S r? ) is_closedr& r= r r r )rA r r r@ r@ rC r
s
z
Database.initc C s
t | jS r? )r r r r@ r@ rC exception_wrapper s zDatabase.exception_wrapperc
C sd | j T | jrtd| 2 | j| jf| j| j_d| j_ |
| jj W 5 Q R X W 5 Q R X d S )NzBError, database not properly initialized before opening connectionF) r r= Exceptionr
_connectr r r r r initialize_connectionr r@ r@ rC connect s
zDatabase.connectc C s d S r? r@ rA r r@ r@ rC r
s z Database.initialize_connectionc
C sN | j > | jrtd| | | jj d| j_W 5 Q R X W 5 Q R X d S )NzBError, database not properly initialized before closing connectionT)r r= r r
_closer r r r r@ r@ rC r& s
zDatabase.closec C s: | j jr"| j jd j}|d k r"|S | j jr2| | j jS r< )r r
connectionr r r r r@ r@ rC get_conn s zDatabase.get_connc C s | j jS r? )r r r r@ r@ rC r ) s zDatabase.is_closedc C s | S r? )r r r r@ r@ rC
get_cursor, s zDatabase.get_cursorc C s | d S r? )r& r r@ r@ rC r / s zDatabase._closec K s t d S r? rw )rA r r r@ r@ rC r 2 s zDatabase._connectc C s t | j|| _d S r? )r rw )r. r r@ r@ rC register_fields5 s zDatabase.register_fieldsc C s t | j|| _d S r? )r rx )r. r r@ r@ rC register_ops9 s zDatabase.register_opsc C sl |t kr| jrtS tS |tkr"tS |tkr8| jr4tS tS |t krN| jrJt
S t S |t krZt
S | jrdtS tS d S r? )r r r r2 r r6 r r r/ r r r5 r rO )rA Z wrapper_typer@ r@ rC r = s&
zDatabase.get_result_wrapperc C s |j jr|jS d S r? )rz r Z lastrowid)rA r r r@ r@ rC r O s zDatabase.last_insert_idc C s |j S r? )Zrowcount)rA r r@ r@ rC r S s zDatabase.rows_affectedc C s | | j| j| j| jS r? )compiler_classrn ro rw rx r r@ r@ rC r V s zDatabase.compilerc C s | j | | S r? )rx r r )rA r r@ r@ rC r{ [ s zDatabase.executec C s t ||f | d | }z|||p.d W n, tk
r` | rZ| jrZ| Y nX |rv| rv| W 5 Q R X |S r )
loggerdebugr
r r{ r get_autocommitr rollbackcommit)rA ri rj ry r r@ r@ rC rx ^ s
zDatabase.execute_sqlc C s d S r? r@ r r@ r@ rC beginm s zDatabase.beginc C s | d S r? )r r r r@ r@ rC r p s zDatabase.commitc C s | d S r? )r r r r@ r@ rC r s s zDatabase.rollbackc C s || j _d S r? )r r )rA r r@ r@ rC set_autocommitv s zDatabase.set_autocommitc C s | j jd kr| | j | j jS r? )r r r r r@ r@ rC r y s zDatabase.get_autocommitc C s | j j| d S r? )r r r~ rA transactionr@ r@ rC push_execution_context~ s z Database.push_execution_contextc C s | j j d S r? )r r r# r r@ r@ rC pop_execution_context s z Database.pop_execution_contextc C s t | jjS r? )r r r r r@ r@ rC execution_context_depth s z Database.execution_context_depthc C s t | |dS )N)with_transaction)ExecutionContext)rA r# r@ r@ rC execution_context s zDatabase.execution_contextc C s | j j| d S r? )r r r~ r r@ r@ rC push_transaction s zDatabase.push_transactionc C s | j j d S r? )r r r# r r@ r@ rC pop_transaction s zDatabase.pop_transactionc C s t | jjS r? )r r r r r@ r@ rC transaction_depth s zDatabase.transaction_depthc C s t | S r? r r r@ r@ rC r s zDatabase.transactionc s t fdd}|S )Nc
s* | |W 5 Q R S Q R X d S r? r) r| r rA r@ rC r s
z)Database.commit_on_success..innerr )rA r r r@ r* rC commit_on_success s zDatabase.commit_on_successc C s | j s
tt| |S r? )
savepointsr savepointrA sidr@ r@ rC r- s zDatabase.savepointc C s t | S r? )_atomicr r@ r@ rC atomic s zDatabase.atomicc C s t d S r? rw rA r r@ r@ rC
get_tables s zDatabase.get_tablesc C s t d S r? rw rA r r r@ r@ rC get_indexes s zDatabase.get_indexesc C s t d S r? rw r4 r@ r@ rC get_columns s zDatabase.get_columnsc C s t d S r? rw r4 r@ r@ rC get_primary_keys s zDatabase.get_primary_keysc C s t d S r? rw r4 r@ r@ rC get_foreign_keys s zDatabase.get_foreign_keysc C s t d S r? rw )rA seqr@ r@ rC sequence_exists s zDatabase.sequence_existsc C s | }| j||| S r? )r rx r )rA r} r qcr@ r@ rC r s zDatabase.create_tablec C s t ||d d S )N)r )create_model_tables)rA r r r@ r@ rC
create_tables s zDatabase.create_tablesc sH | }t|ttfs"td| fdd|D }| j| || S )Nz=Fields passed to "create_index" must be a list or tuple: "%s"c s& g | ] }t |tr jj| n|qS r@ rQ r rz r r r} r@ rC r s z)Database.create_index..)r rQ r rV rw rx r )rA r} r r r; fobjsr@ r? rC r s
zDatabase.create_indexc sH | }t|ttfs"td| fdd|D }| j| || S )Nz;Fields passed to "drop_index" must be a list or tuple: "%s"c s& g | ] }t |tr jj| n|qS r@ r> r r? r@ rC r s z'Database.drop_index..)r rQ r rV rw rx r
)rA r} r r r; r@ r@ r? rC r
s
zDatabase.drop_indexc C s | }| j|||| S r? )r rx r )rA r} r r r; r@ r@ rC r s zDatabase.create_foreign_keyc C s" | j r | }| j|| S d S r? ) sequencesr rx r rA r9 r; r@ r@ rC r s zDatabase.create_sequencec C s | }| j|||| S r? )r rx r )rA r} r r r; r@ r@ rC r s zDatabase.drop_tablec C s t |||d d S )N)r r )drop_model_tables)rA r r r r@ r@ rC drop_tables s zDatabase.drop_tablesc C s | }| j|||| S r? )r rx r )rA r} r r r; r@ r@ rC r s zDatabase.truncate_tablec C s" t t|D ]}||| q d S r? )r rq r )rA r r r r r@ r@ rC truncate_tables s zDatabase.truncate_tablesc C s" | j r | }| j|| S d S r? )rA r rx r rB r@ r@ rC r s zDatabase.drop_sequencec C s t t|td|S Nr r ZEXTRACTr r4 rA r Z
date_fieldr@ r@ rC r s zDatabase.extract_datec C s t ||S r? )r Z
DATE_TRUNCrH r@ r@ rC
truncate_date s zDatabase.truncate_datec C s t dS )NzDEFAULT VALUESr r[ r@ r@ rC r s z Database.default_insert_clausec C s dS )NzSELECT 0 WHERE 0r@ r r@ r@ rC r s zDatabase.get_noop_sqlc C s t S r? )r r r@ r@ rC r s zDatabase.get_binary_type)TTNNFT)NT)T)N)N)N)N)N)N)F)F)F)F)N)FF)FF)FF)FF)^rE rF rG r rh r r r distinct_on drop_cascaderw foreign_keysr for_update_nowaitr r ro r rx rn reserved_tablesr r, rA subquery_delete_same_tabler window_functionsr# r r r$ r% r, r- r2 r r r r
r r
r& r r r r r rR r r r r r r r{ rx r r r r r r r! r" r% r& r' r( r r+ r- r1 r3 r5 r6 r7 r8 r: r r= r r
r r r rD r rE r r rI r r r r@ r@ r@ rC r
s
r c s e Zd ZeZddddZdZeo(ejdkZ dZ
e j de j
diZd Zd* fd d Zd
d Zdd Zdd Zd+ddZd,ddZd-ddZd.ddZd/d d Zd0d d Zd1d d!Zd2d"d#Zd$d% Z d&d' Z d(d) Z Z S )3r6 ri rj )r/ r r F)rM r r r= ZGLOBr zINSERT OR REPLACE INTONc sF |pg | _ |dd }|r*| j d|f tt| j|f|| d S )N journal_mode)_pragmasr# r~ r r6 r )rA r Zpragmasr r rR r r@ rC r s
zSqliteDatabase.__init__c K sJ t s tdt j|f|}d |_z| | W n | Y nX |S )Nz&pysqlite or sqlite3 must be installed.)sqlite3r! r isolation_level_add_conn_hooksr& )rA r r r r@ r@ rC r s zSqliteDatabase._connectc C s8 | | |ddt |ddt |ddt d S )Nr rL Z
date_truncr ) _set_pragmasZcreate_functionr r r r r@ r@ rC rV & s
z SqliteDatabase._add_conn_hooksc C s< | j r8| }| j D ]\}}|d||f q| d S )NzPRAGMA %s = %s;)rS r r{ r& )rA r r Zpragmar^ r@ r@ rC rW , s
zSqliteDatabase._set_pragmasDEFERREDc C s | j d| dd d S )NzBEGIN %sFry rx )rA Z lock_typer@ r@ rC r 3 s zSqliteDatabase.beginc C s t dd S )NzBSQLite does not support ALTER TABLE statements to add constraints.)r- )rA r} r r r@ r@ rC r 6 s z!SqliteDatabase.create_foreign_keyc C s | dd}dd | D S )Nz