Your IP : 216.73.216.213

Current Path : /opt/alt/python35/lib64/python3.5/site-packages/__pycache__/
Current File : //opt/alt/python35/lib64/python3.5/site-packages/__pycache__/peewee.cpython-35.pyc


R6�Wh���@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlZddl
mZddl
mZddlmZddlmZyddlmZWnek
r
eZYnXddlmZddlmZdd	lmZd
Zddd
ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=g3Zydd>lmZWn+ek
r!Gd?d@�d@ej�ZYnXej dA�Z!e!j"e��dBZ#e$dCdD�Z%ej&ddEkZ'ej&ddFkZ(ej&ddE�dEdGfkZ)e(rddl*Z*ddHlm+Z+ddIlm,Z,dJdK�Z-e.Z/e0Z1e.Z2e3e*dL�Z4dMdK�Z5e6Z7ddNdO�Z8n=e'rOe9Z/e2Z1e:Z5dPdQ�Z4e;dR�ne<dS��e)rzdTdGZ=dUdK�Z>ndVdK�Z>yddWl?m@ZAWnek
r�dZAYnXyddlBZBWnek
r�eAZBYnXeAo�eAjCeBjCkr�eAZByddXlDmEZEeEjF�Wnek
r0YnXy ddlGZGddYlGmHZIWnek
rkdZGYnXyddlJZKWn=ek
r�yddlLZKWnek
r�dZKYnXYnXy4ddZlMmNZNdd[lMmOZOdd\lMmPZPWn9ek
r+dd]d^�ZNd_d`�ZOdadb�ZPYnXyDddclMmQZQdddlMmRZRddelMmSZSddflMmTZTWn$ek
r�dZQZRZSZTYnXeBr�eBjUejVe.�eBjUejWe.�eBjUej
e.�dgdhdidjdkdlgZXeYeX�ZZdmdndodpdqdrfZ[dsdt�Z\dgdudhdvdidodjdwdkdxdldmiZ]e]j�Z^dye^dk<dze^dl<d{d|�Z_d}d~�Z`Gdd��d�e�Zaead�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d��Zbead�d�d�d�d�d�d�d��ZcecjdZeecjfZgecjhZid�ZjdEZkdFZld�Zmd�Znd�ebjod�ebjpd�ebjqd�ebjrd�ebjsd�ebjtd�ebjud�ebjvd�ebjwd�ebjxd�ebjyiZzd�d��Z{d�d��Z|d�d��Z}Gd�d3�d3e$�Z~Gd�d�de$�ZGd�d��d�e$�Z�Gd�d��d�e$�Z�Gd�d7�d7e��Z�e�Z�Gd�d��d�e��Z�Gd�d��d�e��Z�e�d�Z�Gd�d��d�e��Z�Gd�d.�d.e��Z�Gd�d��d�e��Z�Gd�d�de��Z�Gd�d��d�e��Z�Gd�d��d�e��Z�Gd�d=�d=e��Z�d�d�Z�Gd�d�de��Z�Gd�d��d�e��Z�ed�d�d�d�d�d�d�d�d�d�d�d�f�Z�Gd�d��d�ed�d�d�d�d�f��Z�Gd�d��d�e$�Z�Gd�d�de��Z�Gd�d�de��Z�Gd�d"�d"e��Z�Gd�d�de��Z�Gd�d5�d5e��Z�Gd�d1�d1e��Z�Gd�d��d�e��Z�Gd�d�de��Z�Gd�d�de��Z�Gd�d�de��Z�ddd�Z�Gdd�de��Z�Gdd�de��Z�Gdd8�d8e��Z�Gdd
�d
e��Z�Gdd<�d<e��Z�dd	�Z�Gd
d�de��Z�Gdd�de��Z�Gd
d�de��Z�Gdd9�d9e��Z�Gdd:�d:e��Z�Gdd�de��Z�Gdd�de��Z�Gdd�de$�Z�Gdd�de$�Z�Gdd �d e��Z�Gdd�de$�Z�Gdd�de$�Z�Gdd�de$�Z�Gdd�de��Z�Gdd �d e$�Z�Gd!d"�d"e$�Z�Gd#d$�d$e��Z�Gd%d&�d&e��Z�eTdkr
e�ZTGd'd(�d(e��Z�eRdkr6
e�ZRGd)d*�d*e��Z�eQdkr^
e�ZQGd+d,�d,e��Z�ed-d.d�f�Z�Gd/d0�d0e��Z�Gd1d2�d2e��Z�Gd3d4�d4e��Z�d5d6�Z�Gd7d8�d8e��Z�Gd9d:�d:e��Z�Gd;d<�d<e��Z�Gd=d>�d>e��Z�Gd?d@�d@e��Z�GdAdB�dBe��Z�GdCdD�dDe��Z�edEdFdGdHdIdJf�Z�edKdFdLdMd�dJf�Z�edNdOdPdQdJf�Z�GdRdS�dSe��Z�GdTd!�d!e��Z�GdUd�de��Z�GdVd�de��Z�GdWd#�d#e��Z�GdXd$�d$e��Z�GdYd%�d%e��Z�GdZd,�d,e��Z�Gd[d-�d-e��Z�Gd\d2�d2e��Z�Gd]d^�d^e$�Z�Gd_d`�d`e$�Z�Gdadb�dbe�e	j��Z�Gdcdd�dde$�Z�Gded6�d6e��Z�Gdfd/�d/e��Z�Gdgd+�d+e��Z�Gdhdi�die$�Z�Gdjdk�dke��Z�Gdld;�d;e��Z�Gdmdn�dne��Z�Gdodp�dpe��Z�Gdqdr�dre��Z�Gdsdt�dte��Z�Gdudv�dve��Z�Gdwdx�dxe$�Z�eSdkr,Gdydz�dze$�ZSGd{d�de��Z�eBrWe�d|�Z�ndZ�Gd}d~�d~e$�Z�Gdd��d�e��Z�Gd�d*�d*e%e���Z�d�d��Z�d�d��Z�ed�d�d�d�d�d�d�f�Z�Gd�d��d�e��Z�d�d0�Z�d�d��Z�d�d��Z�dS(��N)�bisect_left)�bisect_right)�deque)�
namedtuple)�OrderedDict)�deepcopy)�wraps)�isclassz2.8.3�	BareField�BigIntegerField�	BlobField�BooleanField�	CharField�Check�Clause�CompositeKey�
DatabaseError�	DataError�	DateField�
DateTimeField�DecimalField�DeferredRelation�DoesNotExist�DoubleField�DQ�Field�FixedCharField�
FloatField�fn�ForeignKeyField�ImproperlyConfigured�IntegerField�IntegrityError�InterfaceError�
InternalError�JOIN�	JOIN_FULL�
JOIN_INNER�JOIN_LEFT_OUTER�Model�
MySQLDatabase�NotSupportedError�OperationalError�Param�PostgresqlDatabase�prefetch�PrimaryKeyField�ProgrammingError�Proxy�R�SmallIntegerField�SqliteDatabase�SQL�	TextField�	TimeField�TimestampField�Using�	UUIDField�Window)�NullHandlerc@seZdZdd�ZdS)r=cCsdS)N�)�self�recordr>r>�9/opt/alt/python35/lib64/python3.5/site-packages/peewee.py�emitiszNullHandler.emitN)�__name__�
__module__�__qualname__rBr>r>r>rAr=hsr=�peewee�_metaclass_helper_cCs|t|fi�S)N)�_METACLASS_)�meta�baser>r>rA�with_metaclassssrK���)�Callable)�reducecCs
t|t�S)N)�
isinstancerO)�cr>r>rA�<lambda>}srS�printcCst|jd��S)N�raw_unicode_escape)�bytes�encode)�sr>r>rArS�scCs(|j|k	r|j|��|�dS)N)�
__traceback__�with_traceback)�tp�value�tbr>r>rA�reraise�sr^cCs$tjj|�tjjd�dS)N�
)�sys�stdout�write)rXr>r>rA�print_�srcz4def reraise(tp, value, tb=None): raise tp, value, tbzUnsupported python version.�
cCs)|jd|j|jddttS)Ng�i)�microseconds�seconds�days�_M)�tr>r>rArS�scCs
|j�S)N)�
total_seconds)rjr>r>rArS�s)�dbapi2)�compat)�
extensions)�format_date_time)�sort_models_topologically)�strip_parenscCsY|pdd�}x@|D]8}y|tjj||��SWqtk
rPYqXqW|S)NcSs|S)Nr>)�xr>r>rArS�sz"format_date_time.<locals>.<lambda>)�datetime�strptime�
ValueError)r\�formats�post_process�fmtr>r>rAro�s

	rocst���t��g�����fdd��dd�}x*t�d|dd�D]}�|�q[Wtt���S)z@Sort models topologically so that parents will precede children.cs`|�kr\|�kr\�j|�x'|jjj�D]}�|j�q8W�j|�dS)N)�add�_meta�reverse_rel�values�model_class�append)�model�foreign_key)�dfs�models�ordering�seenr>rAr��s

z&sort_models_topologically.<locals>.dfscSs|jj|jjfS)N)rz�name�db_table)�mr>r>rArS�sz+sort_models_topologically.<locals>.<lambda>�key�reverseT)�set�sorted�list�reversed)r��namesr�r>)r�r�r�r�rArp�s	rpcCsC|s|ddkr|Sd}}t|�}xV||kr�||dkr�||ddkr�|d7}|d7}|d8}q4Pq4W|r$d}d}x{t|||�D]f}||dkr�|d7}n||dkr�|d8}|dkr	|d7}d}||kr�Pq�W||8}|dkr?|||�S|S)Nr�(��))�len�range)rX�ct�i�l�
unbalanced_ct�requiredr>r>rArq�s6
$






rq)�_DictQueryResultWrapper)�_ModelQueryResultWrapper)�_SortedFieldList)�_TuplesQueryResultWrapper�year�month�day�hour�minute�secondz%Y-%m-%d %H:%M:%Sz%Y-%m-%d %H:%M:%S.%fz%Y-%m-%dz%H:%M:%Sz%H:%M:%S.%fz%H:%McCs8|tkst�|sdSt|t�}t||�S)N)�DATETIME_LOOKUPS�AssertionErrorro�SQLITE_DATETIME_FORMATS�getattr)�lookup_type�datetime_string�dtr>r>rA�_sqlite_date_parts
r�z%Yz%Y-%mz%Y-%m-%d %Hz%Y-%m-%d %H:%Mz%Y-%m-%d %H:%iz%Y-%m-%d %H:%i:%ScCs<|tkst�|sdSt|t�}|jt|�S)N)�SQLITE_DATE_TRUNC_MAPPINGr�ror��strftime)r�r�r�r>r>rA�_sqlite_date_trunc&s
r�cCstj||tj�dk	S)N)�re�search�I)�regexr\r>r>rA�_sqlite_regexp-sr�c@seZdZdd�ZdS)�attrdictcCs||S)Nr>)r?�attrr>r>rA�__getattr__1szattrdict.__getattr__N)rCrDrEr�r>r>r>rAr�0sr��AND�and�OR�or�ADD�+�SUB�-�MUL�*�DIV�/�BIN_AND�&�BIN_OR�|�XOR�^�MOD�%�EQ�=�LT�<�LTEz<=�GT�>�GTEz>=�NEz!=�IN�in�NOT_INznot in�IS�is�IS_NOTzis not�LIKE�like�ILIKE�ilike�BETWEEN�between�REGEXP�regexp�CONCATz||�INNER�
LEFT_OUTERz
LEFT OUTER�RIGHT_OUTERzRIGHT OUTER�FULLr����eq�lt�lte�gt�gte�necCs|j�}|j|�|S)N)�copy�update)�source�	overrides�mergedr>r>rA�
merge_dictrs
r�cs�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.
    cs |j�}�|||�|S)N)�clone)r?�args�kwargsr�)�funcr>rA�inner}szreturns_clone.<locals>.inner)�
call_local)r�r�r>)r�rA�
returns_clonews	r�cs�fdd�}|S)zu
    Method decorator to indicate a method is not allowed to be called.  Will
    raise a `NotImplementedError`.
    cs#td�t|�jf��dS)Nz!%s is not allowed on %s instances)�NotImplementedError�typerC)r?r�r�)r�r>rAr��sznot_allowed.<locals>.innerr>)r�r�r>)r�rA�not_allowed�sr�csdeZdZdZddgZdd�Zdd�Zdd	�Zd
d�Z�fdd
�Z	�S)r2zg
    Proxy class useful for situations when you wish to defer the initialization
    of an object.
    �obj�
_callbackscCsg|_|jd�dS)N)r��
initialize)r?r>r>rA�__init__�s	zProxy.__init__cCs+||_x|jD]}||�qWdS)N)r�r�)r?r��callbackr>r>rAr�s	zProxy.initializecCs|jj|�|S)N)r�r~)r?rr>r>rA�attach_callback�szProxy.attach_callbackcCs+|jdkrtd��t|j|�S)NzCannot use uninitialized Proxy.)r��AttributeErrorr�)r?r�r>r>rAr��szProxy.__getattr__cs4||jkrtd��tt|�j||�S)NzCannot set attribute on proxy.)�	__slots__r�superr2�__setattr__)r?r�r\)�	__class__r>rAr�szProxy.__setattr__)
rCrDrE�__doc__rrrrr�rr>r>)rrAr2�sc@sReZdZe�Zddd�Zdd�Zdd�Zedd	��Z	dS)
rNcCs/|dk	r+|j�|_|jj|�dS)N)�lower�_rel_model_name�_unresolvedry)r?�rel_model_namer>r>rAr�szDeferredRelation.__init__cCs||_||_||_dS)N)r}�fieldr�)r?r}rr�r>r>rA�	set_field�s		zDeferredRelation.set_fieldcCs)||j_|jj|j|j�dS)N)r�	rel_model�add_to_classr}r�)r?rr>r>rA�	set_model�szDeferredRelation.set_modelcCsYttj�}xC|D];}|j|jj�kr|j|�tjj|�qWdS)N)r�rrrrCr
r�discard)�	model_cls�
unresolved�drr>r>rA�resolve�s


zDeferredRelation.resolve)
rCrDrEr�rrrr�staticmethodrr>r>r>rAr�s
	c@seZdZddd�ZdS)�_CDescriptorNcCs|dk	rt|j�S|S)N)�Entity�_alias)r?�instance�
instance_typer>r>rA�__get__�s
z_CDescriptor.__get__)rCrDrErr>r>r>rAr�src@seZdZdZe�ZdZdd�Zedddd��Z	d	d
�Z
dd�Zed
d��Z
eddd��Zedd��Zedd��Zedd��Zdd�Zdd�Zddd�Zeej�Zeej�Zeej�Zeej�Zeej�Zeej �Z!Z"eej#�Z$eejdd�Z%eejdd�Z&eejdd�Z'eej dd�Z(Z)eejdd�Z*eejdd�Z+eej#dd�Z,dd �Z-d!d"�Z.eej/�Z0eej1�Z2eej3�Z4eej5�Z6eej7�Z8eej9�Z:eej;�Z<eej=�Z>eej?�Z@eejA�ZBd#d$�ZCd%d&�ZDdd'd(�ZEd)d*�ZFd+d,�ZGd-d.�ZHd/d0�ZId1d2�ZJd3d4�ZKdS)5�Nodez=Base-class for any part of a query which shall be composable.�nodecCs(d|_d|_d|_d|_dS)NF)�_negatedr�_bind_to�	_ordering)r?r>r>rAr�s			z
Node.__init__NFcs���fdd�}|S)Ncs5�p|j}�r!t|�}t�||�|S)N)rCr��setattr)�method�method_name)r��clsr�r>rA�	decorator�s
zNode.extend.<locals>.decoratorr>)r'r�r�r(r>)r�r'r�rA�extend�szNode.extendcCs
t|��S)N)r�)r?r>r>rA�
clone_base�szNode.clone_basecCs@|j�}|j|_|j|_|j|_|j|_|S)N)r*r!rr#r")r?�instr>r>rAr��sz
Node.clonecCs|j|_dS)N)r!)r?r>r>rA�
__invert__�szNode.__invert__cCs
||_dS)N)r)r?�ar>r>rA�alias�sz
Node.aliascCs
||_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")r?�btr>r>rA�bind_to�szNode.bind_tocCs
d|_dS)N�ASC)r#)r?r>r>rA�asc�szNode.asccCs
d|_dS)N�DESC)r#)r?r>r>rA�descsz	Node.desccCs
|j�S)N)r2)r?r>r>rA�__pos__szNode.__pos__cCs
|j�S)N)r4)r?r>r>rA�__neg__
szNode.__neg__cs��fdd�}|S)z�
        Lightweight factory which returns a method that builds an Expression
        consisting of the left-hand and right-hand operands, using `op`.
        cs&�rt|�|�St|�|�S)N)�
Expression)r?�rhs)�inv�opr>rAr�szNode._e.<locals>.innerr>)r:r9r�r>)r9r:rA�_e
szNode._er9TcCs2|dkrt|tjd�St|tj|�S)N)r7�OPr�r�)r?r8r>r>rA�__eq__'szNode.__eq__cCs2|dkrt|tjd�St|tj|�S)N)r7r<r�r�)r?r8r>r>rA�__ne__+szNode.__ne__cCst|tj|�S)N)r7r<r�)r?r8r>r>rA�in_=szNode.in_cCst|tj|�S)N)r7r<r�)r?r8r>r>rA�not_in?szNode.not_incCs,|rt|tjd�St|tjd�S)N)r7r<r�r�)r?�is_nullr>r>rArAAszNode.is_nullcCst|tjd|�S)Nz%%%s%%)r7r<r�)r?r8r>r>rA�containsEsz
Node.containscCst|tjd|�S)Nz%s%%)r7r<r�)r?r8r>r>rA�
startswithGszNode.startswithcCst|tjd|�S)Nz%%%s)r7r<r�)r?r8r>r>rA�endswithIsz
Node.endswithcCs%t|tjt|td�|��S)Nr�)r7r<r�rr3)r?�low�highr>r>rAr�KszNode.betweencCst|tj|�S)N)r7r<r�)r?�
expressionr>r>rAr�MszNode.regexpcCst|tj|�S)N)r7r<r�)r?r8r>r>rA�concatOszNode.concat)LrCrDrEr	rrR�
_node_typer�classmethodr)r*r�r�r,r.r0r2r4r5r6r;r<r��__and__r��__or__r��__add__r��__sub__r��__mul__r��__div__�__truediv__r��__xor__�__radd__�__rsub__�__rmul__�__rdiv__�__rtruediv__�__rand__�__ror__�__rxor__r=r>r��__lt__r��__le__r��__gt__r��__ge__r��
__lshift__r��
__rshift__r��__mod__r��__pow__r��bin_andr��bin_orr?r@rArBrCrDr�r�rHr>r>r>rAr�sh			
rcs:eZdZdZdZ�fdd�Zdd�Z�S)r6z2An unescaped SQL string, with optional parameters.�sqlcs)||_||_tt|�j�dS)N)r\�paramsrr6r)r?r\rf)rr>rArVs		zSQL.__init__cCst|j|j�S)N)r6r\rf)r?r>r>rAr*[szSQL.clone_base)rCrDrEr	rIrr*r>r>)rrAr6RscsFeZdZdZdZ�fdd�Zdd�Zdd�Z�S)	rz/A quoted-name or entity, e.g. "table"."column".�entitycs tt|�j�||_dS)N)rrr�path)r?rh)rr>rArcszEntity.__init__cCs
t|j�S)N)rrh)r?r>r>rAr*gszEntity.clone_basecCsttd|j|f��S)N)r�filterrh)r?r�r>r>rAr�jszEntity.__getattr__)rCrDrEr	rIrr*r�r>r>)rrAr_s
rcs|eZdZdZdZed�Z�fdd�Zeddd	��Z	d
d�Z
dddd
d�Zdd�Z�S)�FunczAn arbitrary SQL function call.r��count�sumcsM||_||_|r-|j�|jknd|_tt|�j�dS)NF)r��	argumentsr
�
_no_coerce�_coercerrjr)r?r�rm)rr>rArrs		$z
Func.__init__TcCs
||_dS)N)ro)r?�coercer>r>rArpxszFunc.coercecCs%t|j|j�}|j|_|S)N)rjr�rmro)r?�resr>r>rAr*|szFunc.clone_baseNcCspt|t�r!|dkr!|}|dkrKtd|d|�j�}nt|j�}t|td�|�S)N�partition_by�order_by�OVER)rQr<�__sql__r6rr)r?rrrs�windowrer>r>rA�over�sz	Func.overcs�fdd�}|S)Ncst�||�S)N)rj)r�r�)r�r>rA�dec�szFunc.__getattr__.<locals>.decr>)r?r�rxr>)r�rAr��szFunc.__getattr__)rkrl)
rCrDrEr	rIr�rnrr�rpr*rwr�r>r>)rrArjms
rjcs=eZdZdZdZd�fdd�Zdd�Z�S)r7z0A binary expression, e.g `foo + 1` or `bar < 7`.rGFcs;tt|�j�||_||_||_||_dS)N)rr7r�lhsr:r8�flat)r?ryr:r8rz)rr>rAr�s
			zExpression.__init__cCst|j|j|j|j�S)N)r7ryr:r8rz)r?r>r>rAr*�szExpression.clone_base)rCrDrEr	rIrr*r>r>)rrAr7�sr7cs=eZdZdZdZd�fdd�Zdd�Z�S)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.
    �paramNcs)||_||_tt|�j�dS)N)r\�adaptrr-r)r?r\r|)rr>rAr�s		zParam.__init__cCst|j|j�S)N)r-r\r|)r?r>r>rAr*�szParam.clone_base)rCrDrEr	rIrr*r>r>)rrAr-�sc@seZdZdZdS)�Passthrough�passthroughN)rCrDrErIr>r>r>rAr}�sr}csFeZdZdZdZdZdZ�fdd�Zdd�Z�S)	rz8A SQL clause, one or more Node objects joined by spaces.�clause� FcsXd|kr|d|_d|kr2|d|_tt|�j�t|�|_dS)N�glue�parens)r�r�rrrr��nodes)r?r�r�)rr>rAr�s

zClause.__init__cCs+t|j�}|j|_|j|_|S)N)rr�r�r�)r?r�r>r>rAr*�szClause.clone_base)	rCrDrEr	rIr�r�rr*r>r>)rrAr�sc@seZdZdZdZdS)�CommaClausez5One or more Node objects joined by commas, no parens.z, N)rCrDrEr	r�r>r>r>rAr��sr�c@seZdZdZdZdS)�EnclosedClausezAOne or more Node objects joined by commas and enclosed in parens.TN)rCrDrEr	r�r>r>r>rAr��sr�cs@eZdZdd�fdd�Zdd�Zdd�Z�S)r<Ncs;tt|�j�||_||_|jp1d|_dS)N�w)rr<rrrrsr)r?rrrs)rr>rAr�s		zWindow.__init__cCsrg}|jr4|jttd�t|j���|jrb|jttd�t|j���tt|��S)NzPARTITION BYzORDER BY)rrr~rr6r�rsr�)r?�over_clausesr>r>rAru�s						zWindow.__sql__cCst|j|j�S)N)r<rrrs)r?r>r>rAr*�szWindow.clone_base)rCrDrErrur*r>r>)rrAr<�scCstd|�S)Nz
CHECK (%s))r6)r\r>r>rAr�scs4eZdZdZ�fdd�Zdd�Z�S)rz:A "django-style" filter expression, e.g. {'foo__eq': 'x'}.cs tt|�j�||_dS)N)rrr�query)r?r�)rr>rAr�szDQ.__init__cCs
t|j�S)N)rr�)r?r>r>rAr*�sz
DQ.clone_base)rCrDrEr	rr*r>r>)rrAr�scs(eZdZdZ�fdd�Z�S)�_StripParensrqcs tt|�j�||_dS)N)rr�rr )r?r )rr>rAr�sz_StripParens.__init__)rCrDrErIrr>r>)rrAr��sr��JoinMetadata�	src_model�
dest_model�src�destr��primary_keyr��
is_backrefr.�is_self_join�
is_expressionc@sUeZdZddd�Zdd�Zdd�Zdd	�Zed
d��ZdS)�JoinNcCs|t|t�st|t�r"dS|jj||�}|dk	rM|dfS|jj||�}|dk	rx|dfSdS)NFT)NN)NN)rQ�SelectQueryrz�
rel_for_model�reverse_rel_for_model)r?r�r�r�fk_fieldr{r>r>rA�get_foreign_key
s

zJoin.get_foreign_keycCs|jptjS)N)�	join_typer%r�)r?r>r>rA�
get_join_typeszJoin.get_join_typecCs0t|t�r|jSt|t�r,|jS|S)N)rQ�
ModelAliasr}r�)r?�model_or_aliasr>r>rA�model_from_aliass
zJoin.model_from_aliasc
Cs�|j|j�}|j|j�}t|jt�rB|jjpEd}t|jttt	f�}t|jt
tf�r�|jp�d}|r�|}|j|j
jk}nQ|j|||j�\}}|dkr�|jdk	r�|j||�\}}|dk	r|j}nd}|s�|dk	rP|rD|j
j}	q�|j}	q�y|jjj}	Wq�tk
r�|j
j}	Yq�Xnd}	td|d|d|jd|jd|p�|	d|d|d|d	|d
||kd|�S)Nr�r�r�r�r�r�r�r�r.r�r�)r�r�r�rQ�onrrr7rjr6r�
FieldProxyr�rz�fieldsr��to_fieldr�ryrr�)
r?r�r��
join_aliasr��on_fieldr�r�r��target_attrr>r>rA�_join_metadatasH$'
		zJoin._join_metadatacCs%t|d�s|j�|_|jS)N�_cached_metadata)�hasattrr�r�)r?r>r>rA�metadataPsz
Join.metadata)	rCrDrEr�r�r�r��propertyr�r>r>r>rAr�	s
1r��_Joinr�r�c@s7eZdZdd�Zddd�Zdd�ZdS)�FieldDescriptorcCs||_|jj|_dS)N)rr��att_name)r?rr>r>rArYs	zFieldDescriptor.__init__NcCs&|dk	r|jj|j�S|jS)N)�_data�getr�r)r?rrr>r>rAr]szFieldDescriptor.__get__cCs'||j|j<|jj|j�dS)N)r�r��_dirtyry)r?rr\r>r>rA�__set__bszFieldDescriptor.__set__)rCrDrErrr�r>r>r>rAr�Vsr�cseZdZdZdZdZdZdZdddddddddddd�fdd�Zd	d
�Z	dd�Z
d
d�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zddd�Zdd�Zdd �Zd!d"�Z�S)#rzA column on a table.rr�unknownFNc

s�||_||_||_||_||_||_||_||_|	|_|
|_	||_
||_tj
d7_
tj
|_|jr�dp�d|jf|_d|_tt|�j�dS)Nr�rLF)�null�index�unique�verbose_name�	help_text�	db_column�default�choicesr��sequence�constraints�schemar�_field_counter�_order�	_sort_key�	_is_boundrr)
r?r�r�r�r�r�r�r�r�r�r�r�r�)rr>rArms"												!	zField.__init__cKs�t|�d|jd|jd|jd|jd|jd|jd|jd|jd	|j	d
|j
d|jd|j|�}|j
r�|j|_|j|_|j
|_
|S)
Nr�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>rAr*�s&													zField.clone_basecCs�||_||_|jp!|j|_|jsNtjdd|�j�|_|jj|�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`.
        z_+r�TN)r�r}r�r�r��sub�titlerz�	add_fieldr$r�r�)r?r}r�r>r>rAr�s			zField.add_to_classcCs
|jjjS)N)r}rz�database)r?r>r>rA�get_database�szField.get_databasecCs%|j�}|j�j�j|�S)N)�get_db_fieldr��compiler�get_column_type)r?�
field_typer>r>rAr��szField.get_column_typecCs|jS)N)�db_field)r?r>r>rAr��szField.get_db_fieldcCsdS)Nr>)r?r>r>rA�
get_modifiers�szField.get_modifierscCs|S)Nr>)r?r\r>r>rArp�szField.coercecCs|dkr|S|j|�S)z5Convert the python value for storage in the database.N)rp)r?r\r>r>rA�db_value�szField.db_valuecCs|dkr|S|j|�S)z/Convert the database value to a pythonic value.N)rp)r?r\r>r>rA�python_value�szField.python_valuecCs,|rt|jjj|j�St|j�S)N)rr}rzr�r�)r?�
with_tabler>r>rA�	as_entity�szField.as_entitycCsB|j�}|r8td|djtt|��f�St|�S)z2Return the column type, e.g. VARCHAR(255) or REAL.z%s(%s)z, )r�r6�join�map�str)r?�column_type�	modifiersr>r>rA�__ddl_column__�s
#zField.__ddl_column__cCs�|j�|j|�g}|js7|jtd��|jrS|jtd��|jrv|jtd|j��|jr�|j|j�|S)z8Return a list of Node instances that defines the column.zNOT NULLzPRIMARY KEYzDEFAULT NEXTVAL('%s'))	r�r�r�r~r6r�r�r�r))r?r��ddlr>r>rA�__ddl__�s				z
Field.__ddl__cCst|jd|jj�S)N�.)�hashr�r}rC)r?r>r>rA�__hash__�szField.__hash__)rCrDrEr	r�r�rIr�rr*rr�r�r�r�rpr�r�r�r�r�r�r>r>)rrArfs,		
cs=eZdZdZd�fdd�Z�fdd�Z�S)r
�bareNcs2tt|�j||�|dk	r.||_dS)N)rr
rrp)r?rpr�r�)rr>rAr�szBareField.__init__cstt|�jd|j|�S)Nrp)rr
r*rp)r?r�)rr>rAr*�szBareField.clone_base)rCrDrEr�rr*r>r>)rrAr
�sc@seZdZdZeZdS)r!�intN)rCrDrEr�r�rpr>r>r>rAr!�sc@seZdZdZdS)r�bigintN)rCrDrEr�r>r>r>rAr�sc@seZdZdZdS)r4�smallintN)rCrDrEr�r>r>r>rAr4�scs(eZdZdZ�fdd�Z�S)r0r�cs'd|d<tt|�j||�dS)NTr�)rr0r)r?r�r�)rr>rAr�s
zPrimaryKeyField.__init__)rCrDrEr�rr>r>)rrAr0�scs(eZdZdZ�fdd�Z�S)�_AutoPrimaryKeyFieldNcsH||jkr+tdt|�|f��tt|�j||�dS)Nz%s must be named `%s`.)�_column_namerur�rr�r)r?r}r�)rr>rArsz!_AutoPrimaryKeyField.add_to_class)rCrDrEr�rr>r>)rrAr��sr�c@seZdZdZeZdS)r�floatN)rCrDrEr�r�rpr>r>r>rArsc@seZdZdZdS)r�doubleN)rCrDrEr�r>r>r>rAr	scsjeZdZdZdddd�fdd�Z�fdd	�Zd
d�Zdd
�Zdd�Z�S)r�decimalrdr�FNcsM||_||_||_|p*tjj|_tt|�j||�dS)N)	�
max_digits�decimal_places�
auto_roundr��DefaultContext�roundingrrr)r?r�r�r�r�r�r�)rr>rArs
			zDecimalField.__init__c
s:tt|�jd|jd|jd|jd|j|�S)Nr�r�r�r�)rrr*r�r�r�r�)r?r�)rr>rAr*s			zDecimalField.clone_basecCs|j|jgS)N)r�r�)r?r>r>rAr�szDecimalField.get_modifierscCsrtj}|s)|dkr|S|d�S|jrn|d�|j}|j}|t|��j|d|�S|S)Nrrdr�)r��Decimalr�r�r�r��quantize)r?r\�D�expr�r>r>rAr�"s			zDecimalField.db_valuecCs9|dk	r5t|tj�r"|Stjt|��SdS)N)rQr�r�r�)r?r\r>r>rAr�,szDecimalField.python_value)	rCrDrEr�rr*r�r�r�r>r>)rrArs	
zutf-8cCs9t|t�r|St|t�r/|j|�St|�S)N)rQ�unicode_type�string_type�decode)rX�encodingr>r>rA�coerce_to_unicode2s

r�csUeZdZdZd�fdd�Z�fdd�Zdd�Zd	d
�Z�S)r�string�cs&||_tt|�j||�dS)N)�
max_lengthrrr)r?r�r�r�)rr>rAr<s	zCharField.__init__cstt|�jd|j|�S)Nr�)rrr*r�)r?r�)rr>rAr*@szCharField.clone_basecCs|jr|jgpdS)N)r�)r?r>r>rAr�EszCharField.get_modifierscCst|pd�S)N�)r�)r?r\r>r>rArpHszCharField.coerce)rCrDrEr�rr*r�rpr>r>)rrAr9s
cs(eZdZdZ�fdd�Z�S)r�
fixed_charcs.tt|�j|�}|r*|j�}|S)N)rrr��strip)r?r\)rr>rAr�NszFixedCharField.python_value)rCrDrEr�r�r>r>)rrArKsc@s"eZdZdZdd�ZdS)r7�textcCst|pd�S)Nr�)r�)r?r\r>r>rArpWszTextField.coerceN)rCrDrEr�rpr>r>r>rAr7TscsFeZdZdZeZ�fdd�Zdd�Zdd�Z�S)r�blobcsDt|jjt�r+|jjj|j�tt|�j||�S)N)	rQrzr�r2r�_set_constructorrrr)r?r}r�)rr>rAr^szBlobField.add_to_classcCs|j�|_dS)N)�get_binary_type�_constructor)r?r�r>r>rAr�cszBlobField._set_constructorcCs>t|t�r|jd�}t|t�r:|j|�S|S)NrU)rQr�rW�
basestringr�)r?r\r>r>rAr�fs

zBlobField.db_value)	rCrDrEr��binary_constructr�rr�r�r>r>)rrArZs
c@s.eZdZdZdd�Zdd�ZdS)r;�uuidc	Cs@t|tj�r|jSytj|�jSWn|SYnXdS)N)rQr��UUID�hex)r?r\r>r>rAr�pszUUIDField.db_valuecCs3t|tj�r|S|dkr&dStj|�S)N)rQr�r)r?r\r>r>rAr�xszUUIDField.python_valueN)rCrDrEr�r�r�r>r>r>rAr;mscs�fdd�}|S)Ncs|jjjj�|�S)N)r}rzr��extract_date)r?)�	date_partr>rArx~sz_date_part.<locals>.decr>)rrxr>)rrA�
_date_part}srcs=eZdZdZd�fdd�Z�fdd�Z�S)�_BaseFormattedFieldNcs2|dk	r||_tt|�j||�dS)N)rvrrr)r?rvr�r�)rr>rAr�s	z_BaseFormattedField.__init__cstt|�jd|j|�S)Nrv)rrr*rv)r?r�)rr>rAr*�sz_BaseFormattedField.clone_base)rCrDrErvrr*r>r>)rrAr�src@s�eZdZdZdddgZdd�Zeed��Zeed��Z	eed	��Z
eed
��Zeed��Zeed��Z
d
S)rrsz%Y-%m-%d %H:%M:%S.%fz%Y-%m-%d %H:%M:%Sz%Y-%m-%dcCs)|r%t|t�r%t||j�S|S)N)rQr�rorv)r?r\r>r>rAr��szDateTimeField.python_valuer�r�r�r�r�r�N)rCrDrEr�rvr�r�rr�r�r�r�r�r�r>r>r>rAr�s	c@sgeZdZdZdddgZdd�Zeed��Zeed��Z	eed	��Z
d
S)r�datez%Y-%m-%dz%Y-%m-%d %H:%M:%Sz%Y-%m-%d %H:%M:%S.%fcCsZ|r4t|t�r4dd�}t||j|�S|rVt|tj�rV|j�S|S)NcSs
|j�S)N)r)rrr>r>rArS�sz(DateField.python_value.<locals>.<lambda>)rQr�rorvrsr)r?r\�ppr>r>rAr��s
zDateField.python_valuer�r�r�N)rCrDrEr�rvr�r�rr�r�r�r>r>r>rAr�s	c@smeZdZdZdddddgZdd�Zeed	��Zeed
��Z	eed��Z
dS)
r8�timez%H:%M:%S.%fz%H:%M:%Sz%H:%Mz%Y-%m-%d %H:%M:%S.%fz%Y-%m-%d %H:%M:%ScCs�|rPt|t�r4dd�}t||j|�St|tj�rP|j�S|dk	r�t|tj�r�tjj|j�S|S)NcSs
|j�S)N)r)rrr>r>rArS�sz(TimeField.python_value.<locals>.<lambda>)rQr�rorvrsr�	timedelta�min)r?r\rr>r>rAr��s
zTimeField.python_valuer�r�r�N)rCrDrEr�rvr�r�rr�r�r�r>r>r>rAr8�s	cs_eZdZdd�ed�D�Z�fdd�Zdd�Zdd	�Zd
d�Z�S)r9cCsg|]}d|�qS)rdr>)�.0r�r>r>rA�
<listcomp>�s	zTimestampField.<listcomp>�cs�|jdd�pd|_|j|jkrVtddjdd�|jD����|jdd�pkd|_tj}|jr�|jn|j|_	|jr�|j
n|j}|jd	|�t
t|�j||�dS)
N�
resolutionr�z,TimestampField resolution must be one of: %sz, css|]}t|�VqdS)N)r�)rr�r>r>rA�	<genexpr>�sz*TimestampField.__init__.<locals>.<genexpr>�utcFr�)�popr�valid_resolutionsrur�rrs�utcfromtimestamp�
fromtimestamp�_conv�utcnow�now�
setdefaultrr9r)r?r�r��_dt�_default)rr>rAr�s#	zTimestampField.__init__cCs|jdkr|jStjS)Nr�)rr�r)r?r>r>rAr��szTimestampField.get_db_fieldcCs�|dkrdSt|tj�r%nJt|tj�rXtj|j|j|j�}ntt||j��S|j	r�t
j|j��}nt
j|j��}||jd7}|jdkr�||j9}tt|��S)Ng���ư>r�)rQrsrr�r�r�r��roundrr�calendar�timegm�utctimetupler�mktime�	timetuple�microsecond)r?r\�	timestampr>r>rAr��s!	
zTimestampField.db_valuecCs�|dk	r�t|tttf�r�|dkr4dS|jdkrtt||j�\}}|j|�jd|�S|j|�S|S)Nrr�r!)rQr�r��longr�divmodr�replace)r?r\rfr>r>rAr��s$
zTimestampField.python_value)	rCrDrEr�rrr�r�r�r>r>)rrAr9�s

c@seZdZdZeZdS)r
�boolN)rCrDrEr�r&rpr>r>r>rAr
scsOeZdZdZ�fdd�Zdd�Zddd�Zd	d
�Z�S)�RelationDescriptorzEForeign-key abstraction to replace a related PK with a related model.cs#||_tt|�j|�dS)N)rrr'r)r?rr)rr>rArs	zRelationDescriptor.__init__cCs�|jj|j�}|dk	s3|j|jkr�|j|jkrs|jj|jj|k�}||j|j<|j|jS|jjs�|jj�|S)N)	r�r�r��
_obj_cacherrr�r�r)r?r�rel_idr�r>r>rA�get_object_or_idsz#RelationDescriptor.get_object_or_idNcCs |dk	r|j|�S|jS)N)r*r)r?rrr>r>rArs
zRelationDescriptor.__get__cCs�t||j�rGt||jjj�|j|j<||j|j<nP|jj	|j�}||j|j<||kr�|j|jkr�|j|j=|j
j|j�dS)N)rQrr�rr�r�r�r�r(r�r�ry)r?rr\�
orig_valuer>r>rAr� s
zRelationDescriptor.__set__)rCrDrEr	rr*rr�r>r>)rrAr'
s
r'c@s1eZdZdZdd�Zddd�ZdS)�ReverseRelationDescriptorz<Back-reference to expose related objects as a `SelectQuery`.cCs||_|j|_dS)N)rr}r)r?rr>r>rAr.s	z"ReverseRelationDescriptor.__init__NcCsA|dk	r=|jj�j|jt||jjj�k�S|S)N)r�select�whererr�r�r�)r?rrr>r>rAr2s"z!ReverseRelationDescriptor.__get__)rCrDrEr	rrr>r>r>rAr,,sr,c@s=eZdZdZdd�Zddd�Zdd�ZdS)	�ObjectIdDescriptorz(Gives direct access to the underlying idcCs"|j|_tj|�|_dS)N)r��	attr_name�weakref�refr)r?rr>r>rAr:szObjectIdDescriptor.__init__NcCs)|dk	r|jj|j�S|j�S)N)r�r�r0r)r?rrr>r>rAr>szObjectIdDescriptor.__get__cCst||j|�dS)N)r$r0)r?rr\r>r>rAr�CszObjectIdDescriptor.__set__)rCrDrEr	rrr�r>r>r>rAr/8sr/cs�eZdZddddd�fdd�Z�fdd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	�fdd�Z
�fdd�Zdd�Zdd�Z
dd�Z�S)rNc	s�|dkr>t|ttf�r>t|t�r>td��||_||_t|ttf�|_||_	||_
||_||_t
t|�j||�dS)Nr?z[Unexpected value for `rel_model`.  Expected `Model`, `Proxy`, `DeferredRelation`, or "self")rQr2r�
issubclassr)�	TypeErrorr�
_related_name�deferred�	on_delete�	on_update�extrar�rrr)	r?r�related_namer7r8r9r�r�r�)rr>rArGs						zForeignKeyField.__init__csOtt|�jd|jd|j�d|jd|jd|jd|j|�S)Nrr:r7r8r9r�)	rrr*r�_get_related_namer7r8r9r�)r?r�)rr>rAr*Ws				zForeignKeyField.clone_basecCst||j�S)N)r'r)r?r>r>rA�_get_descriptoraszForeignKeyField._get_descriptorcCs
t|�S)N)r/)r?r>r>rA�_get_id_descriptordsz"ForeignKeyField._get_id_descriptorcCs
t|�S)N)r,)r?r>r>rA�_get_backref_descriptorgsz'ForeignKeyField._get_backref_descriptorcCs?|jr%t|j�r%|j|�S|jp>d|jjjS)Nz%s_set)r5�callabler}rzr�)r?r>r>rAr;js
z!ForeignKeyField._get_related_namecs�t�jt�r>���fdd�}�jj|�dSt�jt�rj�jj����dS��_��_�jp�d�j�_���jkr��d7��j	s�t
jdd��j��_	�j
j���j��_�jdkr�j�_�jdk	rOt�jt�sat�j�j��_n�jj
j�_�j
jr����fdd	�}�j�jj
jkr�|d
�n"�j�jj
jkr�|d���j
jkr�|d�n��jkr|d
�t���j��t���j��t�j�j�j��d�_��j
j�j<��jj
j�j<dS)Ncs|�_�j���dS)N)rr)r)r}r�r?r>rArqs	z.ForeignKeyField.add_to_class.<locals>.callbackz%s_id�_idz_+r�r?csC|jdd�jj�fd�jd��t||��dS)Nrz%s.%s�backref�obj_id_name)r�rzr�r:r)�msg�context)r}r�rBr?r>rA�invalid�s
		z-ForeignKeyField.add_to_class.<locals>.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)rQrr2rrrr�r}r�r�r�r�r�rzr�r;r:r�rr�r��validate_backrefsr�r{�__dict__r$r<r=r>r��rel)r?r}r�rrEr>)r}r�rBr?rArosP		
	



	
	zForeignKeyField.add_to_classcs2t|jt�s|jj�Stt|�j�S)zr
        Overridden to ensure Foreign Keys use same column type as the primary
        key they point to.
        )rQr�r0r�rr)r?)rr>rAr��s
zForeignKeyField.get_db_fieldcs2t|jt�s|jj�Stt|�j�S)N)rQr�r0r�rr)r?)rr>rAr��s
zForeignKeyField.get_modifierscCs|jj|�S)N)r�rp)r?r\r>r>rArp�szForeignKeyField.coercecCs.t||j�r|j�}|jj|�S)N)rQr�
_get_pk_valuer�r�)r?r\r>r>rAr��szForeignKeyField.db_valuecCs&t||j�r|S|jj|�S)N)rQrr�r�)r?r\r>r>rAr��szForeignKeyField.python_value)rCrDrErr*r<r=r>r;rr�r�rpr�r�r>r>)rrArFs
@	c@sgeZdZdZdZdd�Zdd�Zddd�Zd	d
�Zdd�Z	d
d�Z
dS)rz+A primary key composed of multiple columns.NcGs
||_dS)N)�field_names)r?rJr>r>rAr�szCompositeKey.__init__cCs&||_||_t|||�dS)N)r�r}r$)r?r}r�r>r>rAr�s		zCompositeKey.add_to_classcs0�dk	r,t�fdd�|jD��S|S)Ncsg|]}t�|��qSr>)r�)r�
field_name)rr>rAr�s	z(CompositeKey.__get__.<locals>.<listcomp>)�tuplerJ)r?rrr>)rrAr�szCompositeKey.__get__cCsdS)Nr>)r?rr\r>r>rAr��szCompositeKey.__set__cs5�fdd�t�j|�D�}ttj|�S)Ncs/g|]%\}}�jjj||k�qSr>)r}rzr�)rrr\)r?r>rAr�s	z'CompositeKey.__eq__.<locals>.<listcomp>)�ziprJrP�operator�and_)r?�other�expressionsr>)r?rAr=�szCompositeKey.__eq__cCst|jj|jf�S)N)r�r}rCrJ)r?r>r>rAr��szCompositeKey.__hash__)rCrDrEr	r�rrrr�r=r�r>r>r>rAr�sc@sdeZdZdZddd�Zdd�Zddd	�Zd
d�Zdd
�Zdd�Z	dS)�AliasMaprjrcCsi|_||_dS)N)�
_alias_map�_counter)r?�startr>r>rAr�s	zAliasMap.__init__cCsd|jS)Nz<AliasMap: %s>)rS)r?r>r>rA�__repr__�szAliasMap.__repr__NcCsI||jkrdS|jd7_|p;d|j|jf|j|<dS)Nr�z%s%s)rSrT�prefix)r?r�r.r>r>rAry�szAliasMap.addcCs'||jkr|j|�|j|S)N)rSry)r?r�r>r>rA�__getitem__�s
zAliasMap.__getitem__cCs
||jkS)N)rS)r?r�r>r>rA�__contains__�szAliasMap.__contains__cCsC|r?x6|jj�D]%\}}||kr||j|<qW|S)N)rS�items)r?�	alias_mapr�r.r>r>rAr�s
zAliasMap.update)
rCrDrErWrrVryrXrYr�r>r>r>rArR�srRc2@seZdZddddddddd	d
ddd
dddddddddddddddddddiZejdejdejd ejd!ej	d"ej
d#ejd$ejd%ej
d&ejd'ejd(ejd)ejd*ejd+ejd,ejd-ejd.ejd/ejd0ejd1ejd2ejd3ejd4ejd5ejd6iZej d7ej!d8ej"d9ej#d:iZ$e%Z&d;d<d=d=d>d?�Z'd@dA�Z(dBdC�Z)dDdE�Z*dFdG�Z+dHdI�Z,dJdK�Z-dLdM�Z.dNdO�Z/dPdQ�Z0dRdS�Z1dTdU�Z2dVdW�Z3dXdY�Z4dZd[�Z5d\d]�Z6d^d_�Z7d`da�Z8dbdc�Z9d=d=ddde�Z:d=dfdgdh�Z;d=didj�Z<d=dkdl�Z=dmdn�Z>d=dodp�Z?dqdr�Z@eAdsdt�ZBdudv�ZCdwdx�ZDdydz�ZEd{d|�ZFd}d~�ZGd=dd��ZHeGd��ZId�d�d��ZJeGd��ZKd�d�d�d��ZLeGd��ZMd�d�d�d��ZNeGd��ZOd�d��ZPd�d��ZQeGd��ZRd�d�d��ZSeGd��ZTd�d��ZUeGd��ZVd�d��ZWeGd��ZXd=S)��
QueryCompilerr�r�r��BIGINTr��BLOBr&�SMALLINTr�DATErs�DATETIMEr��DECIMALr��REALr��CHARr�r��INTEGERr�r�r��VARCHARr��TEXTr�TIMEr�r�z<=r�z>=z!=r�zNOT INr�zIS NOTr�r�r�r�r�r�r�r�r��#r�r�r�r�z||z
INNER JOINzLEFT OUTER JOINzRIGHT OUTER JOINz	FULL JOIN�"�?NcCsm||_||_t|j|p$i�|_t|j|p?i�|_|j�|_t	dg�|_
dS)Nr{)�
quote_char�
interpolationr��	field_map�
_field_map�op_map�_op_map�
get_parse_map�
_parse_mapr��_unknown_types)r?rlrm�field_overrides�op_overridesr>r>rArFs		zQueryCompiler.__init__cCsgd|jd|jd|jd|jd|jd|jd|jd|jd	|jd
|j	d|j
iS)NrGr{r~r�rrgrre�select_query�compound_select_queryrq)�_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_parens)r?r>r>rArrOs										zQueryCompiler.get_parse_mapcCsd|j||jfS)Nz%s%s%s)rl)r?rXr>r>rA�quote`szQueryCompiler.quotecCs$||jkr|j|S|j�S)N)ro�upper)r?�fr>r>rAr�cszQueryCompiler.get_column_typecCs|j|S)N)rq)r?�qr>r>rA�get_opfszQueryCompiler.get_opcCst|j�ddd��S)Nr�cSs|djS)Nr)r�)r�r>r>rArSjsz.QueryCompiler._sorted_fields.<locals>.<lambda>)r�rZ)r?�
field_dictr>r>rA�_sorted_fieldsiszQueryCompiler._sorted_fieldscCs|j|gfS)N)rm)r?r r[�convr>r>rA�_parse_defaultlszQueryCompiler._parse_defaultc
Cs�t|jt�r|j}|j|j||�\}}|j|j||�\}}|jrfdnd}|||j|j�|f}	|	||fS)Nz%s %s %sz
(%s %s %s))rQryr�
parse_noder8rzr�r:)
r?r r[r�ry�lparamsr8�rparams�templaterer>r>rAryos	zQueryCompiler._parse_expressioncCs;|jr(|j|j|j�|d�S|j|jgfS)N)r|r�r\rm)r?r r[r�r>r>rAr{xs	z QueryCompiler._parse_passthroughcCs�|jrF|r'|j|jkr'd}|j|j|j�||�S|dk	rn|j|j|j�|�S|j|jgfSdS)N)r|r�r�r\rm)r?r r[r�r>r>rArz}s	zQueryCompiler._parse_paramcCsP|jr|pd}|j|j||�\}}d|jt|�f|fS)Nz%s(%s))ro�parse_node_listrmr�rq)r?r r[r�rerfr>r>rAr|�szQueryCompiler._parse_funccCsG|j|j|||j�\}}|jr=dt|�}||fS)Nz(%s))r�r�r�r�rq)r?r r[r�rerfr>r>rAr}�s
	zQueryCompiler._parse_clausecCs"djt|j|j��gfS)Nr�)r�r�r�rh)r?r r[r�r>r>rAr~�szQueryCompiler._parse_entitycCs|jt|j�fS)N)r\r�rf)r?r r[r�r>r>rAr��szQueryCompiler._parse_sqlcCsV|r:dj|j||j�|j|j�f�}n|j|j�}|gfS)Nr�)r�r�r}r�)r?r r[r�rer>r>rAr�szQueryCompiler._parse_fieldcCsPd}|jj|krF|jj|krF|j|j}}d}n|j|j}}d}|j�}|j|kr�|j|_|j||�\}	}
|j||j||��\}}|r�|||	|
f\}
}}}n|	|
||f\}
}}}|jjr,d|
|j	|f}nd|
|j	|f}|||fS)NrxTFz((%s) %s (%s))z
(%s %s %s))
r8rIry�alias_map_classrT�generate_select�calculate_alias_mapr��compound_select_parenthesesrN)r?r r[r��csq�first_q�second_qr9�new_map�first�first_pr��second_pr��lp�r�rprer>r>rAr��s($	!z*QueryCompiler._parse_compound_select_querycCs}|j�}|jsQ|r6t|t�r6|j}n|jjj}|f|_|j	||�\}}dt
|�|fS)Nz(%s))r��_explicit_selectionrQrr�r}rzr��_selectr�rq)r?r r[r�r��select_fieldr�rfr>r>rAr��s	z!QueryCompiler._parse_select_querycCs.|j|j||�\}}t|�|fS)N)r�r rq)r?r r[r�rerfr>r>rAr��sz!QueryCompiler._parse_strip_parenscCst|dd�}d}||jkrs|j||||�\}}||jkom|jdkom|dk}n�t|tttf�r�|j|||�\}}d|}nKt|t	�r@|j
}|r(t|t�r(|j}t|t�r
|j
|�}	q4|j
t||j��}	n|j�}	|	g}n�t|�r[t|t	�sjt|t�r�|j�j||�}
|j|
||�\}}n]|dk	r�|j
|�}	|j|	|d�\}}}n!|j||d�\}}d}|||fS)NrIFz(%s)T)r�rsrtr|rQr�rLr�r�r)rmrr�r�r�rIr	r3r�r�r.r��_parser�)r?r r[r��	node_typer�rerfr�r\rg�_r>r>rAr��s:
		!zQueryCompiler._parsec
s,|j||��\}}}|rO�dk	rO|rO�fdd�|D�}t|t�r�|jrqd|}|jr�dj|d|jf�}|jr�dj||jf�}|r"tdd�|D��r"g}g}x>t|�D]0\}	}
t|
t�r�|j	|
�\}}q�W||fS)Ncsg|]}�j|��qSr>)r�)rr�)r�r>rAr�s	z,QueryCompiler.parse_node.<locals>.<listcomp>zNOT %sr��AScss|]}t|t�VqdS)N)rQr)r�pr>r>rArsz+QueryCompiler.parse_node.<locals>.<genexpr>)
r�rQrr!rr�r#�any�	enumerater�)
r?r r[r�rerfr��clean_params�	clean_sql�idxr{�csql�cparamsr>)r�rAr��s"	
		zQueryCompiler.parse_nodez, c
Cseg}g}xC|D];}|j|||�\}}	|j|�|j|	�qW|j|�|fS)N)r�r~r)r�)
r?r�r[r�r�rerfr �node_sql�node_paramsr>r>rAr�
s

zQueryCompiler.parse_node_listcCs�|j�}|dk	r$|j|_|j|j|jjj�x�|jj�D]}\}}|j||jj�xX|D]P}t|j	t
�r�|j|j	|j	j�qy|j|j	|j	jj�qyWqPW|j|�S)N)
r�rTryr}rz�table_alias�_joinsrZrQr�rr.r�)r?r�r[r�r��
joined_models�join_objr>r>rAr�s
$z!QueryCompiler.calculate_alias_mapcCs|jt|�|�S)N)r�r)r?�clausesr[r>r>rA�build_query&szQueryCompiler.build_querycCs�g}t�}|g}x�|r�|j�}||ks||krHq|j|�x�||D]�}|}	|j}
t|jtttt	f�r�|jj
�j�}n�|j}|j
r�|j}
|j}n|j}
|j}|j}|r?t|
|j�}t||jj�}|j
r0||}}||k}ntd��t|
t�rc|
}n&|j|
�|
j�j||
�}|j�}||jkr�t|j|�}nt|�}|jt||td�|��q`WqW|S)Nz Missing required join predicate.�ON)r�rryr�rQr�r7rjrrr�r.r�r�r�r�r�r�r�rurr~r�r��join_mapr6)r?�joinsr}r[r�r�r��currr�r�r��
constraintr��fk_model�pk_model�fkryr8�dest_nr��join_sqlr>r>rA�generate_joins)sL			
								
	
$zQueryCompiler.generate_joinscCs�|j}|jj}|j||�}t|t�rHt|�g}n�|jsctd�g}n=td�g}|jdkr�|td�t	|j�g7}t
|j�}d|_|j
|td�f�|jdkr|j|j�j||��n|jt|j��|jdk	r^|jtd��|jtd	d
�|jD���|j|j||�}|r�|j
|�|jdk	r�|j
td�|jg�|jr�|j
td�t|j�g�|jr|j
td
�|jg�|jr/|j
td�t|j�g�|jdk	sP|jr�|jr�|jdk	rh|jn|j}|jtd|��|jdk	r�|jtd|j��|j\}	}
|	r�|
r�dnd}|jt|��|j||�S)N�SELECTzSELECT DISTINCTTFr�z, �FROM�WINDOWcSs7g|]-}tt|j�td�|j���qS)r�)rr6rru)rrvr>r>rArzs	z1QueryCompiler.generate_select.<locals>.<listcomp>�WHEREzGROUP BY�HAVINGzORDER BYzLIMIT %sz	OFFSET %szFOR UPDATE NOWAITz
FOR UPDATE)TF)r}rzr�r�rQ�CompoundSelectr��	_distinctr6r�rr�r�r)�_fromr~r�r.r��_windowsr�r��_where�	_group_by�_having�	_order_by�_limit�_offset�	limit_max�_for_updater�)r?r�r[r�dbr��
select_clause�join_clauses�limit�
for_update�no_wait�stmtr>r>rAr�^sT			#		
	"		"!!zQueryCompiler.generate_selectc
Cst|j}|j�}|j||jj�|jrDd|j}nd}t|�|j�td�g}g}xx|j|j	�D]d\}}t
|ttf�s�t
|d|j�}|jt|jdd�tj|dd��q�W|jt|��|jr$|jtd	�|jg�|jdk	rdt|j�}	d
|	_|jtd�|	g�|j||�S)NzUPDATE OR %s�UPDATE�SETr|r�FrzTr�z, �	RETURNING)r}r�ryrzr��_on_conflictr6r�r��_updaterQrr)r-r�r~r7r<r�r�r�r)�
_returningrr�r�)
r?r�rr[�	statementr�r�rr\�returning_clauser>r>rA�generate_update�s0		!			zQueryCompiler.generate_updatecCs|dd�|D��S)NcSs"g|]}|jdd��qS)r�F)r�)rrr>r>rAr�s	z3QueryCompiler._get_field_clause.<locals>.<listcomp>r>)r?r��clause_typer>r>rA�_get_field_clause�szQueryCompiler._get_field_clausecCs�|j}|j}|j�}|j||jj�|jrL|jj}n|jred|j}nd}t	|�|j
�g}|jdk	r�|jr�|j
|j|j��|j
t|j��n0|jdk	r�gg}}d}	x�|j�D]�}
|	s/t|
j�dtjd��}d}	g}xO|D]G}|
|}
t|
ttf�svt|
d|j�}
|j
|
�q<W|j
t|��q�W|r�|j|j|�t	d�t|�g�n+|jjjr�|j
|jj|j��|j r7|jt	d	�|j|j!�d
t�g�n@|j"dk	rwt#|j"�}d|_$|jt	d	�|g�|j%||�S)NzINSERT OR %s INTOzINSERT INTOFr�r�Tr|�VALUESr�r�z, )&r}rzr�ryr��_upsertr��
upsert_sqlr�r6r��_query�_fieldsr~r�r��_rows�
_iter_rowsr��keysrN�
attrgetterrQrr)r-r�r�r)r��auto_increment�default_insert_clause�is_insert_returning�get_primary_key_fieldsr�rr�r�)r?r�rrIr[r�r�r��
value_clauses�have_fields�row_dictr|rr\r�r>r>rA�generate_insert�s`					

	
			zQueryCompiler.generate_insertcCs�|j}td�|j�g}|jrF|jtd�|jg�|jdk	r�t|j�}d|_|jtd�|g�|j|�S)NzDELETE FROMr�z, r�)	r}r6r�r�r)r�rr�r�)r?r�rr�r�r>r>rA�generate_delete�s			zQueryCompiler.generate_deletecCs.|j|j��}|j|�}t|�S)N)r�r�r�r)r?rr�r�r>r>rA�field_definition	szQueryCompiler.field_definitioncCs�td�t|j��td�|jj�t|jj��g}|jrh|jtd|j��|jr�|jtd|j��t|�S)NzFOREIGN KEY�
REFERENCESzON DELETE %szON UPDATE %s)	r6r�r�rr�r7r~r8r)r?rr�r>r>rA�foreign_key_constraints				z$QueryCompiler.foreign_key_constraintcs�fdd�}|S)Ncs%t|��}|j|||��S)N)r�r�)r?r�r�r)�
function_namer>rAr�sz/QueryCompiler.return_parsed_node.<locals>.innerr>)r�r�r>)r�rA�return_parsed_nodesz QueryCompiler.return_parsed_nodecCsk|p(d|jj|j|jjjf}|j|�}ttd�|j�td�t|�|j	�S)Nzfk_%s_%s_refs_%szALTER TABLEzADD CONSTRAINT)
rzr�r�rr�rr6r�rr�)r?r}rr��	fk_clauser>r>rA�_create_foreign_key#s						z!QueryCompiler._create_foreign_keyr�Fc	s<|rdnd}|j�gg}}�jrr�fdd��jjD�}|jttd�t|���xV�jD]K}|j|j	|��t
|t�r||jr||j|j
|��q|W|jjrx<|jjD].}t
|t�st|�}|j|�q�Wtt|�|j�t||��S)NzCREATE TABLE IF NOT EXISTSzCREATE TABLEcs#g|]}�j|j��qSr>)r�r�)rr�)rIr>rAr7s	z/QueryCompiler._create_table.<locals>.<listcomp>zPRIMARY KEY)rz�
composite_keyr�rJr~rr6r��declared_fieldsr�rQrr6r�r�rr�)	r?r}�safer��columnsr��pk_colsrr�r>)rIrA�
_create_table1s*	
				zQueryCompiler._create_tablercCsM|rdnd}t|�|j�g}|rC|jtd��t|�S)NzDROP TABLE IF EXISTSz
DROP TABLE�CASCADE)r6r�r~r)r?r}�
fail_silently�cascader�r�r>r>rA�_drop_tableLs
zQueryCompiler._drop_tablercCsTtd�|j�g}|r1|jtd��|rJ|jtd��t|�S)NzTRUNCATE TABLEzRESTART IDENTITYr)r6r�r~r)r?r}�restart_identityrr�r>r>rA�_truncate_tableTszQueryCompiler._truncate_tabler	cCsqd|dj|�f}t|�dkrmtj|jd��j�}d|dd�|dd�f}|S)Nz%s_%sr��@zutf-8�7�)r�r��hashlib�md5rW�	hexdigest)r?�tablerr��
index_hashr>r>rA�
index_name^s
$zQueryCompiler.index_namecGs�|jj}|rdnd}|j|dd�|D��}tt|�t|�td�|j�tdd�|D��|�S)NzCREATE UNIQUE INDEXzCREATE INDEXcSsg|]}|j�qSr>)r�)rr�r>r>rArhs	z/QueryCompiler._create_index.<locals>.<listcomp>r�cSsg|]}|j��qSr>)r�)rrr>r>rArns	)rzr�rrr6rr�r�)r?r}r�r�r9�tbl_namer�rr>r>rA�
_create_indexes				zQueryCompiler._create_indexrcCsV|jj}|rdnd}|j|dd�|D��}tt|�t|��S)NzDROP INDEX IF EXISTSz
DROP INDEXcSsg|]}|j�qSr>)r�)rr�r>r>rArus	z-QueryCompiler._drop_index.<locals>.<listcomp>)rzr�rrr6r)r?r}r�rrr�rr>r>rA�_drop_indexrszQueryCompiler._drop_indexrcCsttd�t|��S)NzCREATE SEQUENCE)rr6r)r?�
sequence_namer>r>rA�_create_sequenceyszQueryCompiler._create_sequencercCsttd�t|��S)Nz
DROP SEQUENCE)rr6r)r?rr>r>rA�_drop_sequence}szQueryCompiler._drop_sequencer)YrCrDrErnr<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�rpr%r�r�r�r�r�rRr�rrrr�r�r�r�r�ryr{rzr|r}r~r�rr�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_tablerr�create_indexr�
drop_indexr�create_sequencer�
drop_sequencer>r>r>rAr\	s�																													
	'	5?>
r\c@s"eZdZdddd�ZdS)�SqliteQueryCompilerFcCs|j�j�S)N)�deletere)r?r}rrr>r>rAr�sz"SqliteQueryCompiler.truncate_tableN)rCrDrErr>r>r>rAr!�sr!c@s.eZdZdd�Zdd�ZeZdS)�ResultIteratorcCs||_d|_dS)Nr)�qrw�_idx)r?r$r>r>rAr�s	zResultIterator.__init__cCs�|j|jjkr+|jj|j}nI|jjsn|jj�}|jjj|�|jjd7_nt�|jd7_|S)Nr�)r%r$�_ct�
_result_cache�
_populated�iterater~�
StopIteration)r?r�r>r>rA�next�szResultIterator.nextN)rCrDrErr+�__next__r>r>r>rAr#�sr#c@s�eZdZdZddd�Zdd�Zedd��Zd	d
�Zdd�Z	d
d�Z
dd�Zdd�ZeZ
ddd�ZdS)�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
    NcCst||_||_d|_d|_g|_d|_d|_|dk	r`|\|_|_nd|_|_dS)NrF)	r�cursorr&r%r'r(�_initialized�column_meta�	join_meta)r?rr.rIr>r>rAr�s							zQueryResultWrapper.__init__cCs$|jrt|j�St|�SdS)N)r(�iterr'r#)r?r>r>rA�__iter__�s	
zQueryResultWrapper.__iter__cCs|j�|jS)N)�
fill_cacher&)r?r>r>rArk�s
zQueryResultWrapper.countcCs|jS)N)rk)r?r>r>rA�__len__�szQueryResultWrapper.__len__cCs|S)Nr>)r?�rowr>r>rA�process_row�szQueryResultWrapper.process_rowcCs{|jj�}|sId|_t|jdd�s@|jj�t�n%|jsn|j|jj�d|_|j	|�S)NTr�)
r.�fetchoner(r��closer*r/r�descriptionr7)r?r6r>r>rAr)�s	
			zQueryResultWrapper.iterateccsx|j�VqWdS)N)r))r?r>r>rA�iterator�szQueryResultWrapper.iteratorcCs�|j|jkr5|j|j}|jd7_|S|jrDt�|j�}|jj|�|jd7_|jd7_|S)Nr�)r%r&r'r(r*r)r~)r?r+r�r>r>rAr+�s	zQueryResultWrapper.nextcCs~|ptd�}|dkr*td��|j|_xA|jry||jkryyt|�Wq9tk
ruPYq9Xq9WdS)N�Infrz"Negative values are not supported.)r�rur&r%r(r+r*)r?�nr>r>rAr4�s
zQueryResultWrapper.fill_cache)rCrDrEr	rr3r�rkr5r7r)r;r+r,r4r>r>r>rAr-�s
r-c@seZdZdd�ZdS)�ExtQueryResultWrapperc
Csk|j}g}dd�}x@tt|��D],}|}||d}d}|jdk	r	y|j|}	Wntk
r�Yn�Xt|	t�r�|	j}|	jp�|	j	}d}nSt|	t
�r	t|	j�r	t|	jdt�r	|	jr|	jdj}d}|rD||j
jkrD|j
j|}
|
j	}|
j}|j|||f�q.W||_dS)NcSs|S)Nr>)rrr>r>rArS�sz2ExtQueryResultWrapper.initialize.<locals>.<lambda>rFT)rr�r�r0�
IndexErrorrQrr�rr�rjrmrorzrr~r�)r?r:rr��identityr�r��column�found�
select_column�	field_objr>r>rAr�s8	
					z ExtQueryResultWrapper.initializeN)rCrDrErr>r>r>rAr>�sr>c@seZdZdd�ZdS)�TuplesQueryResultWrappercs#t�fdd�t|�D��S)Ncs-g|]#\}}�j|d|��qS)rL)r�)rr��col)r?r>rAr	s	z8TuplesQueryResultWrapper.process_row.<locals>.<listcomp>)rLr�)r?r6r>)r?rAr7	sz$TuplesQueryResultWrapper.process_rowN)rCrDrEr7r>r>r>rArE	srEc@seZdZdd�ZdS)�NaiveQueryResultWrappercCsQ|j�}x4|jD])\}}}t|||||��qW|j�|S)N)rr�r$�_prepare_instance)r?r6rr�rAr�r>r>rAr7	s

z#NaiveQueryResultWrapper.process_rowN)rCrDrEr7r>r>r>rArG	srGc@seZdZdd�ZdS)�DictQueryResultWrappercCs;i}x.|jD]#\}}}|||�||<qW|S)N)r�)r?r6rqr�rAr�r>r>rAr7#	sz"DictQueryResultWrapper.process_rowN)rCrDrEr7r>r>r>rArI"	srIc@s[eZdZdd�Zdd�Zdd�Zdd�Zd	d
d�Zdd
�Zd	S)�ModelQueryResultWrappercCsJ|j�\|_}tdd�|jD��|_|j|�|_dS)Ncss$|]}t|t�r|VqdS)N)rQr)rrFr>r>rAr/	sz5ModelQueryResultWrapper.initialize.<locals>.<genexpr>)�generate_column_map�
column_mapr�r0�_col_set�generate_join_list�	join_list)r?r:�	model_setr>r>rAr-	sz"ModelQueryResultWrapper.initializec	CsHg}t|jg�}x#t|j�D]\}}d}}t|t�r�t|t�r}|j}|j}|jj	}n|j
}}|j	}|jp�|j}nl|j
dkr�|j}}n
|j
}}t|t�r�|jr�|j}nt|t�r|jd}|j||||f�|j|�q(W||fS)Nr����)r�rr�r0rQrr��_model_alias�field_instancer�r}rr�r"rrrhr~ry)	r?rLr�r�r r�r�r��constructorr>r>rArK3	s,
		
	

z+ModelQueryResultWrapper.generate_column_mapcCsg}|j}|jg}x�|r|j�}||kr?qx�||D]�}|j}|j|ksw|j|krJ|jdk	r�|j|jk}|j|jk}	|jj	o�|p�|	}
nd}
}}	|j
||
||	f�|j
|j�qJWqW|S)NF)r1rrr�r�r�r�rMr�r�r~)r?r�rOr��stack�currentr�r��
fk_present�
pk_present�checkr>r>rArNO	s.				
z*ModelQueryResultWrapper.generate_join_listcCsA|j|�}|j|�}x|D]}|j�q%W|dS)Nr)�construct_instances�follow_joinsrH)r?r6�	collected�	instancesr�r>r>rAr7l	s

z#ModelQueryResultWrapper.process_rowNcCs�i}x�t|j�D]�\}\}}}}|dk	rI||krIq||}	||krl|�||<||}
|dkr�|jj|d}|dk	r�||	�}	t|
||	�qW|S)Nr)r�rLr.r:r$)r?r6r��collected_modelsr�r�rTr�r�r\rr>r>rArZs	s(

z+ModelQueryResultWrapper.construct_instancesc
Cs�||jg}x�|jD]�\}}}}||j}y||j}Wntk
rk||j}YnXd}	|r�|r�|jj|jj	�}	n|r�|jj|j
j	�}	|	s�q|j
dk	}
|
o�|j|jko�t||j
j	�dk}|r(t
||j
j	|j|j�|jr�|
o||jdk	o|t||j
j	�dk	o||jj|jj	�dk}|r�t
||jj	|�t
||j|�|j|�qW|S)NT)rrOr�r��KeyErrorr�r�r�r�r�r�r�r�r$r�r~)
r?r\�preparedr��
check_nullrWrXr+�joined_inst�has_fk�mpk�can_populate_joined_pk�can_populate_joined_fkr>r>rAr[�	sJ

			z$ModelQueryResultWrapper.follow_joins)	rCrDrErrKrNr7rZr[r>r>r>rArJ,	srJ�	JoinCacher�csLeZdZ�fdd�Z�fdd�Zdd�Zdd�Z�S)	�AggregateQueryResultWrappercs&g|_tt|�j||�dS)N)�_rowrrhr)r?r�r�)rr>rAr�	s	z$AggregateQueryResultWrapper.__init__csftt|�j|�t�|_x-|jD]"\}}}}|jj|�q,Wt�|_i|_i|_	i|_
x�|jD]�\}}}}|jr�|j
j}n	|j}|jp�|j}|r�|jj|j�n3|j
j|jt��|j
|jj|j�|j	j|ji�td|d|jpE|�|j	|j|j<q�Wi|_i}x�t|j�D]{\}\}}	}
}||jkr�|jj|g�|j|j||
f�|j|g�||j||
f�qWxa|jD]V}||jkr q|j
j|f�}x&|D]}
|j|j||
�q<WqWdS)Nr�r�)rrhrr��
all_modelsrLry�models_with_aggregate�back_references�source_to_dest�dest_to_sourcerOr�r�r:r�r�r�rr�rgr.�columns_to_comparer�r~r�r))r?r:r�r�r�r�r��key_to_columnsr�r}�col_namer��sources�joined_model)rr>rAr�	sH					'	(

z&AggregateQueryResultWrapper.initializecCs`i}xS|jj�D]B\}}g||<x)|D]!\}}||j||�q3WqW|S)N)rorZr~)r?r6r�r}�column_datar�rqr>r>rA�read_model_data�	s
z+AggregateQueryResultWrapper.read_model_datacCs.|jr|jj�}n|jj�}|sdd|_t|jdd�s[|jj�t�n%|js�|j	|jj
�d|_dd�}i}|j|�}||j}x;|j
�D]-\}}t�||<|||||�<q�W|j|�}x|jj�}	|	dkr#Pt�}
|j|	�}x7|j
�D])\}}
|||
krH|
j|�qHW|
s�|jj|	�P|j|
}|j|	|�}xf|j
�D]X\}}d}x)|jj�D]}|dk	r�d}q�W|s�|||||�<q�WqW|jg}|g}x�|r|j�}||jkrTq0x�|j|D]�}y|j||j\}}Wntk
r�wbYnX|js�|jr}x(||j�D]}t||g�q�W|j|kr�qbx||jj
�D]u\}}|dkrqy|||j|jj}Wntk
rRwYnXt||�j|�|j|�qWnz|r�|j|kr�qbx_||j
�D]M\}}||j|j|jj}t||jj|�|j|�q�W|j|j�qbWq0Wx|D]}|j�qW|S)NTr�cs<�jjr2t�fdd��jjjD��S�j�S)Ncsg|]}�j|�qSr>)r�)rrK)rr>rAr
s	zHAggregateQueryResultWrapper.iterate.<locals>._get_pk.<locals>.<listcomp>)rzr�rLr�rJrI)rr>)rrA�_get_pk

s
z4AggregateQueryResultWrapper.iterate.<locals>._get_pkF) rirr.r8r(r�r9r*r/rr:rZrrZrrur�ryr~rjr�r|r1rmr�r_r�r�r$r�r�rH)r?r6rv�identity_map�_constructed�primary_instancer�r�
model_data�cur_row�duplicate_models�cur_row_datar}�data�different_models�
new_instances�all_noner\rUr]rVr�r�r��pkr+rbr>r>rAr)�	s�		
			

	

		
 

	
z#AggregateQueryResultWrapper.iterate)rCrDrErrrur)r>r>)rrArh�	s6rhcs3eZdZdZdZ�fdd�Zdd�Zdd�Zd	d
�Zdd�Z	d
dd�Z
dd�Zedd��Z
edd��Zed
d
dd��Zed
dd��Zd
dd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d(d)d*�Z�S)+�Queryz?Base class representing a database query on one or more tables.Tcs\tt|�j�||_|jj|_d|_||_|jgi|_d|_	dS)NT)
rr�rr}rzr�r��
_query_ctxr�r�)r?r})rr>rAro
s			zQuery.__init__cCs&|j�\}}d|j||fS)Nz%s %s %s)rer})r?rerfr>r>rArVz
szQuery.__repr__cCs.t|�|j�}|j|_|j|�S)N)r�r}r��_clone_attributes)r?r�r>r>rAr�~
szQuery.clonecCs@|jdk	r!|jj�|_|j�|_|j|_|S)N)r�r��_clone_joinsr�r�)r?r�r>r>rAr��
s
zQuery._clone_attributescCs tdd�|jj�D��S)Ncss'|]\}}|t|�fVqdS)N)r�)r�mc�jr>r>rAr�
sz%Query._clone_joins.<locals>.<genexpr>)�dictr�rZ)r?r>r>rAr��
szQuery._clone_joinsNcCs>ttj|�}|dkr"|S|p.tj}|||�S)N)rPrNrO)r?�initialrQ�conjunction�reducedr>r>rA�_add_query_clauses�
s
zQuery._add_query_clausescCs�g}x�|D]�}t|t�r2|j|�q
t|t�rQ|j|�q
t|t�rv|j|j��q
t|�r
t|t	�r
|j|j
j�q
W|S)N)rQrr~r�r�r)�get_proxy_fieldsr	r3r)rzr�)r?r��accum�argr>r>rA�_model_shorthand�
s
zQuery._model_shorthandcGs|j|j|�|_dS)N)r�r�)r?rQr>r>rAr.�
szQuery.wherecGs"|j|j|tj�|_dS)N)r�r�rN�or_)r?rQr>r>rA�orwhere�
sz
Query.orwherecCs�|j}|sRt|t�p:t|�o:|jj|�}|rqtd��nt|t�rq|jj|}|j	j
|g�|j	|jt||||��t|t�s�||_dS)Nz#A join condition must be specified.)
r�rQr�r	rz�
rel_existsrur�r�r�rr~r�)r?r�r�r�r��require_join_conditionr>r>rAr��
s	#z
Query.joincCs|p|j|_dS)z"Change or reset the query context.N)r}r�)r?r}r>r>rA�switch�
szQuery.switchcKsd|j}x0|jj|g�D]}|j|kr|SqW|j|�j|d||�j|�S)Nr�)r�r�r�r�r�r�)r?�lm�rmr��join_kwargs�ctxr�r>r>rA�ensure_join�
s
	zQuery.ensure_joincCsg}g}ttf}x�t|j��D]�\}}|j}d|kr�|jdd�dtkr�|jdd�\}}t|}n	tj}xK|j	d�D]:}	t
||	�}
t|
|�r�|
j}|j
|
�q�W|j
t|
||��q+W||fS)N�__r�)rr,r�rZr}�rsplit�
DJANGO_MAPr<r��splitr�rQrr~r7)r?�qdictr�r��relationshipr�r\r�r:�piece�
model_attrr>r>rA�convert_dict_to_node�
s 	(
		zQuery.convert_dict_to_nodecOs�t�}|r2|ttjdd�|D��M}|rH|t|�M}t|g�}t�}x�|r7|j�}t|t	�s�qcx�d|j
fd|jffD]�\}}t|t�r#|j|j
�\}	}
|j|
�ttj|	�}|j|_|j|_t|||�q�|j|�q�WqcW|j}|j�}	x||D]t}t|t�r�|j|j}
}|}n.t|t�r�|jj|j}
}|j}|	j|
||�}	qTW|	j|�S)NcSsg|]}|j��qSr>)r�)rr-r>r>rAr�
s	z Query.filter.<locals>.<listcomp>ryr8)rrPrNrOrrr��popleftrQr7ryr8r�r�r�r!rr$r~r�rr}rr,rr�r.)r?r�r��dq_noder��dq_joinsr��sider�r�r�rGrr�r�rDr>r>rAri�
s>	#		+
	
		zQuery.filtercCs
|jj�S)N)r�r�)r?r>r>rAr�szQuery.compilercCs
t�dS)N)r�)r?r>r>rAresz	Query.sqlcCs+|j�\}}|jj|||j�S)N)rer��execute_sql�require_commit)r?rerfr>r>rA�_executeszQuery._executecCs
t�dS)N)r�)r?r>r>rA�executesz
Query.executeFcCsJ|r|j�j�}n|j�j�}|rB|rB|dS|SdS)Nr)�tuplesr�r�r8)r?�as_tuple�convertr6r>r>rA�scalars
zQuery.scalar)rCrDrEr	r�rrVr�r�r�r�r�r�r.r�r�r�r�r�rir�rer�r�r�r>r>)rrAr�k
s.
(r�cs�eZdZdZ�fdd�Zdd�Zed�Zed�Zed�Z	e
d	d
d��Ze
d	dd
��Zdd�Z
dd�Zdd�Z�S)�RawQueryzh
    Execute a SQL query, returning a standard iterable interface that returns
    model instances.
    csM||_t|�|_d|_d|_d|_tt|�j|�dS)NF)	�_sqlr��_params�_qr�_tuples�_dictsrr�r)r?rr�rf)rr>rArs				zRawQuery.__init__cCs7t|j|j|j�}|j|_|j|_|S)N)r�r}r�r�r�r�)r?r�r>r>rAr�&szRawQuery.clone�joiningr.r�TcCs
||_dS)N)r�)r?r�r>r>rAr�0szRawQuery.tuplescCs
||_dS)N)r�)r?�dictsr>r>rAr�4szRawQuery.dictscCs|j|jfS)N)r�r�)r?r>r>rAre8szRawQuery.sqlcCs�|jdkr{|jr-|jjt�}n0|jrK|jjt�}n|jjt�}||j|j	�d�|_|jS)N)
r�r�r��get_result_wrapper�RESULTS_TUPLESr��
RESULTS_DICTS�
RESULTS_NAIVEr}r�)r?�QRWr>r>rAr�;s		zRawQuery.executecCst|j��S)N)r2r�)r?r>r>rAr3FszRawQuery.__iter__)rCrDrEr	rr�r�r�r.r�r�r�r�rer�r3r>r>)rrAr�sr�cKsL|jdd�}|r$td��|r>|p3g|p=dS|rH|SdS)Nr)Fz,"extend" is the only valid keyword argument.)rru)�orig�new_valr�r)r>r>rA�allow_extendIsr�cs�eZdZdZ�fdd�Z�fdd�Zdd�Zee�Zed�Z	ed	�Z
ed
�Zdd�Zd
d�Z
dd�Zee�Zedd��Zedd��Zedd��Zedd��Zedd��Zedd��Zedd��Zedd d!��Zed"d#d$��Zed"d%d&d'��Zed"d(d)��Zed"d*d+��Zed"d,d-��Zed"d.d/��Zed0d1d2��Zd0d3d4�Z d0d5d6�Z!d0d"d7d8�Z"d%d9d:�Z#d%d;d<�Z$d=d>�Z%d?d@�Z&dAdBdC�Z'dAdDdE�Z(dFdG�Z)dHdI�Z*dJdK�Z+dLdM�Z,dNdO�Z-dPdQ�Z.dRdS�Z/dTdU�Z0dVdW�Z1e2r�dXdY�Z3�S)Zr�rwcs�tt|�j|�|jj|_|j|�d|_d|_d|_	d|_
d|_d|_d|_
d|_d|_d|_d|_d|_d|_d|_d|_dS)NF)FF)rr�rr��
commit_selectr��_SelectQuery__selectr�r�r�r�r�r�r�r�r��_naiver�r��_aggregate_rowsrr�)r?r}�	selection)rr>rArUs$
														zSelectQuery.__init__cs�tt|�j|�}|j|_t|j�|_|jdk	r�g|_xI|jD]>}t|t�r�|jj	|j
��qX|jj	|�qXW|jdk	r�t|j�|_|jr�|jj
�|_|j
dk	r�t|j
�|_
|jdk	rt|j�|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|S)N)rr�r�r�r�r�r�rQrr~r�r�r�r�r�r�r�r�r�r�r�r�r�r)r?r�r�)rr>rAr�is6		zSelectQuery._clone_attributescs�fdd�}|S)NcsD|jjjj}�|kr.td���t|j|�|�S)Nz!Your database does not support %s)r}rzr��compound_operationsrur�)r?rP�
supported_ops)rNr>rAr��s

z&SelectQuery.compound_op.<locals>.innerr>)rNr�r>)rNrA�compound_op�szSelectQuery.compound_op�UNION�	INTERSECT�EXCEPTcCsF|jjtd��jt||@�jd��j�}||B|S)Nr�r�)r}r-r6�from_r�r.rs)r?r8�wrapped_rhsr>r>rArR�s"zSelectQuery.__xor__cCstjd�||�S)Nz	UNION ALL)r��_compound_op_static)r?r8r>r>rA�	union_all�szSelectQuery.union_allcGs@t|�dk|_|p'|jjj}|j|�|_dS)Nr)r�r�r}rzr�r�r�)r?r�r>r>rA�__select�szSelectQuery.__selectcGs|rt|�nd|_dS)N)r�r�)r?r�r>r>rAr��szSelectQuery.from_cOs"|r|j|�nd|_dS)N)r�r�)r?r�r�r>r>rA�group_by�szSelectQuery.group_bycGs|j|j|�|_dS)N)r�r�)r?rQr>r>rA�having�szSelectQuery.havingcOs"t|jt|�|�|_dS)N)r�r�r�)r?r�r�r>r>rArs�szSelectQuery.order_bycOs"t|jt|�|�|_dS)N)r�r�r�)r?�windowsr�r>r>rArv�szSelectQuery.windowcCs
||_dS)N)r�)r?�limr>r>rAr��szSelectQuery.limitcCs
||_dS)N)r�)r?�offr>r>rA�offset�szSelectQuery.offset�cCs0|dkr|d8}||_|||_dS)Nrr�)r�r�)r?�page�paginate_byr>r>rA�paginate�s
	zSelectQuery.paginateTcCs
||_dS)N)r�)r?�is_distinctr>r>rA�distinct�szSelectQuery.distinctFcCs||f|_dS)N)r�)r?r��nowaitr>r>rAr��szSelectQuery.for_updatecCs
||_dS)N)r�)r?�naiver>r>rAr��szSelectQuery.naivecCs
||_dS)N)r�)r?r�r>r>rAr��szSelectQuery.tuplescCs
||_dS)N)r�)r?r�r>r>rAr��szSelectQuery.dictscCs
||_dS)N)r�)r?�aggregate_rowsr>r>rAr��szSelectQuery.aggregate_rowsNcCs
||_dS)N)r)r?r.r>r>rAr.�szSelectQuery.aliascCs�|dkr*tj|jj�jd�}|j|krN|j|j�}n|j�}|j	|j|�}|j
s�dd�|jD�|_
t|j�|f|_|S)NrkcSsg|]}|j��qSr>)r.)rrrr>r>rAr�s	z(SelectQuery.annotate.<locals>.<listcomp>)
r�Countrzr�r.r�r�r}r�r�r�r�rL)r?r�
annotationr�r>r>rA�annotate�s	zSelectQuery.annotatecCs=|dkr!tjtd��}|j�}|g|_|S)Nr�)rr�r6rsr�)r?�aggregationr�r>r>rA�
_aggregate�s
zSelectQuery._aggregatecCs|j|�jd|�S)Nr�)r�r�)r?r�r�r>r>rA�	aggregate�szSelectQuery.aggregatecCsJ|js$|js$|js$|jr4|jd|�S|jdd�pIdS)N�clear_limitr�Fr)r�r�r�r��
wrapped_countr�)r?r�r>r>rArk�s$zSelectQuery.countcCsc|j�}|r"d|_|_|j�\}}d|}|jj||�}|j�pbdS)Nz+SELECT COUNT(1) FROM (%s) AS wrapped_selectr)rsr�r�rer}�rawr�)r?r�r�rerf�wrapped�rqr>r>rAr�s
zSelectQuery.wrapped_countcCs4|jdd�}td�g|_t|j��S)Nr��1)r�r6r�r&r�)r?r�r>r>rA�exists
szSelectQuery.existscCs[|jdd�}yt|j��SWn.tk
rV|jjd|j���YnXdS)Nr�z:Instance matching query does not exist:
SQL: %s
PARAMS: %s)r�r+r�r*r}rre)r?r�r>r>rAr�s
	zSelectQuery.getr�cCsN|j�}|j|�|jd|�}|rJ|dkrF|dS|SdS)Nr�r)r�r4r')r?r=rqr�r>r>rA�peeks

zSelectQuery.peekcCs1|j|kr!||_d|_|jd|�S)NTr=)r�r�r�)r?r=r>r>rAr�s		zSelectQuery.firstcCs|j�j|�S)N)r�r�)r?r>r>rAre%szSelectQuery.sqlcCst|j}xd|jD]Y}t|t�r;|j|kr;dSt|t�r|jdk	r|j|krdSqWdS)NFT)r}r�rQrrr")r?r}r r>r>rA�verify_naive(s	zSelectQuery.verify_naivecCs|j|jfS)N)r�r�)r?r>r>rA�get_query_meta2szSelectQuery.get_query_metacCs�|jr|jjt�S|jr2|jjt�S|jsQ|jsQ|j�ra|jjt	�S|j
rz|jjt�S|jjt�SdS)N)
r�r�r�r�r�r�r�r�r�r�r��RESULTS_AGGREGATE_MODELS�RESULTS_MODELS)r?r>r>rA�_get_result_wrapper5s			zSelectQuery._get_result_wrappercCso|js|jdkrd|j}|j�}|j�}|||j�|�|_d|_|jS|jSdS)NF)r�r�r}r�r�r�)r?r}�
query_meta�
ResultWrapperr>r>rAr�As		zSelectQuery.executecCst|j��S)N)r2r�)r?r>r>rAr3LszSelectQuery.__iter__cCst|j�j��S)N)r2r�r;)r?r>r>rAr;OszSelectQuery.iteratorcCsm|j�}t|t�r'|j}n|}|dk	rU|dkrO|dnd}|j|�|j|S)Nrr�)r�rQ�slice�stopr4r')r?r\rqr�r>r>rArXRs
zSelectQuery.__getitem__cCst|j��S)N)r�r�)r?r>r>rAr5]szSelectQuery.__len__cCs
t|�S)N)�id)r?r>r>rAr�aszSelectQuery.__hash__)4rCrDrErIrr�r�rr�rLrKrNrRr�r�r�r-r�r�r�rsrvr�r�r�r�r�r�r�r�r�r.r�r�r�rkr�r�r�r�r�rer�r�r�r�r3r;rXr5�PY3r�r>r>)rrAr�Rsn

	
r�c@s4eZdZdd�Zdd�Zdd�ZdS)�NoopSelectQuerycCs|jj�ffS)N)r��get_noop_sql)r?r>r>rAreeszNoopSelectQuery.sqlcCsdS)N)NNr>)r?r>r>rAr�hszNoopSelectQuery.get_query_metacCs|jjt�S)N)r�r�r�)r?r>r>rAr�ksz#NoopSelectQuery._get_result_wrapperN)rCrDrErer�r�r>r>r>rAr�dsr�csveZdZdZddd�fdd�Z�fdd�Zddd	�Zd
d�Zdd
�Zdd�Z	�S)r�rxNcs8||_||_||_tt|�j|g�dS)N)ryrNr8rr�r)r?r}ryrNr8)rr>rArqs			zCompoundSelect.__init__cs@tt|�j|�}|j|_|j|_|j|_|S)N)rr�r�ryrNr8)r?r�)rr>rAr�ws
z CompoundSelect._clone_attributesFcCs|jd|�S)Nr�)r�)r?r�r>r>rArk~szCompoundSelect.countcCs
|jj�S)N)ryr�)r?r>r>rAr��szCompoundSelect.get_query_metacCs|jj�o|jj�S)N)ryr�r8)r?r>r>rAr��szCompoundSelect.verify_naivecCs�|jr|jjt�S|jr2|jjt�S|jrK|jjt�S|jj	p`|j
j	}|jjp�|j
jp�|j}|s�|s�|j�r�|jjt
�S|jjt�SdS)N)r�r�r�r�r�r�r�r�ryr�r8r�r�r�r�)r?�	has_joins�is_naiver>r>rAr��s			!z"CompoundSelect._get_result_wrapper)
rCrDrErIrr�rkr�r�r�r>r>)rrAr�nsr�cs�eZdZ�fdd�Z�fdd�Zdd�Zeedd���Zeed	d
d���Zeed	dd
���Z	dd�Z
dd�Z�S)�_WriteQuerycs>d|_d|_d|_d|_tt|�j|�dS)NF)r�r�r�r�rr�r)r?r})rr>rAr�s
				z_WriteQuery.__init__csOtt|�j|�}|jrKt|j�|_|j|_|j|_|S)N)rr�r�r�r�r�r�)r?r�)rr>rAr��s	z_WriteQuery._clone_attributescs�fdd�}|S)Ncs>|jjj}|js.tdt|����|||�S)Nz/RETURNING is not supported by your database: %s)r}rzr�r�rur�)r?r�r�r�)r%r>rAr��s
	z-_WriteQuery.requires_returning.<locals>.innerr>)r%r�r>)r%rA�requires_returning�sz_WriteQuery.requires_returningcGsYt|�dkr.|ddkr.d|_n'|sC|jjj}|j|�|_dS)Nr�r)r�r�r}rzr�r�)r?r�r>r>rA�	returning�s
"z_WriteQuery.returningTcCs
||_dS)N)r�)r?r�r>r>rAr��sz_WriteQuery.tuplescCs
||_dS)N)r�)r?r�r>r>rAr��sz_WriteQuery.dictscCsQ|jdk	rA|jr(|jjt�S|jrA|jjt�S|jjt�S)N)r�r�r�r�r�r�r�r�)r?r>r>rAr��s		z_WriteQuery.get_result_wrappercCsI|j�}|j|jgif}||j|j�|�|_|jS)N)r�r�r}r�r�)r?r�rIr>r>rA�_execute_with_result_wrapper�sz(_WriteQuery._execute_with_result_wrapper)rCrDrErr�r�r�r�r�r�r�rr>r>)rrAr��s		r�cs�eZdZd�fdd�Z�fdd�Zeddd��Zed�Zd	d
�Z	dd�Z
d
d�Zdd�Z�S)�UpdateQueryNcs,||_d|_tt|�j|�dS)N)r�r�rrr)r?r}r�)rr>rAr�s		zUpdateQuery.__init__cs:tt|�j|�}t|j�|_|j|_|S)N)rrr�r�r�r�)r?r�)rr>rAr��szUpdateQuery._clone_attributescCs
||_dS)N)r�)r?�actionr>r>rA�on_conflict�szUpdateQuery.on_conflictr�cCs|j�j|�S)N)r�r�)r?r>r>rAre�szUpdateQuery.sqlcCsX|jdk	r(|jdkr(|j�S|jdk	r>|jS|jj|j��SdS)N)r�r�rr��
rows_affectedr�)r?r>r>rAr��s

zUpdateQuery.executecCs.|jjjjstd��t|j��S)NzwUPDATE queries cannot be iterated over unless they specify a RETURNING clause, which is not supported by your database.)r}rzr�r�rur2r�)r?r>r>rAr3�szUpdateQuery.__iter__cCst|j�j��S)N)r2r�r;)r?r>r>rAr;�szUpdateQuery.iterator)
rCrDrErr�r�rr�r�rer�r3r;r>r>)rrAr�srcs�eZdZddddd�fdd�Zdd�Z�fdd�Zed	�Zed
�Ze	ddd
��Z
e	ddd��Ze	ddd��Ze
dd��Zdd�Zdd�Zdd�Z�S)�InsertQueryNFcs�tt|�j|�d|_|dk	p4|dk	|_d|_|dk	r[||_n|pdig|_||_||_||_	d|_
dS)NF)rrrr��_is_multi_row_insert�_return_id_listr�r�r��_validate_fieldsr�)r?r}r��rowsr�r��validate_fields)rr>rAr�s					zInsertQuery.__init__c
#s|jj}|jr0|j��fdd�}|j}|j}x�|jD]�}|j�}t�}xa|D]Y}|jr�||�||j	kr�|j	|}	n|}	||||	<|j
|	�qnW|r�x+|D]#}	|	|kr�||	�||	<q�W|VqLWdS)Ncs |�krtd|��dS)Nz"%s" is not a recognized field.)r_)r)�valid_fieldsr>rA�validate_field
sz.InsertQuery._iter_rows.<locals>.validate_field)r}rzrr�
_default_dict�_default_callablesr�r�r�r�ry)
r?�
model_metar�defaults�	callablesr��	field_rowr�r�rr>)rrAr�
s,					
	

zInsertQuery._iter_rowscs|tt|�j|�}|j|_|j|_|j|_|j|_|j|_|j|_|j	|_	|j
|_
|S)N)rrr�r�r�rr�r�rrr�)r?r�)rr>rAr�(
szInsertQuery._clone_attributesr�zwhere clauseTcCs
||_dS)N)r�)r?�upsertr>r>rAr7
szInsertQuery.upsertcCs
||_dS)N)r�)r?rr>r>rAr;
szInsertQuery.on_conflictcCs
||_dS)N)r)r?�return_id_listr>r>rAr?
szInsertQuery.return_id_listcCs'|jjr#|js|jr#dSdS)NTF)r��insert_returningrr)r?r>r>rAr�C
szInsertQuery.is_insert_returningcCs|j�j|�S)N)r�r�)r?r>r>rAreJ
szInsertQuery.sqlcCsrg}d}|j}xH|jD]=}t|j|�j|j�j�}|r|j|�qW|rj|S|SdS)N)rr�rr}rr�r�r~)r?�id_list�last_idrr6r>r>rA�_insert_with_loopM
s	zInsertQuery._insert_with_loopcCs;|jo1|jdko1|jdko1|jj}|rD|j�S|jdk	rl|jdkrl|j�S|jdk	r�|jS|j�}|js|jj	r�|j
�}|jj}dd�t
|j�|�D�}|jjjr�|S|dS|jj||j�S|jr3ttjd�|j��SdSdS)NcSs%g|]\}}|j|��qSr>)r�)rrrAr>r>rArq
s	z'InsertQuery.execute.<locals>.<listcomp>rT)rr�r�r��insert_manyrr�rr�rr8r}rzrMr�r��last_insert_idrr�rN�
itemgetter�fetchall)r?�insert_with_loopr.�pk_rowrI�
clean_datar>r>rAr�]
s0	


			zInsertQuery.execute)rCrDrErr�r�r�r�r.r�rrrr�r�rerr�r>r>)rrAr�s rc@s4eZdZed�Zdd�Zdd�ZdS)�DeleteQueryr�cCs|j�j|�S)N)r�r�)r?r>r>rAre�
szDeleteQuery.sqlcCsX|jdk	r(|jdkr(|j�S|jdk	r>|jS|jj|j��SdS)N)r�r�rr�rr�)r?r>r>rAr��
s

zDeleteQuery.executeN)rCrDrEr�r�rer�r>r>r>rAr }
sr �
IndexMetadatar�rerr�r�ColumnMetadata�	data_typer��ForeignKeyMetadatarA�
dest_table�dest_columnc@seZdZdS)�PeeweeExceptionN)rCrDrEr>r>r>rAr'�
sr'c@seZdZdS)r N)rCrDrEr>r>r>rAr �
sc@seZdZdS)rN)rCrDrEr>r>r>rAr�
sc@seZdZdS)rN)rCrDrEr>r>r>rAr�
sc@seZdZdS)r"N)rCrDrEr>r>r>rAr"�
sc@seZdZdS)r#N)rCrDrEr>r>r>rAr#�
sc@seZdZdS)r$N)rCrDrEr>r>r>rAr$�
sc@seZdZdS)r+N)rCrDrEr>r>r>rAr+�
sc@seZdZdS)r,N)rCrDrEr>r>r>rAr,�
sc@seZdZdS)r1N)rCrDrEr>r>r>rAr1�
sc@s=eZdZdgZdd�Zdd�Zdd�ZdS)	�ExceptionWrapper�
exceptionscCs
||_dS)N)r))r?r)r>r>rAr�
szExceptionWrapper.__init__cCsdS)Nr>)r?r>r>rA�	__enter__�
szExceptionWrapper.__enter__cCsd|dkrdS|j|jkr`|j|j}trA|}n	|j}t|||�|�dS)N)rCr)�PY26r�r^)r?�exc_type�	exc_value�	traceback�new_type�exc_argsr>r>rA�__exit__�
s		zExceptionWrapper.__exit__N)rCrDrErrr*r1r>r>r>rAr(�
s	r(cs"eZdZ�fdd�Z�S)�_BaseConnectionLocalcsGtt|�j|�d|_d|_d|_g|_g|_dS)NT)rr2r�
autocommit�closed�conn�
context_stack�transactions)r?r�)rr>rAr�
s				z_BaseConnectionLocal.__init__)rCrDrErr>r>)rrAr2�
sr2c@seZdZdS)�_ConnectionLocalN)rCrDrEr>r>r>rAr8�
sr8c@s�eZdZdZeZddddgZdZdZdZ	iZ
dZdZdZ
dZdZdZdZiZd	ZgZdZdZdZdZdZdZd
ededed
ededede de!de"i	Z#dddddddd�Z$dd�Z%dd�Z&dd�Z'dd�Z(dd�Z)dd �Z*d!d"�Z+d#d$�Z,d%d&�Z-d'd(�Z.e/d)d*��Z0e/d+d,��Z1d-d.�Z2d/d0�Z3d1d2�Z4d3d4�Z5d5d6�Z6ddd7d8�Z7d9d:�Z8d;d<�Z9d=d>�Z:d?d@�Z;dAdB�Z<dCdD�Z=dEdF�Z>dGdH�Z?ddIdJ�Z@dKdL�ZAdMdN�ZBdOdP�ZCdQdR�ZDdSdT�ZEddUdV�ZFdWdX�ZGddYdZ�ZHdd[d\�ZIdd]d^�ZJdd_d`�ZKddadb�ZLdcdd�ZMddedf�ZNddgdh�ZOddidj�ZPddkdl�ZQddmdn�ZRdodp�ZSdddqdr�ZTdddsdt�ZUdddudv�ZVdddwdx�ZWdydz�ZXd{d|�ZYd}d~�ZZdd��Z[d�d��Z\d�d��Z]dS)��DatabaseFr�r�r�z	UNION ALLTrkNrj�ConstraintErrorrrr"r#r$r+r,r1c	Ks�i|_|rt�|_nt�|_|j||�tj�|_||_||_	||_
t|j|pvi�|_t|j
|p�i�|_
dS)N)�connect_kwargsr8�_localr2�init�	threading�Lock�
_conn_lockr3�autorollback�use_speedupsr�rurv)	r?r��threadlocalsr3r��opsrArBr;r>r>rAr�
s				zDatabase.__init__cKsB|j�s|j�|dk|_||_|jj|�dS)N)�	is_closedr9r6r�r;r�)r?r�r;r>r>rAr=�
s

	z
Database.initcCs
t|j�S)N)r(r))r?r>r>rA�exception_wrapper�
szDatabase.exception_wrappercCs{|j�k|jrtd��|j��B|j|j|j�|j_d|j_	|j
|jj�WdQRXWdQRXdS)NzBError, database not properly initialized before opening connectionF)r@r6�	ExceptionrF�_connectr�r;r<r5r4�initialize_connection)r?r>r>rA�connects
	
zDatabase.connectcCsdS)Nr>)r?r5r>r>rArI
szDatabase.initialize_connectioncCs]|j�M|jrtd��|j��$|j|jj�d|j_WdQRXWdQRXdS)NzBError, database not properly initialized before closing connectionT)r@r6rGrF�_closer<r5r4)r?r>r>rAr9s
	
zDatabase.closecCsO|jjr/|jjdj}|dk	r/|S|jjrE|j�|jjS)Nr�rQ)r<r6�
connectionr4rJr5)r?r5r>r>rA�get_conns
zDatabase.get_conncCs
|jjS)N)r<r4)r?r>r>rArE"szDatabase.is_closedcCs|j�j�S)N)rMr.)r?r>r>rA�
get_cursor%szDatabase.get_cursorcCs|j�dS)N)r9)r?r5r>r>rArK(szDatabase._closecKs
t�dS)N)r�)r?r�r�r>r>rArH+szDatabase._connectcCst|j|�|_dS)N)r�ru)r'r�r>r>rA�register_fields.szDatabase.register_fieldscCst|j|�|_dS)N)r�rv)r'rDr>r>rA�register_ops2szDatabase.register_opscCs�|tkr|jrtStS|tkr-tS|tkrJ|jrFtStS|t	krg|jrct
StS|tkrwt
S|jr�tStSdS)N)r�rBr�rGr�rJr�r�rEr�r�rIr�rh)r?�wrapper_typer>r>rAr�6s



zDatabase.get_result_wrappercCs|jjr|jSdS)N)rzr��	lastrowid)r?r.rr>r>rArHszDatabase.last_insert_idcCs|jS)N)�rowcount)r?r.r>r>rArLszDatabase.rows_affectedcCs"|j|j|j|j|j�S)N)�compiler_classrlrmrurv)r?r>r>rAr�OszDatabase.compilercCs|j|j�j|��S)N)r�r�r�)r?rr>r>rAr�TszDatabase.executecCs�tj||f�|j��~|j�}y|j||pAf�Wn4tk
r||j�ru|jru|j��YnX|r�|j�r�|j	�WdQRX|S)N)
�logger�debugrFrNr�rG�get_autocommitrA�rollback�commit)r?rerfr�r.r>r>rAr�Ws


zDatabase.execute_sqlcCsdS)Nr>)r?r>r>rA�beginfszDatabase.begincCs|j�j�dS)N)rMrY)r?r>r>rArYiszDatabase.commitcCs|j�j�dS)N)rMrX)r?r>r>rArXlszDatabase.rollbackcCs||j_dS)N)r<r3)r?r3r>r>rA�set_autocommitoszDatabase.set_autocommitcCs,|jjdkr"|j|j�|jjS)N)r<r3r[)r?r>r>rArWrszDatabase.get_autocommitcCs|jjj|�dS)N)r<r6r~)r?�transactionr>r>rA�push_execution_contextwszDatabase.push_execution_contextcCs|jjj�dS)N)r<r6r)r?r>r>rA�pop_execution_contextzszDatabase.pop_execution_contextcCst|jj�S)N)r�r<r6)r?r>r>rA�execution_context_depth}sz Database.execution_context_depthcCst|d|�S)N�with_transaction)�ExecutionContext)r?r`r>r>rA�execution_context�szDatabase.execution_contextcCs|jjj|�dS)N)r<r7r~)r?r\r>r>rA�push_transaction�szDatabase.push_transactioncCs|jjj�dS)N)r<r7r)r?r>r>rA�pop_transaction�szDatabase.pop_transactioncCst|jj�S)N)r�r<r7)r?r>r>rA�transaction_depth�szDatabase.transaction_depthcCs
t|�S)N)r\)r?r>r>rAr\�szDatabase.transactioncs%t����fdd��}|S)Nc
s%�j���||�SWdQRXdS)N)r\)r�r�)r�r?r>rAr��s
z)Database.commit_on_success.<locals>.inner)r)r?r�r�r>)r�r?rA�commit_on_success�s!zDatabase.commit_on_successcCs|jst�t||�S)N)�
savepointsr��	savepoint)r?�sidr>r>rArh�s	zDatabase.savepointcCs
t|�S)N)�_atomic)r?r>r>rA�atomic�szDatabase.atomiccCs
t�dS)N)r�)r?r�r>r>rA�
get_tables�szDatabase.get_tablescCs
t�dS)N)r�)r?rr�r>r>rA�get_indexes�szDatabase.get_indexescCs
t�dS)N)r�)r?rr�r>r>rA�get_columns�szDatabase.get_columnscCs
t�dS)N)r�)r?rr�r>r>rA�get_primary_keys�szDatabase.get_primary_keyscCs
t�dS)N)r�)r?rr�r>r>rA�get_foreign_keys�szDatabase.get_foreign_keyscCs
t�dS)N)r�)r?�seqr>r>rA�sequence_exists�szDatabase.sequence_existscCs%|j�}|j|j||��S)N)r�r�r)r?r}r�qcr>r>rAr�szDatabase.create_tablecCst|d|�dS)Nr)�create_model_tables)r?r�rr>r>rA�
create_tables�szDatabase.create_tablescsf|j�}t|ttf�s1td|���fdd�|D�}|j|j�||��S)Nz=Fields passed to "create_index" must be a list or tuple: "%s"cs5g|]+}t|t�r+�jj|n|�qSr>)rQr�rzr�)rr�)r}r>rAr�s	z)Database.create_index.<locals>.<listcomp>)r�rQr�rLrur�r)r?r}r�r�rs�fobjsr>)r}rAr�s

zDatabase.create_indexcsf|j�}t|ttf�s1td|���fdd�|D�}|j|j�||��S)Nz;Fields passed to "drop_index" must be a list or tuple: "%s"cs5g|]+}t|t�r+�jj|n|�qSr>)rQr�rzr�)rr�)r}r>rAr�s	z'Database.drop_index.<locals>.<listcomp>)r�rQr�rLrur�r)r?r}r�rrsrvr>)r}rAr�s

zDatabase.drop_indexcCs(|j�}|j|j|||��S)N)r�r�r)r?r}rr�rsr>r>rAr�szDatabase.create_foreign_keycCs/|jr+|j�}|j|j|��SdS)N)�	sequencesr�r�r)r?rqrsr>r>rAr�s	zDatabase.create_sequencecCs(|j�}|j|j|||��S)N)r�r�r)r?r}rrrsr>r>rAr�szDatabase.drop_tablecCst|d|d|�dS)Nrr)�drop_model_tables)r?r�rrr>r>rA�drop_tables�szDatabase.drop_tablescCs(|j�}|j|j|||��S)N)r�r�r)r?r}rrrsr>r>rAr�szDatabase.truncate_tablecCs1x*tt|��D]}|j||�qWdS)N)r�rpr)r?r�rrrr>r>rA�truncate_tables�szDatabase.truncate_tablescCs/|jr+|j�}|j|j|��SdS)N)rwr�r�r )r?rqrsr>r>rAr �s	zDatabase.drop_sequencecCstjt|td�|��S)Nr�)r�EXTRACTrr3)r?r�
date_fieldr>r>rAr�szDatabase.extract_datecCstj||�S)N)r�
DATE_TRUNC)r?rr|r>r>rA�
truncate_date�szDatabase.truncate_datecCs
td�S)NzDEFAULT VALUES)r6)r?r}r>r>rAr��szDatabase.default_insert_clausecCsdS)NzSELECT 0 WHERE 0r>)r?r>r>rAr��szDatabase.get_noop_sqlcCstS)N)r�)r?r>r>rAr��szDatabase.get_binary_type)^rCrDrEr�r\rTr�r��distinct_on�drop_cascaderu�foreign_keysr��for_update_nowaitrrrmr�rvrl�reserved_tablesr�rgrw�subquery_delete_same_tabler��window_functionsr"rrr#r$r+r,r1r)rr=rFrJrIr9rMrErNrKrHrJrOrPr�rrr�r�r�rZrYrXr[rWr]r^r_rbrcrdrer\rfrhrkrlrmrnrorprrrrurrrrrryrrzr rr~r�r�r�r>r>r>rAr9�
s�		

r9csBeZdZeZddddddiZdZeoBejd/kZ	d0Z
ejdej
d
iZdZd�fdd�Zdd�Zdd�Zdd�Zddd�Zddd�Zddd�Zddd �Zdd!d"�Zdd#d$�Zdd%d&�Zdd'd(�Zd)d*�Zd+d,�Zd-d.�Z�S)1r5r&rer�r�rgFrMr
�rr��GLOBr�zINSERT OR REPLACE INTONcs]|p	g|_|jdd�}|r=|jjd|f�tt|�j|||�dS)N�journal_mode)�_pragmasrr~rr5r)r?r��pragmasr�r�r�)rr>rArs
zSqliteDatabase.__init__c	KsZtstd��tj||�}d|_y|j|�Wn|j��YnX|S)Nz&pysqlite or sqlite3 must be installed.)�sqlite3r rJ�isolation_level�_add_conn_hooksr9)r?r�r�r5r>r>rArHs	
zSqliteDatabase._connectcCsJ|j|�|jddt�|jddt�|jddt�dS)NrrL�
date_truncr�)�_set_pragmas�create_functionr�r�r�)r?r5r>r>rAr�s
zSqliteDatabase._add_conn_hookscCsT|jrP|j�}x.|jD]#\}}|jd||f�qW|j�dS)NzPRAGMA %s = %s;)r�r.r�r9)r?r5r.�pragmar\r>r>rAr�%s
	zSqliteDatabase._set_pragmas�DEFERREDcCs|jd|dd�dS)NzBEGIN %sr�F)r�)r?�	lock_typer>r>rArZ,szSqliteDatabase.begincCstd��dS)NzBSQLite does not support ALTER TABLE statements to add constraints.)r,)r?r}rr�r>r>rAr/sz!SqliteDatabase.create_foreign_keycCs)|jdd�}dd�|j�D�S)Nz<SELECT name FROM sqlite_master WHERE type = ? ORDER BY name;rcSsg|]}|d�qS)rr>)rr6r>r>rAr6s	z-SqliteDatabase.get_tables.<locals>.<listcomp>)r)r�r)r?r�r.r>r>rArl3s		zSqliteDatabase.get_tablesc	s	d}|j|�df�}t|j���t��|jd��}xG|j�D]9}|d}t|d�dk}|rY�j|�qYWi�xDt��D]6}|jd|�}dd�|j�D��|<q�W����fd	d�t��D�S)
NzQSELECT name, sql FROM sqlite_master WHERE tbl_name = ? AND type = ? ORDER BY namer�zPRAGMA index_list("%s")r�rLzPRAGMA index_info("%s")cSsg|]}|d�qS)rLr>)rr6r>r>rArKs	z.SqliteDatabase.get_indexes.<locals>.<listcomp>cs6g|],}t|�|�||�k���qSr>)r!)rr�)�
index_columns�index_to_sqlr�unique_indexesr>rArNs	)r�r�rr�r�ryr�)	r?rr�r�r.r6r��	is_uniquerr>)r�r�rr�rArm8s 	
!zSqliteDatabase.get_indexescs0|jd��}�fdd�|j�D�S)NzPRAGMA table_info("%s")c	s?g|]5}t|d|d|dt|d����qS)r�rLrMr�)r"r&)rr6)rr>rArXs	z.SqliteDatabase.get_columns.<locals>.<listcomp>)r�r)r?rr�r.r>)rrArnVszSqliteDatabase.get_columnscCs*|jd|�}dd�|j�D�S)NzPRAGMA table_info("%s")cSs$g|]}|dr|d�qS)r�rQr>)rr6r>r>rAr]s	z3SqliteDatabase.get_primary_keys.<locals>.<listcomp>)r�r)r?rr�r.r>r>rAro[szSqliteDatabase.get_primary_keyscs0|jd��}�fdd�|j�D�S)NzPRAGMA foreign_key_list("%s")cs1g|]'}t|d|d|d���qS)rMrLr�)r$)rr6)rr>rAras	z3SqliteDatabase.get_foreign_keys.<locals>.<listcomp>)r�r)r?rr�r.r>)rrArp_szSqliteDatabase.get_foreign_keyscCs
t||�S)N)�savepoint_sqlite)r?rir>r>rArhdszSqliteDatabase.savepointcCstj||�S)N)rr)r?rr|r>r>rArgszSqliteDatabase.extract_datecCstjt||�S)N)rr�r�)r?rr|r>r>rAr~jszSqliteDatabase.truncate_datecCstjS)N)r��Binary)r?r>r>rAr�mszSqliteDatabase.get_binary_type)rMr
r�rrQ)rCrDrEr!rTrur�r��sqlite_version_inforr�r<r�r�rvr�rrHr�r�rZrrlrmrnrorprhrr~r�r>r>)rrAr5�s2	c@sBeZdZdZdZdZdZdddddddd	d
ddd
ddiZdZdZ	dZ
dZej
diZdgZdZdZdZdZddd�Zdd�Zdd�Zddd�Zddd�Zddd �Zdd!d"�Zdd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�ZdS)-r.Tr��BYTEAr&�BOOLEANrs�	TIMESTAMPr��NUMERICr�zDOUBLE PRECISIONr��SERIALr�rz%s�~�userNcKsmtstd��tjd||�}|jrVtjtj|�tjtj|�|ri|j|�|S)Nzpsycopg2 must be installed.r�)	�psycopg2r rJ�register_unicode�
pg_extensions�
register_type�UNICODE�UNICODEARRAY�set_client_encoding)r?r�r�r�r5r>r>rArH�s	
zPostgresqlDatabase._connectcCsR|j}|jdk	r.|jjr.|jjS|jrNd|j|jjfSdS)NFz	%s_%s_seq)rzr�r�r�r�r�)r?rrIr>r>rA�_get_pk_sequence�s
	
	z#PostgresqlDatabase._get_pk_sequencecCs�|j|�}|sdS|j}|jr;d|j}nd}|jd||f�|j�d}|j�r~|j�|S)Nz%s.r�zSELECT CURRVAL('%s"%s"')r)r�rzr�r�r8rWrY)r?r.rr�rIr��resultr>r>rAr�s		
z!PostgresqlDatabase.last_insert_id�publiccCs,d}dd�|j||f�j�D�S)NzSSELECT tablename FROM pg_catalog.pg_tables WHERE schemaname = %s ORDER BY tablenamecSsg|]\}|�qSr>r>)rr�r>r>rAr�s	z1PostgresqlDatabase.get_tables.<locals>.<listcomp>)r�r)r?r�r�r>r>rArl�szPostgresqlDatabase.get_tablescs>d}|j|�d|f�}�fdd�|j�D�S)Na
            SELECT
                i.relname, idxs.indexdef, idx.indisunique,
                array_to_string(array_agg(cols.attname), ',')
            FROM pg_catalog.pg_class AS t
            INNER JOIN pg_catalog.pg_index AS idx ON t.oid = idx.indrelid
            INNER JOIN pg_catalog.pg_class AS i ON idx.indexrelid = i.oid
            INNER JOIN pg_catalog.pg_indexes AS idxs ON
                (idxs.tablename = t.relname AND idxs.indexname = i.relname)
            LEFT OUTER JOIN pg_catalog.pg_attribute AS cols ON
                (cols.attrelid = t.oid AND cols.attnum = ANY(idx.indkey))
            WHERE t.relname = %s AND t.relkind = %s AND idxs.schemaname = %s
            GROUP BY i.relname, idxs.indexdef, idx.indisunique
            ORDER BY idx.indisunique DESC, i.relname;r�csAg|]7}t|d|d|djd�|d���qS)rr�rM�,rL)r!r�)rr6)rr>rAr�s	z2PostgresqlDatabase.get_indexes.<locals>.<listcomp>)r�r)r?rr�r�r.r>)rrArm�szPostgresqlDatabase.get_indexescsVd}|j|�|f�}t|j�|�����fdd�|j�D�S)Nz�
            SELECT column_name, is_nullable, data_type
            FROM information_schema.columns
            WHERE table_name = %s AND table_schema = %s
            ORDER BY ordinal_positioncs=g|]3\}}}t|||dk|�k���qS)�YES)r")rr�r�r�)�pksrr>rAr�s	z2PostgresqlDatabase.get_columns.<locals>.<listcomp>)r�r�ror)r?rr�r�r.r>)r�rrArn�s
zPostgresqlDatabase.get_columnscCs8d}|j|d||f�}dd�|j�D�S)Na�
            SELECT kc.column_name
            FROM information_schema.table_constraints AS tc
            INNER JOIN information_schema.key_column_usage AS kc ON (
                tc.table_name = kc.table_name AND
                tc.table_schema = kc.table_schema AND
                tc.constraint_name = kc.constraint_name)
            WHERE
                tc.constraint_type = %s AND
                tc.table_name = %s AND
                tc.table_schema = %szPRIMARY KEYcSsg|]\}|�qSr>r>)rr6r>r>rAr�s	z7PostgresqlDatabase.get_primary_keys.<locals>.<listcomp>)r�r)r?rr�r�r.r>r>rAro�sz#PostgresqlDatabase.get_primary_keyscs;d}|j|�|f�}�fdd�|j�D�S)Na�
            SELECT
                kcu.column_name, ccu.table_name, ccu.column_name
            FROM information_schema.table_constraints AS tc
            JOIN information_schema.key_column_usage AS kcu
                ON (tc.constraint_name = kcu.constraint_name AND
                    tc.constraint_schema = kcu.constraint_schema)
            JOIN information_schema.constraint_column_usage AS ccu
                ON (ccu.constraint_name = tc.constraint_name AND
                    ccu.constraint_schema = tc.constraint_schema)
            WHERE
                tc.constraint_type = 'FOREIGN KEY' AND
                tc.table_name = %s AND
                tc.table_schema = %scs1g|]'}t|d|d|d���qS)rr�rL)r$)rr6)rr>rAr�s	z7PostgresqlDatabase.get_foreign_keys.<locals>.<listcomp>)r�r)r?rr�rer.r>)rrArp�sz#PostgresqlDatabase.get_foreign_keyscCs)|jd|f�}t|j�d�S)Nz�
            SELECT COUNT(*) FROM pg_class, pg_namespace
            WHERE relkind='S'
                AND pg_class.relnamespace = pg_namespace.oid
                AND relname=%sr)r�r&r8)r?r�rqr>r>rArr�sz"PostgresqlDatabase.sequence_existscGs4djdgt|��}|jd||�dS)Nr�z%szSET search_path TO %s)r�r�r�)r?�search_path�path_paramsr>r>rA�set_search_path�sz"PostgresqlDatabase.set_search_pathcCsdS)NzSELECT 0 WHERE falser>)r?r>r>rAr�szPostgresqlDatabase.get_noop_sqlcCstjS)N)r�r�)r?r>r>rAr�sz"PostgresqlDatabase.get_binary_type) rCrDrEr�r�rr�rur�r�rrmr<r�rvr�r�rwr�r�rHr�rrlrmrnrorprrr�r�r�r>r>r>rAr.psB	c@s*eZdZdZddgZdddddd	d
ddd
ddddiZdZdZd4Ze	j
de	jde	jdiZ
dZdZdZdd�Zddd �Zdd!d"�Zdd#d$�Zdd%d&�Zdd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�ZdS)5r*Tr�z	UNION ALLr&�BOOLr�r�r�zDOUBLE PRECISIONr��FLOATr�zINTEGER AUTO_INCREMENTr��LONGTEXTr�zVARCHAR(40)z%srLr
r�zLIKE BINARYr�r��`FzREPLACE INTOcKsctstd��ddddi}|j|�d|krP|jd�|d<tjd||�S)	Nz%MySQLdb or PyMySQL must be installed.�charset�utf8�use_unicodeT�password�passwdr�)�mysqlr r�rrJ)r?r�r��conn_kwargsr>r>rArHs
zMySQLDatabase._connectNcCsdd�|jd�D�S)NcSsg|]\}|�qSr>r>)rr6r>r>rAr+s	z,MySQLDatabase.get_tables.<locals>.<listcomp>zSHOW TABLES)r�)r?r�r>r>rArl*szMySQLDatabase.get_tablescs�|jd��}t��i�x\|j�D]N}|dsP�j|d��j|dg��|dj|d�q/W���fdd��D�S)NzSHOW INDEX FROM `%s`r�rLr�cs2g|](}t|d�||�k���qS)N)r!)rr�)�indexesrr�r>rAr6s	z-MySQLDatabase.get_indexes.<locals>.<listcomp>)r�r�rryrr~)r?rr�r.r6r>)r�rr�rArm-s	
zMySQLDatabase.get_indexescsPd}|j|�f�}t|j������fdd�|j�D�S)Nz�
            SELECT column_name, is_nullable, data_type
            FROM information_schema.columns
            WHERE table_name = %s AND table_schema = DATABASE()cs=g|]3\}}}t|||dk|�k���qS)r�)r")rr�r�r�)r�rr>rAr@s	z-MySQLDatabase.get_columns.<locals>.<listcomp>)r�r�ror)r?rr�rer.r>)r�rrArn9s
zMySQLDatabase.get_columnscCs*|jd|�}dd�|j�D�S)NzSHOW INDEX FROM `%s`cSs*g|] }|ddkr|d�qS)rL�PRIMARYr�r>)rr6r>r>rArEs	z2MySQLDatabase.get_primary_keys.<locals>.<listcomp>)r�r)r?rr�r.r>r>rAroCszMySQLDatabase.get_primary_keyscs8d}|j|�f�}�fdd�|j�D�S)Na@
            SELECT column_name, referenced_table_name, referenced_column_name
            FROM information_schema.key_column_usage
            WHERE table_name = %s
                AND table_schema = DATABASE()
                AND referenced_table_name IS NOT NULL
                AND referenced_column_name IS NOT NULLcs.g|]$\}}}t|||���qSr>)r$)rrAr%r&)rr>rArQs	z2MySQLDatabase.get_foreign_keys.<locals>.<listcomp>)r�r)r?rr�r�r.r>)rrArpGszMySQLDatabase.get_foreign_keyscCs%tjtt|�td�|��S)Nr�)rr{rr3)r?rr|r>r>rArTszMySQLDatabase.extract_datecCstj|t|�S)N)r�DATE_FORMAT�MYSQL_DATE_TRUNC_MAPPING)r?rr|r>r>rAr~WszMySQLDatabase.truncate_datecCstt|jj�td��S)NzVALUES (DEFAULT))rr�rzr�r6)r?r}r>r>rAr�Zsz#MySQLDatabase.default_insert_clausecCsdS)NzDO 0r>)r?r>r>rAr�_szMySQLDatabase.get_noop_sqlcCstjS)N)r�r�)r?r>r>rAr�bszMySQLDatabase.get_binary_typell����)rCrDrEr�r�rur�rmr�r<r�r�r�rvrlr�r�rHrlrmrnrorprr~r�r�r�r>r>r>rAr*s:		

c@seZdZdd�ZdS)�_callable_context_managercs%t����fdd��}|S)Nc
s���||�SWdQRXdS)Nr>)r�r�)rr?r>rAr�hsz1_callable_context_manager.__call__.<locals>.inner)r)r?rr�r>)rr?rA�__call__gs!z"_callable_context_manager.__call__N)rCrDrEr�r>r>r>rAr�fsr�c@s7eZdZddd�Zdd�Zdd�ZdS)	raTcCs||_||_d|_dS)N)r�r`rL)r?r�r`r>r>rAros		zExecutionContext.__init__c
Cst|jj�a|jj|�|jj|jj|jj�|_|jri|jj�|_|jj	�WdQRX|S)N)
r�r@r]rHr;rLr`r\�txnr*)r?r>r>rAr*ts
			zExecutionContext.__enter__cCs�|jj��|jdkr,|jj�n]z9|jrd|sN|jjd�|jj|||�Wd|jj�|jj|j�XWdQRXdS)NF)	r�r@rLr^r`r�rYr1rK)r?r,�exc_val�exc_tbr>r>rAr1s
	
zExecutionContext.__exit__N)rCrDrErr*r1r>r>r>rAransracsIeZdZd�fdd�Z�fdd�Z�fdd�Z�S)r:Tcs&tt|�j||�||_dS)N)rr:rr�)r?r�r�r`)rr>rAr�szUsing.__init__csUg|_x6|jD]+}|jj|jj�|j|j_qWtt|�j�S)N)�_origr�r~rzr�rr:r*)r?r)rr>rAr*�s
	zUsing.__enter__csStt|�j|||�x0t|j�D]\}}|j||j_q,WdS)N)rr:r1r�r�r�rzr�)r?r,r�r�r�r)rr>rAr1�szUsing.__exit__)rCrDrErr*r1r>r>)rrAr:�sc@s4eZdZdd�Zdd�Zdd�ZdS)rjcCs
||_dS)N)r�)r?r�r>r>rAr�sz_atomic.__init__cCsI|jj�dkr*|jj�|_n|jj�|_|jj�S)Nr)r�rer\�_helperrhr*)r?r>r>rAr*�sz_atomic.__enter__cCs|jj|||�S)N)r�r1)r?r,r�r�r>r>rAr1�sz_atomic.__exit__N)rCrDrErr*r1r>r>r>rArj�srjc@s^eZdZdd�Zdd�Zddd�Zddd	�Zd
d�Zdd
�ZdS)r\cCs
||_dS)N)r�)r?r�r>r>rAr�sztransaction.__init__cCs|jj�dS)N)r�rZ)r?r>r>rA�_begin�sztransaction._beginTcCs!|jj�|r|j�dS)N)r�rYr�)r?rZr>r>rArY�s
ztransaction.commitcCs!|jj�|r|j�dS)N)r�rXr�)r?rZr>r>rArX�s
ztransaction.rollbackcCsU|jj�|_|jjd�|jj�dkrA|j�|jj|�|S)NFr)r�rWr�r[rer�rc)r?r>r>rAr*�s
ztransaction.__enter__cCs�z[|r|jd�nA|jj�dkrZy|jd�Wn|jd��YnXWd|jj|j�|jj�XdS)NFr�)rXr�rerYr[r�rd)r?r,r�r�r>r>rAr1�s
ztransaction.__exit__N)	rCrDrErr�rYrXr*r1r>r>r>rAr\�sr\c@s[eZdZddd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�ZdS)rhNcCsJ||_|j�}|p+dtj�j|_|j|j�|_dS)NrX)r�r�r��uuid4rrir��
quoted_sid)r?r�ri�	_compilerr>r>rAr�s	zsavepoint.__init__cCs|jj|dd�dS)Nr�F)r�r�)r?r�r>r>rAr��szsavepoint._executecCs|jd|j�dS)NzRELEASE SAVEPOINT %s;)r�r�)r?r>r>rArY�szsavepoint.commitcCs|jd|j�dS)NzROLLBACK TO SAVEPOINT %s;)r�r�)r?r>r>rArX�szsavepoint.rollbackcCs:|jj�|_|jjd�|jd|j�|S)NFz
SAVEPOINT %s;)r�rW�_orig_autocommitr[r�r�)r?r>r>rAr*�szsavepoint.__enter__cCsXz=|r|j�n&y|j�Wn|j��YnXWd|jj|j�XdS)N)rXrYr�r[r�)r?r,r�r�r>r>rAr1�s

zsavepoint.__exit__)	rCrDrErr�rYrXr*r1r>r>r>rArh�srhcs4eZdZ�fdd�Z�fdd�Z�S)r�csR|jj�}|jdk	r6|j|_d|_n	d|_tt|�j�S)N)r�rMr��_orig_isolation_levelrr�r*)r?r5)rr>rAr*�s	zsavepoint_sqlite.__enter__csLz tt|�j|||�SWd|jdk	rG|j|jj�_XdS)N)rr�r1r�r�rMr�)r?r,r�r�)rr>rAr1s
zsavepoint_sqlite.__exit__)rCrDrEr*r1r>r>)rrAr��sr�c@sXeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)r�cCs%||_|jj|_||_dS)N)rRr}rrS)r?r.rSr>r>rArs	zFieldProxy.__init__cCst|j|j�S)N)r�rRrS)r?r>r>rAr*szFieldProxy.clone_basecCs|jj|�S)N)rSrp)r?r\r>r>rArpszFieldProxy.coercecCs|jj|�S)N)rSr�)r?r\r>r>rAr�szFieldProxy.python_valuecCs|jj|�S)N)rSr�)r?r\r>r>rAr�szFieldProxy.db_valuecCs#|dkr|jSt|j|�S)Nr})rRr�rS)r?r�r>r>rAr�szFieldProxy.__getattr__N)	rCrDrErr*rpr�r�r�r>r>r>rAr�sr�c@s[eZdZdd�Zdd�Zdd�Zddd	�Zd
d�Zdd
�ZdS)r�cCs||jd<dS)Nr})rG)r?r}r>r>rAr#szModelAlias.__init__cCs2t|j|�}t|t�r.t||�S|S)N)r�r}rQrr�)r?r�r�r>r>rAr�&s
zModelAlias.__getattr__cCstd��dS)Nz-Cannot set attributes on ModelAlias instances)r)r?r�r\r>r>rAr,szModelAlias.__setattr__Fcs;�jj}|r|jn|j}�fdd�|D�S)Ncsg|]}t�|��qSr>)r�)rr�)r?r>rAr2s	z/ModelAlias.get_proxy_fields.<locals>.<listcomp>)r}rzr��
sorted_fields)r?r��mmr�r>)r?rAr�/szModelAlias.get_proxy_fieldscGsF|s|j�}t||�}|jjrB|j|jj�}|S)N)r�r�rzrs)r?r�r�r>r>rAr-4szModelAlias.selectcKs
|j|�S)N)r})r?r�r>r>rAr�<szModelAlias.__call__N)	rCrDrErr�rr�r-r�r>r>r>rAr�"sr�c@sjeZdZdZdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)r��_keys�_itemscCsg|_g|_dS)N)r�r�)r?r>r>rArCs	z_SortedFieldList.__init__cCs|j|S)N)r�)r?r�r>r>rArXGsz_SortedFieldList.__getitem__cCs
t|j�S)N)r2r�)r?r>r>rAr3Jsz_SortedFieldList.__iter__cCsD|j}t|j|�}t|j|�}||j||�kS)N)r�rr�rr�)r?�item�kr�r�r>r>rArYMs	z_SortedFieldList.__contains__cCs|jj|j�S)N)r�r�r�)r?rr>r>rAr�Ssz_SortedFieldList.indexcCsE|j}t|j|�}|jj||�|jj||�dS)N)r�rr��insertr�)r?r�r�r�r>r>rAr�Vs	z_SortedFieldList.insertcCs'|j|�}|j|=|j|=dS)N)r�r�r�)r?r�r�r>r>rA�remove\s
z_SortedFieldList.removeN)r�r�)rCrDrErrrXr3rYr�r�r�r>r>r>rAr�@sr�c@seZdZdS)rN)rCrDrEr>r>r>rArasz	peewee.dbc
@s�eZdZddddddddddddd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Zdd�Z	dd�Z
dd�Zdddd�Zdddd�Z
dd�Zddd�ZdS)�ModelOptionsNTFc
Ks�||_|jj�|_i|_i|_i|_i|_i|_i|_	g|_
t�|_g|_
g|_t�|_g|_|dk	r�|nt|_||_||_t|p�g�|_||_||_||_|	|_|
|_||_||_d|_d|_ i|_!i|_"x*|
j#�D]\}}t$|||�q?Wt|
j%��|_&|jr�|jr�|j|�|_dS)NF)'r}rCr
r�r�rr�_default_by_namer
r�_default_callable_listr��_sorted_field_listr��sorted_field_namesr�rr��default_databaser�r��
db_table_funcr�r�rsr�r�r�r�rF�only_save_dirtyr�r�rHr{rZr$r��_additional_keys)r?r'r�r�r�r�rsr�r�r�r�rFr�r�r�r\r>r>rArisD																								zModelOptions.__init__cCsd|jj|jfS)Nz<%s: %s>)rrCr�)r?r>r>rArV�szModelOptions.__repr__cCs�|jr�g}x�|jD]�}t|t�rV|jdkrCdnd}||j}|j|jd�}|jd�r�|j|j	��q|j|j
��qW||_dS)Nr3r�r�)rsrQrr#r�r��lstriprCr~r4r2)r?�
norm_order_byr�rWrr>r>rAr`�s	
zModelOptions.preparedcCs�t|j�|_dd�|jD�|_t|jj��t|jj��Bt|jf�B|_	dd�|jD�|_
dS)NcSsg|]}|j�qSr>)r�)rr�r>r>rAr�s	z4ModelOptions._update_field_lists.<locals>.<listcomp>cSs%g|]}t|t�s|�qSr>)rQr�)rrr>r>rAr�s	)r�r�r�r�r�r�r�r|r�rr�)r?r>r>rA�_update_field_lists�s
%z ModelOptions._update_field_listscCs�|j|j�||j|j<||j|j<|jj|�|j�|jdk	r�|j|j	|<t
|j�r�|j|j|<|jj
|j|jf�n#|j|j|<|j|j|j<dS)N)�remove_fieldr�r�rr�r�r�r�r�rr?rr�r~r
r�)r?rr>r>rAr��s
zModelOptions.add_fieldcCs�||jkrdS|jj|�}|j|j=|jj|�|j�|jdk	r�|j|=|j	j|d�r�xlt
|j�D]/\}\}}||kr�|jj|�Pq�Wn)|jj|d�|j
j|jd�dS)N)r�rrr�r�r�r�r�rrr�r�r
r�r�)r?rK�originalr�r�r�r>r>rAr��s


"zModelOptions.remove_fieldcCs:|jj�}x$|jD]\}}|�||<qW|S)N)r�r�r�)r?�ddrKr�r>r>rA�get_default_dict�szModelOptions.get_default_dictcCs1y|jj|�SWntk
r,dSYnXdS)Nr�rQ)r�r�ru)r?rr>r>rA�get_field_index�s
zModelOptions.get_field_indexcs0�jr&�fdd��jjD�S�jgS)Ncsg|]}�j|�qSr>)r�)rrK)r?r>rAr�s	z7ModelOptions.get_primary_key_fields.<locals>.<listcomp>)r�r�rJ)r?r>)r?rAr��s	z#ModelOptions.get_primary_key_fieldsc	Cs�t|t�}|o"t|t�}|r1g}x�|jD]}}t|t�r;|j|kr;|dkp�|r�|j|jkp�|o�|j|jk}|r;|s�|S|j|�q;W|r�|SdS)N)	rQrrr�rrr�rr~)	r?rrD�multi�is_field�is_noder�r�is_matchr>r>rAr��szModelOptions.rel_for_modelcCs|jj|j||�S)N)rzr�r})r?rrDr�r>r>rAr��sz"ModelOptions.reverse_rel_for_modelcCs|j|�p|j|�S)N)r�r�)r?rr>r>rAr��szModelOptions.rel_existscCs�g}|jg}x�|r�|j�}||kr6q|j|�x*|jjj�D]}|j|j�qVW|rx*|jjj�D]}|j|j�q�WqW|S)N)r}rr~rzrHr|rr{)r?�backrefsr�rUrr�r>r>rA�related_models�s	
zModelOptions.related_models)rCrDrErrVr`r�r�r�r�r�r�r�r�r�r�r>r>r>rAr�hs 			(	r�c
sUeZdZeddddddddd	g	�Z�fd
d�Zdd
�Z�S)�	BaseModelr�r�r�r�rsr�r�rFr�cs8|tks|djtkr>tt��j�|||�Si}|jdd�}|r�x6|jj�D]%\}}|jd�sl|||<qlWt	|dd�}d}	x�|D]�}
t
|
d�s�q�t	|
d�}|	dkr�t|j�}	�j
|jB}x?|jj�D].\}}||kr||kr|||<qWxZ|
jj�D]I\}}||krsqXt|t�rX|jjsXt|j�||<qXWq�Wtt��j�|||��|jdt�}
|
�|��_d�_t�jj��j_�jjs?tjdd�jj���j_g}xv�jj�D]e\}}t|t�rU|jr�|r�td��qU|jr�||}}qU|j||f�qUWd	}|dkr|	r�|	|	j}}qt dd
�d}}nt|t!�rd}d
}|d	k	ru|j"�|�|�j_t|t �p`t#|j$��j_%|�j_&x$|D]\}}|j"�|�q|Wt
�d
�r�t'�d�fdd��d�j}d�j(i}t)|t*f|�}|�_*�jj+�t
�d�r'�j,�t-j.���S)Nr�Metar�r�rz�model_options_basez[^\w]+zprimary key is overdetermined.FTr��_composite_key�__unicode__rVcsd�j|j�fS)Nz<%s: %r>)rCr�)r?)r'r>rArSbsz#BaseModel.__new__.<locals>.<lambda>z%sDoesNotExistrD�validate_model)/rHrCrr��__new__rrGrZrCr�r�rr��inheritabler�rQr�rr�r�rzr�r�r�r�r�r�r
rrur~r�r0rrr&r�r�r�r$rDr�rr`r�rr)r'r��bases�attrs�meta_optionsrIr��v�model_pk�	parent_pk�b�	base_meta�all_inheritable�ModelOptionsBaser�r��pk_namer�r�exc_name�	exc_attrs�exception_class)r)r'rAr�s�
!	$	
	


zBaseModel.__new__cCst|j��S)N)r2r-)r?r>r>rAr3rszBaseModel.__iter__)rCrDrEr�r�r�r3r>r>)rrAr�
s
`r�c@s�eZdZdd�Zedd��Zedd��Zeddd	��Zedd
d��Zedd
d��Z	edd��Z
edd��Zedd��Zedd��Z
edd��Zedd��Zedd��Zedd��Zedd ��Zed!d"d#��Zed$d%��Zed&d'��Zed(d)��Zed!d*d+��Zed,d-��Zed!d!d.d/��Zed!d!d0d1��Zed2d3��Zed4d5��Zd6d7�ZeZd8d9�ZeZ d:d;�Z!d<d=�Z"d>d?�Z#d@dA�Z$dBdC�Z%d!ddDdE�Z&dFdG�Z'e(dHdI��Z)d!dJdK�Z*d!d!dLdM�Z+dNdO�Z,dPdQ�Z-dRdS�Z.dS)Tr)cOs^|jj�|_t|j�|_i|_x*|j�D]\}}t|||�q:WdS)N)rzr�r�r�r�r(rZr$)r?r�r�r�r�r>r>rArvs
	zModel.__init__cCs
t|�S)N)r�)r'r>r>rAr.~szModel.aliascGs4t||�}|jjr0|j|jj�}|S)N)r�rzrs)r'r�r�r>r>rAr-�szModel.selectNcs<|p	i}|j��fdd��D��t�|�S)Ncs*g|] }�jj|�|f�qSr>)rzr�)rr�)r'r�r>rAr�s	z Model.update.<locals>.<listcomp>)r�r)r'�_Model__datar��fdictr>)r'r�rAr��s#zModel.updatecs<|p	i}|j��fdd��D��t�|�S)Ncs*g|] }�jj|�|f�qSr>)rzr�)rr�)r'r�r>rAr�s	z Model.insert.<locals>.<listcomp>)r�r)r'rr�rr>)r'r�rAr��s#zModel.insertTcCst|d|d|�S)Nr	r
)r)r'r	r
r>r>rAr�szModel.insert_manycCst|d|d|�S)Nr�r�)r)r'r�r�r>r>rA�insert_from�szModel.insert_fromcCs
t|�S)N)r )r'r>r>rAr"�szModel.deletecGst|||�S)N)r�)r'rerfr>r>rAr��sz	Model.rawcKs*||�}|jdd�|j�|S)N�force_insertT)�saverH)r'r�r+r>r>rA�create�s
zModel.createcOsF|j�j�}|r'|j|�}|r<|j|�}|j�S)N)r-r�r.rir�)r'r�r��sqr>r>rAr��sz	Model.getcKsd|jdi�}|j�}x\|j�D]N\}}d|kr[|j||i�}q+|jt||�|k�}q+Wy|j�dfSWn�|jk
r_y]tdd�|j�D��}|j	|�|j
jj��|j
|�dfSWdQRXWnWtk
rZ}z7y|j�dfSWn|jk
rG|�YnXWYdd}~XnXYnXdS)Nrr�Fcss-|]#\}}d|kr||fVqdS)r�Nr>)rr�r�r>r>rAr�s	z&Model.get_or_create.<locals>.<genexpr>T)rr-rZrir.r�r�rr�r�rzr�rkrr")r'r�rr�rr\rf�excr>r>rA�
get_or_create�s&"
zModel.get_or_createcKs�y1|jjj��|j|�dfSWdQRXWn|tk
r�g}xN|j�D]@\}}t||�}|js�|jrT|j	||k�qTW|j
|�dfSYnXdS)NTF)rzr�rkrr"rZr�r�r�r~r�)r'r�r�rKr\rr>r>rA�
create_or_get�s
zModel.create_or_getcOs|j�j||�S)N)r-ri)r'�dqr�r>r>rAri�szModel.filtercCsAi}|jjr"|jj|d<|jj|jjj|�kS)Nr�)rzr�r�r�rl)r'r�r>r>rA�table_exists�szModel.table_existsFcCs�|r|j�rdS|jj}|jj}|jrn|dk	rn|jrn|j|j�sn|j|j�|j|�|j	�dS)NF)
rrzr�r�rwr�rrrr�_create_indexes)r'rr�r�r>r>rAr�s
zModel.create_tablecCsdg}xW|jjD]I}|jr%qt|j|jt|t�f�}|r|j|�qW|S)N)	rzr�r�r�r�r�rQrr~)r'r�r�requires_indexr>r>rA�_fields_to_index�s	zModel._fields_to_indexcCs/tjdd�|j�D�|jjp+f�S)NcSs"g|]}|f|jf�qSr>)r�)rrr>r>rArs	z%Model._index_data.<locals>.<listcomp>)�	itertools�chainrrzr�)r'r>r>rA�_index_dataszModel._index_datacCs:x3|j�D]%\}}|jjj|||�q
WdS)N)rrzr�r)r'�
field_listr�r>r>rArszModel._create_indexescCs:x3|j�D]%\}}|jjj|||�q
WdS)N)rrzr�r)r'rrr�r>r>rA�
_drop_indexes
szModel._drop_indexescsg}�jjj�}�jj}�jjjrU|jrU|j|j|j��|j|j���x6�j	�D](}|j|j
�|g|j��qxW�jjrxO�jjD]A\}}�fdd�|D�}|j|j
�||��q�Wdd�|D�S)Ncs g|]}�jj|�qSr>)rzr�)rr�)r'r>rArs	z Model.sqlall.<locals>.<listcomp>cSsg|]\}}|�qSr>r>)rrer�r>r>rAr s	)
rzr�r�r�rwr�r~rrrrr�r�)r'�queriesr�r�rrJr�r�r>)r'rA�sqlalls& zModel.sqlallcCs|jjj|||�dS)N)rzr�r)r'rrr>r>rAr"szModel.drop_tablecCs|jjj|||�dS)N)rzr�r)r'rrr>r>rAr&szModel.truncate_tablecCs5|jjr%t|jj|jj�St|jj�S)N)rzr�rr�)r'r>r>rAr�*szModel.as_entitycOst|||�S)N)r�)r'r�r�r>r>rA�noop0sz
Model.noopcCst||jjj�S)N)r�rzr�r�)r?r>r>rArI4szModel._get_pk_valuecCs)|jjs%t||jjj|�dS)N)rzr�r$r�r�)r?r\r>r>rA�
_set_pk_value8szModel._set_pk_valuecCs|jj|j�kS)N)rzr�rI)r?r>r>rA�_pk_expr=szModel._pk_exprcCs|jj�|j�dS)N)r��clearr`)r?r>r>rArH@s
zModel._prepare_instancecCsdS)Nr>)r?r>r>rAr`DszModel.preparedcCs>i}x1|D])}|j|kr
||j||j<q
W|S)N)r�)r?r��only�new_datarr>r>rA�
_prune_fieldsGs

zModel._prune_fieldscCs�x�|jjD]y}||jkoS||koS||dkoS|jj|�dk	}|r
t||t||��|j|||<q
WdS)N)rzrHr�r(r�r$r�r�)r?r�r��
conditionsr>r>rA�_populate_unsaved_relationsNsz!Model._populate_unsaved_relationsc	Cs�t|j�}|jjdk	r<|jj}|j�}n
d}}|ra|j||�}n?|jjr�|r�|j||j�}|s�|jj	�dS|j
|�|dk	r-|r-|jjr�x7|jD]}|j
|d�q�Wn|j
|jd�|j|�j|j��j�}nb|dkrU|j|�j�d}n:|j|�j�}|dk	r||}|j|�d}|jj	�|S)NFr�)r�r�rzr�rIrr��dirty_fieldsr�rr!r�rJrr�r�r.rr�r�r)	r?rrr��pk_field�pk_value�pk_part_namer	�pk_from_cursorr>r>rArYs>


'	

z
Model.savecCs
t|j�S)N)r&r�)r?r>r>rA�is_dirty~szModel.is_dirtycs�fdd��jjD�S)Ncs(g|]}|j�jkr|�qSr>)r�r�)rr�)r?r>rAr�s	z&Model.dirty_fields.<locals>.<listcomp>)rzr�)r?r>)r?rAr"�szModel.dirty_fieldsccs=t|�}|j�j|j��}t|�|fg}t�}x�|r8|j�\}}||kroqH|j|�x�|jjj	�D]�\}}|j
}	|j|kr�||j|j
jk}
|	j�j|
�}n||>}
|	j�j|
�}|js|r&|j|	|f�|
|fVq�WqHWdS)N)r�r-r.rr�rryrzr{rZr}rr�r�r�r�r~)r?�search_nullabler}r�rUr��klass�rel_namer�rr �subqueryr>r>rA�dependencies�s&		
	
zModel.dependenciescCs�|r�|j|�}xztt|��D]f\}}|j}|jru|ru|j|jdi�j|�j�q(|j	�j|�j�q(W|j	�j|j
��j�S)N)r,r�r�r}r�r�r�r.r�r"r)r?�	recursive�delete_nullabler,r�r�rr>r>rA�delete_instance�s	(zModel.delete_instancecCst|j|j�f�S)N)r�rrI)r?r>r>rAr��szModel.__hash__cCs:|j|jko9|j�dk	o9|j�|j�kS)N)rrI)r?rPr>r>rAr=�szModel.__eq__cCs||kS)Nr>)r?rPr>r>rAr>�szModel.__ne__)/rCrDrErrJr.r-r�r�rrr"r�rr�rrrirrrrrrrrrr�rrI�get_idr�set_idrrHr`rr!rr'r�r"r,r/r�r=r>r>r>r>rAr)usb	

%cCs |j�}d|_|_|S)N)r�r�r�)r�r>r>rA�clean_prefetch_subquery�sr2csat|�g}xKt|�D]=\}}t|t�rF|\}}nd}t|t�rwt|t�rw|j�}|j�d}}x�t	t
|d��D]�}||}	|	j}
|	j��j
j�dd�}|r�fdd�|D�}�fdd�|D�}n�j
j�dd�}|s:|r�|�ksR|dkr�Pq�W|p`|s�|rsd|nd}
td	||
f��|r�t|
��ttj�fd
d�t||�D��}|j|�}|jt||d��q|rt|
��ttj�fdd�|D��}|j|�}|jt||d��qW|S)
Nr�r�Tcs"g|]}t�|j��qSr>)r�r�)rr�)�subquery_modelr>rAr�s	z)prefetch_add_subquery.<locals>.<listcomp>cs%g|]}t�|jj��qSr>)r�r�r�)rr�)�
last_modelr>rAr�s	z	 using %sr�z1Error: unable to find foreign key for query: %s%scs)g|]\}}|�j|�>�qSr>)r-)rr�r�)�cleanedr>rAr�s	Fcs&g|]}|j�j|�>�qSr>)r�r-)rrA)r5r>rAr�s	)�PrefetchResultr�rQrLr�r3r)r-r}r�r�r�rrzr�rr2rPrNr�rMr.r~)r	�
subqueries�
fixed_queriesr�r+�target_model�fksr�r��prefetch_result�
last_query�relsr��tgt_err�exprr>)r5r4r3rA�prefetch_add_subquery�sT	

						
r@�__prefetchedr�r�rA�
rel_models�
field_to_namercsIeZdZddddd�fdd�Zdd�Zdd�Z�S)r6Ncs�|rp|r5dd�|D�}dd�|D�}n&dd�|D�}dd�|D�}tt||��}|j}tt|�j|||||||�S)NcSsg|]}|j�qSr>)r})rrr>r>rAr�s	z*PrefetchResult.__new__.<locals>.<listcomp>cSsg|]}|jj�qSr>)r�r�)rrr>r>rAr�s	cSsg|]}|j�qSr>)r)rrr>r>rAr�s	cSsg|]}|j�qSr>)r�)rrr>r>rAr�s	)r�rMr}rr6r�)r'r�r�rArBrCr�foreign_key_attrs)rr>rAr��s	zPrefetchResult.__new__c
Cs�|jr_x�|jD]E}|j|j}||f}||krt||j||�qWn�x�|jD]{\}}|j|jj}||f}|j|g�}d|j}x|D]}	t|	||�q�Wt|||�qiWdS)Nz%s_prefetch)	rAr�r�r�r$rCr�r�r:)
r?r�id_mapr�
identifierr��attname�
rel_instancesr�r+r>r>rA�populate_instance�s	

z PrefetchResult.populate_instancecCszxs|jD]h\}}|jj|j|�}||f}|jrQ|||<q
|j|g�||j|�q
WdS)N)rCr�r�r�rArr~)r?rrErrGr@r�r>r>rA�store_instance
s	
zPrefetchResult.store_instance)rCrDrEr�rIrJr>r>)rrAr6�s	
r6cGs|s
|St||�}i}i}x�t|�D]�}|j}|jrx2|jD]'}|j|g�||j|�qTWi||<||}t|j|��}	x\|j	D]Q}
|jr�|j
|
|�|	r�x)||D]}|j|
||j�q�Wq�Wq2W|j	S)N)r@r�rr�rBrr~r&r�r�rJrI)r	r7r8�deps�rel_mapr;�query_modelrrE�
has_relationsrrHr>r>rAr/s*		

	#cKs(x!t|�D]}|j|�q
WdS)z8Create tables for all given models (in the right order).N)rpr)r��create_table_kwargsr�r>r>rArt4srtcKs.x'tt|��D]}|j|�qWdS)z6Drop tables for all given models (in the right order).N)r�rpr)r��drop_table_kwargsr�r>r>rArx9srx)�rrsr�r
r�loggingrNr�r`r>rr�r1�bisectrr�collectionsrrr�ImportErrorr�r�r�	functoolsr�inspectr	�__version__�__all__r=�Handler�	getLoggerrU�
addHandlerrH�objectrK�version_info�PY2r�r+�builtinsrOrPr?r�r�rVr�r�r�rcr�r�r#r^�unicode�buffer�exec�RuntimeErrorrirkZ	pysqlite2rl�pysq3r�r�Zpsycopg2cffirm�registerr�rnr�ZMySQLdbr�ZpymysqlZplayhouse._speedupsrorprqr�r�r�r��register_adapterr�r�DATETIME_PARTSr�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�r�r�r�r�r2rrrr6r3rrjrr7r-r}rr�r�r<rrr�r�r�r�rr
r!rr4r0r�rrrr�rrr7rr;rrrrr8r9r
r'r,r/rrrRr\r!r#r-r>rErGrIrJrgrhr�r�r�r�r�r�r�rrr r!r"r$rGr'r rrr"r#r$r+r,r1r(r2�localr8r9r5r.r*r�rar:rjr\rhr�r�r�rr�r�r�r�r)r2r@rAr6r/rtrxr>r>r>rA�<module>s�
	









	"
	

															

�%	+M{&	7"� ��{R#���0	�
(<'�	�;u�`'#!�h�?2+