📁 File Manager Pro
v10.0.3 | PHP: 8.1.34
Server: Apache
2026-06-21 20:38:36
📂
/ (Root)
/
opt
/
alt
/
python38
/
lib64
/
python3.8
/
site-packages
/
__pycache__
📍 /opt/alt/python38/lib64/python3.8/site-packages/__pycache__
🔄 Refresh
✏️
Editing: peewee.cpython-38.pyc
Read Only
U S��W�� � 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;d<d=g3Z zd d>lm Z W n( ek �r� G d?d@� d@ej �Z Y nX e� dA�Z!e!�"e � � dBZ#e$fdCdD�Z%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*dL�Z4dMdK� Z5e6Z7�d]dNdO�Z8n,e'�rde9Z/e2Z1e:Z5dPdQ� Z4e;dR� ne<dS��e)�r�dTZ=dUdK� Z>ndVdK� Z>zd dWl?m@ZA W n ek �r� dZAY nX z d dlBZBW n ek �r� eAZBY nX eA�r�eAjCeBjCk�r�eAZBzd dXlDmEZE eE�F� 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 eB�r�eB�UejVe.� eB�UejWe.� eB�Uej e.� dgdhdidjdkdlgZXeYeX�ZZdmZ[dndo� Z\dpdqdrdsdtdudv�Z]e]�� Z^dwe^dk< dxe^dl< dydz� Z_d{d|� Z`G d}d~� d~e�Zaeadd�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��ZcecjdZeecjfZgecjhZid�ZjdEZkdFZld�Zmd�Znebjoebjpebjqebjrebjsebjtebjuebjvebjwebjxebjyd�� Zzd�d�� Z{d�d�� Z|d�d�� Z}G d�d3� d3e$�Z~G d�d� de$�ZG d�d�� d�e$�Z�G d�d�� d�e$�Z�G d�d7� d7e��Z�e�Z�G d�d�� d�e��Z�G d�d�� d�e��Z�e�d�Z�G d�d�� d�e��Z�G d�d.� d.e��Z�G d�d�� d�e��Z�G d�d� de��Z�G d�d�� d�e��Z�G d�d�� d�e��Z�G d�d=� d=e��Z�d�d� Z�G d�d� de��Z�G d�d�� d�e��Z�ed�dÃZ�G d�dń d�ed�dǃ�Z�G d�dɄ d�e$�Z�G d�d � d e��Z�G d�d � d e��Z�G d�d"� d"e��Z�G d�d � d e��Z�G d�d5� d5e��Z�G d�d1� d1e��Z�G d�dф d�e��Z�G d�d � d e��Z�G d�d� de��Z�G d�d� de��Z��d_d�dׄZ�G d�d� de��Z�G d�d � d e��Z�G d�d8� d8e��Z�G d�d � d e��Z�G d�d<� d<e��Z�d�dބ Z�G d�d� d�e��Z�G d�d� de��Z�G d�d� de��Z�G d�d9� d9e��Z�G d�d:� d:e��Z�G d�d� de��Z�G d�d� d�e��Z�G d�d� d�e$�Z�G d�d� d�e$�Z�G d�d � d e��Z�G d�d� de$�Z�G d�d� d�e$�Z�G d�d� d�e$�Z�G d�d� d�e��Z�G d�d�� d�e$�Z�G d�d�� d�e$�Z�G d�d�� d�e��Z�G d�d�� d�e��Z�eTdk� r*e�ZTG d�d�� d�e��Z�eRdk� rHe�ZRG d�d�� d�e��Z�eQdk� rfe�ZQG �d �d� �de��Z�e�d�d�Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d � �d e��Z��d �d � Z�G �d �d � �d e��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �deZ�G �d�d� �deZ�G �d�d� �deZ�e�d�d�Z�e�d �d �Z�e�d �d �Z�G �d �d!� �d!eɃZ�G �d"d!� d!eʃZ�G �d#d� deʃZ�G �d$d� dẽZ�G �d%d#� d#ẽZ�G �d&d$� d$eʃZ�G �d'd%� d%ẽZ�G �d(d,� d,ẽZ�G �d)d-� d-ẽZ�G �d*d2� d2ẽZ�G �d+�d,� �d,e$�Z�G �d-�d.� �d.e$�Z�G �d/�d0� �d0e�e jփZ�G �d1�d2� �d2e$�Z�G �d3d6� d6eZ�G �d4d/� d/eZ�G �d5d+� d+eZ�G �d6�d7� �d7e$�Z�G �d8�d9� �d9e܃Z�G �d:d;� d;e݃Z�G �d;�d<� �d<e܃Z�G �d=�d>� �d>e܃Z�G �d?�d@� �d@e܃Z�G �dA�dB� �dBe�Z�G �dC�dD� �dDe��Z�G �dE�dF� �dFe$�Z�eSdk� r�G �dG�dH� �dHe$�ZSG �dId� deɃZ�eB� r�eِdJ�Z�ndZ�G �dK�dL� �dLe$�Z�G �dM�dN� �dNe�Z�G �dOd*� d*e%e��Z�dP�dQ� Z�dR�dS� Z�e�dT�dU�Z�G �dV�dW� �dWe�Z�dXd0� Z�dY�dZ� Z�d[�d\� Z�dS (` � N)� bisect_left)� bisect_right)�deque)� namedtuple)� OrderedDict)�deepcopy��wraps)�isclassz2.8.4� 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 @ s e Zd Zdd� ZdS )r> c C s d S �N� )�self�recordr@ r@ �9/opt/alt/python38/lib64/python3.8/site-packages/peewee.py�emiti s zNullHandler.emitN)�__name__� __module__� __qualname__rD r@ r@ r@ rC r> h s r> �peewee�_metaclass_helper_c C s | t |fi �S r? )� _METACLASS_)�meta�baser@ r@ rC �with_metaclasss s rM � � )rN � )�Callable)�reducec C s t | t�S r? )� isinstancerQ )�cr@ r@ rC �<lambda>} � rU �printc C s t | �d��S �N�raw_unicode_escape)�bytes�encode��sr@ r@ rC rU � rV c C s |j |k r|�|��|�d S r? )� __traceback__�with_traceback)�tp�value�tbr@ r@ rC �reraise� s rc c C s t j�| � t j�d� d S )N� )�sys�stdout�writer\ r@ r@ rC �print_� s rh 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 )� microseconds�seconds�days�_M��tr@ r@ rC rU � rV c C s | � � S r? )� total_secondsro r@ r@ rC rU � rV )�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 rU � rV z"format_date_time.<locals>.<lambda>)�datetime�strptime� ValueError)ra �formats� post_process�fmtr@ r@ rC ru � s ru c sT t ���t � �g �� ���fdd�� dd� }t�|dd�D ] }� |� q:tt���S )z@Sort models topologically so that parents will precede children.c sD | �kr@| �kr@�� | � | jj�� D ]}� |j� q&��| � d S r? )�add�_meta� reverse_rel�values� model_class�append)�model� foreign_key��dfs�models�ordering�seenr@ rC r� � s z&sort_models_topologically.<locals>.dfsc S s | j j| j jfS r? )r� �name�db_table)�mr@ r@ rC rU � rV z+sort_models_topologically.<locals>.<lambda>T)�key�reverse)�set�sorted�list�reversed)r� �namesr� r@ r� rC rv � s rv c C s� | r| d dkr| S d }}t | �}||k rf| | dkrf| |d dkrf|d7 }|d7 }|d8 }q$qfq$|r�d}d}t||| �D ]N}| | dkr�|d7 }n| | dkr�|d8 }|dk r�|d7 }d}||kr� q�q�||8 }|dkr�| || � S | S )Nr �(� �))�len�range)r] �ct�i�l� unbalanced_ct�requiredr@ r@ rC rw � s6 rw )�_DictQueryResultWrapper)�_ModelQueryResultWrapper)�_SortedFieldList)�_TuplesQueryResultWrapper�year�month�day�hour�minute�second)�%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_LOOKUPS�AssertionErrorru �SQLITE_DATETIME_FORMATS�getattr�� lookup_type�datetime_string�dtr@ 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� ru r� �strftimer� r@ r@ rC �_sqlite_date_trunc& s r� c C s t �| |t j�d k S r? )�re�search�I)�regexra 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� �and�or�+�-�*�/�&�|�^�%�=�<�<=�>�>=�!=�inznot in�iszis not�like�ilike�between�regexp�||)�AND�OR�ADD�SUB�MUL�DIV�BIN_AND�BIN_OR�XOR�MOD�EQ�LT�LTE�GT�GTE�NE�IN�NOT_IN�IS�IS_NOT�LIKE�ILIKE�BETWEEN�REGEXP�CONCAT�INNERz LEFT OUTERz RIGHT OUTER�FULL)r� � LEFT_OUTER� RIGHT_OUTERr� r� � � ) �eq�lt�lte�gt�gte�ner� r� r� r� r� c C s | � � }|�|� |S r? )�copy�update)�source� overrides�mergedr@ 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 �args�kwargsr ��funcr@ rC �inner} s z returns_clone.<locals>.inner)� call_local�r 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)�NotImplementedError�typerE �rA r r r r@ rC r � s �znot_allowed.<locals>.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 � initialize�rA 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 krtd��t| j |�S )Nz Cannot use uninitialized Proxy.)r �AttributeErrorr� r� r@ r@ rC r� � s zProxy.__getattr__c s$ || j krtd��tt| ��||�S )Nz Cannot set attribute on proxy.)� __slots__r �superr3 � __setattr__�rA r� ra �� __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 dd�Zdd� Zdd� Zedd � �Z dS ) r Nc C s( g | _ |d k r$|�� | _| j�| � d S r? )�fields�lower�_rel_model_name� _unresolvedr� )rA �rel_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)� model_cls� unresolved�drr@ r@ rC �resolve� s zDeferredRelation.resolve)N) rE rF rG r� r) r r, r/ � staticmethodr4 r@ r@ r@ rC r � s c @ s e Zd Zddd�ZdS )� _CDescriptorNc C s |d k rt |j�S | S r? )�Entity�_alias�rA �instance� instance_typer@ r@ rC �__get__� s z_CDescriptor.__get__)N)rE rF rG r< r@ r@ r@ rC r6 � s r6 c @ s e Zd ZdZe� ZdZdd� Zed5dd��Z d d � Z d d � Z e d d� �Z e d6dd��Ze dd� �Ze dd� �Ze dd� �Zdd� Zdd� Zd7dd �Zeej�Zeej�Zeej�Zeej �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/�Z0eej1�Z2eej3�Z4eej5�Z6eej7�Z8eej9�Z:eej;�Z<eej=�Z>eej?�Z@eejA�ZBd#d$� ZCd%d&� ZDd8d'd(�ZEd)d*� ZFd+d,� ZGd-d.� ZHd/d0� ZId1d2� ZJd3d4� ZKdS )9�Nodez=Base-class for any part of a query which shall be composable.�nodec C s d| _ d | _d | _d | _d S �NF)�_negatedr8 �_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)�method� method_name�r �clsr� r@ rC � decorator� s z Node.extend.<locals>.decoratorr@ )rG r� r rH r@ rF 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? )rJ r@ r8 rB rA )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? �r8 )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)rA )rA �btr@ r@ rC �bind_to� s z Node.bind_toc C s d| _ d S )N�ASC�rB r r@ r@ rC �asc� s zNode.ascc C s d| _ d S )N�DESCrS r r@ r@ rC �desc s z Node.descc C s | � � S r? )rT r r@ r@ rC �__pos__ s z Node.__pos__c C s | � � S r? )rV 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? )� Expression�rA �rhs��inv�opr@ rC r s zNode._e.<locals>.innerr@ )r^ r] r r@ r\ rC �_e s zNode._eT)r] c C s$ |d krt | tjd �S t | tj|�S r? )rY �OPr� r� rZ r@ r@ rC �__eq__' s z Node.__eq__c C s$ |d krt | tjd �S t | tj|�S r? )rY r` r� r� rZ r@ r@ rC �__ne__+ s z Node.__ne__c C s t | tj|�S r? )rY r` r� rZ r@ r@ rC �in_= s zNode.in_c C s t | tj|�S r? )rY r` r� rZ r@ r@ rC �not_in? s z Node.not_inc C s |rt | tjd �S t | tjd �S r? )rY r` r� r� )rA �is_nullr@ r@ rC re A s z Node.is_nullc C s t | tjd| �S )Nz%%%s%%�rY r` r� rZ r@ r@ rC �containsE s z Node.containsc C s t | tjd| �S )Nz%s%%rf rZ r@ r@ rC � startswithG s zNode.startswithc C s t | tjd| �S )Nz%%%srf rZ r@ r@ rC �endswithI s z Node.endswithc C s t | tjt|td�|��S )Nr� )rY r` r� r r4 )rA �low�highr@ r@ rC r� K s z Node.betweenc C s t | tj|�S r? )rY r` r� )rA � expressionr@ r@ rC r� M s z Node.regexpc C s t | tj|�S r? )rY r` r� rZ r@ r@ rC �concatO s z Node.concat)NF)N)F)T)LrE rF rG r$ r6 rT � _node_typer � classmethodrI rJ r r rL rO rQ rT rV rW rX r_ r` r� �__and__r� �__or__r� �__add__r� �__sub__r� �__mul__r� �__div__� __truediv__r� �__xor__�__radd__�__rsub__�__rmul__�__rdiv__� __rtruediv__�__rand__�__ror__�__rxor__ra rb r� �__lt__r� �__le__r� �__gt__r� �__ge__r� � __lshift__r� � __rshift__r� �__mod__r� �__pow__r� �bin_andr� �bin_orrc rd re rg rh ri r� r� rm r@ r@ r@ rC r= � sp r= c s, e Zd ZdZdZ� fdd�Zdd� Z� ZS )r7 z2An unescaped SQL string, with optional parameters.�sqlc s || _ || _tt| ��� d S r? )ra �paramsr r7 r )rA ra r� r" r@ rC r V s z SQL.__init__c C s t | jf| j�� S r? )r7 ra r� r r@ r@ rC rJ [ s zSQL.clone_base�rE rF rG r$ rn r rJ r% r@ r@ r" rC r7 R s c s4 e Zd ZdZdZ� fdd�Zdd� Zdd� Z� ZS ) r7 z/A quoted-name or entity, e.g. "table"."column".�entityc s t t| ��� || _d S r? )r r7 r �path)rA r� r" r@ rC r c s zEntity.__init__c C s t | j� S r? )r7 r� r r@ r@ rC rJ g s zEntity.clone_basec C s t td | j|f �� S r? )r7 �filterr� r� r@ r@ rC r� j s zEntity.__getattr__) rE rF rG r$ rn r rJ r� r% r@ r@ r" rC r7 _ s r7 c sT e Zd ZdZdZed�Z� fdd�Zeddd��Z d d � Z dd d �Z dd� Z � Z S )�Funcz An arbitrary SQL function call.r )�count�sumc s6 || _ || _|r |�� | jknd| _tt| ��� d S r? )r� � argumentsr' � _no_coerce�_coercer r� r )rA r� r� r" r@ rC r r s z Func.__init__Tc C s || _ d S r? )r� )rA �coercer@ r@ rC r� x s z Func.coercec C s t | jf| j�� }| j|_|S r? )r� r� r� r� )rA �resr@ r@ rC rJ | s zFunc.clone_baseNc C sJ t |t�r|d kr|}|d kr0t||d��� }n t|j�}t| td�|�S )N)� partition_by�order_by�OVER)rS r= �__sql__r7 r8 r )rA r� r� �windowr� r@ r@ rC �over� s � z Func.overc s � fdd�}|S )Nc s t � f| �|�S r? )r� �r r �r� r@ rC �dec� s z Func.__getattr__.<locals>.decr@ )rA r� r� r@ r� rC r� � s zFunc.__getattr__)T)NNN)rE rF rG r$ rn r� r� r r r� rJ r� r� r% r@ r@ r" rC r� m s r� c s. e Zd ZdZdZd� fdd� Zdd� Z� ZS ) rY z0A binary expression, e.g `foo + 1` or `bar < 7`.rl Fc s* t t| ��� || _|| _|| _|| _d S r? )r rY r �lhsr^ r[ �flat)rA r� r^ r[ r� r" r@ rC r � s zExpression.__init__c C s t | j| j| j| j�S r? )rY r� r^ r[ r� r r@ r@ rC rJ � s zExpression.clone_base)Fr� r@ r@ r" rC rY � s rY 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? )ra �adaptr r. r )rA ra r� r" r@ rC r � s zParam.__init__c C s t | j| j�S r? )r. ra r� r r@ r@ rC rJ � s zParam.clone_base)Nr� r@ r@ r" rC r. � s c @ s e Zd ZdZdS )� Passthrough� passthroughN)rE rF rG rn r@ r@ r@ rC r� � s r� c s4 e Zd ZdZdZdZdZ� fdd�Zdd� 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 )N�glue�parens)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 rJ � s zClause.clone_base) rE rF rG r$ rn r� r� r rJ 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 )N�w)r r= r r� r� r8 )rA r� r� r" r@ rC r � s zWindow.__init__c C sP g }| j r$|�ttd�t| j � �� | jrD|�ttd�t| j� �� tt|� �S )Nz PARTITION BY�ORDER BY)r� r� r r7 r� r� r� )rA � over_clausesr@ r@ rC r� � s ��zWindow.__sql__c C s t | j| j�S r? )r= r� r� r r@ r@ rC rJ � s zWindow.clone_base)NN)rE rF rG r r� rJ r% r@ r@ r" rC r= � s c C s t d| �S )Nz CHECK (%s)�r7 )ra r@ r@ rC r � s c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r z:A "django-style" filter expression, e.g. {'foo__eq': 'x'}.c s t t| ��� || _d S r? )r r r �query�rA r� r" r@ rC r � s z DQ.__init__c C s t f | j�S r? )r r� r r@ r@ rC rJ � s z DQ.clone_base)rE rF rG r$ r rJ r% r@ r@ r" rC r � s c s e Zd ZdZ� fdd�Z� ZS )� _StripParensrw c s t t| ��� || _d S r? )r r� r r> )rA r> r" r@ rC r � s z_StripParens.__init__)rE rF rG rn r r% r@ r@ r" rC r� � s r� � JoinMetadata� � src_model� dest_model�src�destr� � primary_keyr� � is_backrefrO � is_self_join� is_expressionc @ s: e Zd Zd dd�Zdd� Zdd� Zdd � Zed d � �ZdS ) �JoinNc C sX t |t�st |t�rdS |j�||�}|d k r6|dfS |j�||�}|d k rT|dfS dS )N�NNFT)rS � SelectQueryr� � rel_for_model�reverse_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 |t�r|jS t |t�r |jS |S r? )rS � ModelAliasr� r� )rA �model_or_aliasr@ r@ rC �model_from_alias s zJoin.model_from_aliasc C sB | � | j�}| � | j�}t| jt�r,| jjp.d }t| jttt f�}t| jt t f�rX| 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� rS �onr= r8 rY r� r7 r � FieldProxyr� r� r& r� �to_fieldr� r� r r� ) rA r� r� � join_aliasr� �on_fieldr� r� r� � target_attrr@ r@ rC �_join_metadata sJ �zJoin._join_metadatac C s t | d�s| �� | _| 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� �_Join)r� r� r� r� c @ s& e Zd Zdd� Zddd�Zdd� ZdS ) �FieldDescriptorc C s || _ | j j| _d S r? )r+ r� �att_name�rA r+ r@ r@ rC r Y s zFieldDescriptor.__init__Nc C s |d k r|j �| j�S | jS r? )�_data�getr� r+ r9 r@ r@ rC r< ] s zFieldDescriptor.__get__c C s ||j | j< |j�| j� d S r? )r� r� �_dirtyr� �rA r: ra 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� rN F)�null�index�unique� verbose_name� help_text� db_column�default�choicesr� �sequence� constraints�schemar �_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 rK r@ r@ rC rJ � s* � �zField.clone_basec C sZ || _ || _| jp| j | _| js4t�dd|��� | _|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� �sub�titler� � add_fieldrC r� r� �rA r� r� r@ r@ rC r. � s zField.add_to_classc C s | j jjS r? )r� r� �databaser r@ r@ rC � get_database� s zField.get_databasec C s | � � }| �� �� �|�S r? )� get_db_fieldr� �compiler�get_column_type)rA � 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 ra r@ r@ rC r� � s z Field.coercec C s |dkr |S | � |�S )z5Convert the python value for storage in the database.N�r� 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| j�S t | j�S r? )r7 r� r� r� r� )rA � with_tabler@ r@ rC � as_entity� s zField.as_entityc C s0 | � � }|r(td|d�tt|��f �S t|�S )z2Return the column type, e.g. VARCHAR(255) or REAL.�%s(%s)r� )r� r7 �join�map�str)rA � column_type� 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� rI )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� rn r� r rJ r. r� r� r� r� r� r r r r r r r% r@ r@ r" rC r f s: � c s. e Zd ZdZd� fdd� Z� fdd�Z� ZS )r �bareNc s$ t t| �j||� |d k r || _d S r? )r r r r� )rA r� r r r" r@ rC r � s zBareField.__init__c s t t| �jf d| ji|��S )Nr� )r r rJ r� �rA r r" r@ rC rJ � s zBareField.clone_base)N)rE rF rG r� r rJ r% r@ r@ r" rC r � s c @ s e Zd ZdZeZdS )r" �intN)rE rF rG r� r r� r@ r@ r@ rC r" � s c @ s e Zd ZdZdS )r �bigintN�rE 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� fdd�Z� 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� fdd�Z� ZS )�_AutoPrimaryKeyFieldNc s4 || j kr tdt| �|f ��tt| ��||� d S )Nz%s must be named `%s`.)� _column_namer| 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 r� 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_digits�decimal_places� auto_roundr �DefaultContext�roundingr 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 rJ r r r r" r r" r@ rC rJ 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 |d�S | jrP|d�| j }| j}|t|��j||d�S |S )Nr r )r" )r �Decimalr r r" r �quantize)rA ra �D�expr" r@ r@ rC r " s zDecimalField.db_valuec C s* |d k r&t |tj�r|S t�t|��S d S r? )rS r r# r r� r@ r@ rC r , s zDecimalField.python_value)r r� FN) rE rF rG r� r rJ r� r r r% r@ r@ r" rC r s � �utf-8c C s* t | t�r| S t | t�r"| �|�S t| �S r? )rS � unicode_type� string_type�decode)r] �encodingr@ r@ rC �coerce_to_unicode2 s r, c s> e Zd ZdZd � fdd� Z� fdd�Zdd� 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 rJ r/ r r" r@ rC rJ @ 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 |pd�S �N� �r, r� r@ r@ rC r� H s zCharField.coerce)r. ) rE rF rG r� r rJ r� r� r% r@ r@ r" rC r 9 s c s e Zd ZdZ� fdd�Z� 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 |pd�S r0 r2 r� r@ r@ rC r� W s zTextField.coerceN)rE rF rG r� r� r@ r@ r@ rC r8 T s c s4 e Zd ZdZeZ� fdd�Zdd� Zdd� Z� Z S )r �blobc s0 t |jjt�r |jj�| j� tt| ��||�S r? ) rS r� 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 r7 c s zBlobField._set_constructorc C s, t |t�r|�d�}t |t�r(| �|�S |S rX )rS r( r[ � basestringr9 r� r@ r@ rC r f s zBlobField.db_value) rE rF rG r� �binary_constructr9 r. r7 r r% r@ r@ r" rC r Z s c @ s e Zd ZdZdd� Zdd� ZdS )r< �uuidc C s6 t |tj�r|jS zt�|�jW S | Y S X d S r? )rS r< �UUID�hexr� r@ r@ rC r p s zUUIDField.db_valuec C s& t |tj�r|S |d kr d S t�|�S r? )rS 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� r� r� � extract_dater �� date_partr@ rC r� ~ s z_date_part.<locals>.decr@ )rA r� r@ r@ rC � _date_part} s rB c s. e Zd ZdZd� fdd� Z� fdd�Z� ZS )�_BaseFormattedFieldNc s$ |d k r|| _ tt| �j||� d S r? )r} r rC r )rA r} r r r" r@ rC r � s z _BaseFormattedField.__init__c s t t| �jf d| ji|��S )Nr} )r rC rJ r} r r" r@ rC rJ � s ��z _BaseFormattedField.clone_base)N)rE rF rG r} r rJ r% r@ r@ r" rC rC � s rC c @ sj e Zd ZdZdddgZdd� Zeed��Zeed��Z eed ��Z eed ��Z eed ��Z eed ��Z d S )r rz r� r� r� c C s |rt |t�rt|| j�S |S r? )rS r: ru r} r� r@ r@ rC r � s zDateTimeField.python_valuer� r� r� r� r� r� N)rE rF rG r� r} r r� rB r� r� r� r� r� r� r@ r@ r@ rC r � s � c @ sF e Zd ZdZdddgZdd� Zeed��Zeed��Z eed ��Z d S ) r �dater� r� r� c C s@ |r$t |t�r$dd� }t|| j|�S |r<t |tj�r<|�� S |S )Nc S s | � � S r? )rD rx r@ r@ rC rU � rV z(DateField.python_value.<locals>.<lambda>)rS r: ru r} rz rD �rA ra �ppr@ r@ rC r � s zDateField.python_valuer� r� r� N) rE rF rG r� r} r r� rB 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 |t�r$dd� }t|| j|�S t |tj�r8|�� S |d k r\t |tj�r\tjj| �� S |S )Nc S s | � � S r? )rG rx r@ r@ rC rU � rV z(TimeField.python_value.<locals>.<lambda>)rS r: ru r} rz rG � timedelta�minrE r@ r@ rC r � s zTimeField.python_valuer� r� r� N) rE rF rG r� r} r r� rB r� r� r� r@ r@ r@ rC r9 � s � c sF e Zd Zdd� ed�D �Z� fdd�Zdd� Zdd � Zd d � Z� Z S ) r: c C s g | ] }d | �qS )r r@ ��.0r� r@ r@ rC � <listcomp>� s zTimestampField.<listcomp>� c s� |� dd�pd| _| j| jkr<tdd�dd� | jD �� ��|� dd�pJd| _tj}| jr`|jn|j| _ | jrt|j n|j }|� d |� t t| �j||� d S ) N� resolutionr� z,TimestampField resolution must be one of: %sr� c s s | ]}t |�V qd S r? )r rJ r@ r@ rC � <genexpr>� s z*TimestampField.__init__.<locals>.<genexpr>�utcFr� )�poprN �valid_resolutionsr| r rP rz �utcfromtimestamp� fromtimestamp�_conv�utcnow�now� setdefaultr r: r )rA r r �_dt�_defaultr" r@ rC r � s � zTimestampField.__init__c C s | j dkr| jS tjS �Nr� )rN r� r r r@ r@ rC r� � s �zTimestampField.get_db_fieldc C s� |d kr d S t |tj�rn4t |tj�r<t�|j|j|j�}ntt|| j ��S | j rdt � |� � �}nt �|�� �}||jd 7 }| jdkr�|| j9 }tt|��S )Ng���ư>r� )rS rz rD r� r� r� r �roundrN rP �calendar�timegm� utctimetuplerG �mktime� timetuple� microsecond)rA ra � timestampr@ r@ rC r � s zTimestampField.db_valuec C sp |d k rlt |tttf�rl|dkr$d S | jdkrbd| j }t|| j�\}}|| }| �|�j|d�S | �|�S |S )Nr r� ri )rb )rS r r �longrN �divmodrU �replace)rA ra �ticks_to_microsecond�ticksrk r@ r@ rC r � s zTimestampField.python_value) rE rF rG r� rR r r� r r r% r@ r@ r" rC r: � s c @ s e Zd ZdZeZdS )r �boolN)rE rF rG r� ri r� r@ r@ r@ rC r s c s: e Zd ZdZ� fdd�Zdd� Zd dd�Zd d � Z� ZS ) �RelationDescriptorzEForeign-key abstraction to replace a related PK with a related model.c s || _ tt| ��|� d S r? )r- r rj 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: �rel_idr r@ r@ rC �get_object_or_id s z#RelationDescriptor.get_object_or_idNc C s |d k r| � |�S | jS r? )rm r+ r9 r@ r@ rC r< s zRelationDescriptor.__get__c C s| t || j�r2t|| jjj�|j| j< ||j| j< n8|j� | j�}||j| j< ||krj| j|jkrj|j| j= |j � | j� d S r? ) rS r- r� r+ r� r� r� r� rk r� r� r� )rA r: ra � orig_valuer@ r@ rC r� " s � zRelationDescriptor.__set__)N) rE rF rG r$ r rm r< r� r% r@ r@ r" rC rj s rj c @ s" e Zd ZdZdd� Zddd�ZdS )�ReverseRelationDescriptorz<Back-reference to expose related objects as a `SelectQuery`.c C s || _ |j| _d S r? )r+ r� r- r� r@ r@ rC r 0 s z"ReverseRelationDescriptor.__init__Nc C s. |d k r*| j �� �| jt|| jjj�k�S | S r? )r- �select�wherer+ r� r� r� r9 r@ r@ rC r< 4 s �z!ReverseRelationDescriptor.__get__)N)rE rF rG r$ r r< r@ r@ r@ rC ro . s ro c @ s* e Zd ZdZdd� Zd dd�Zdd� ZdS ) �ObjectIdDescriptorz(Gives direct access to the underlying idc C s |j | _t�|�| _d S r? )r� � attr_name�weakref�refr+ r� r@ r@ rC r < s zObjectIdDescriptor.__init__Nc C s |d k r|j �| j�S | �� S r? )r� r� rs r+ r9 r@ r@ rC r< @ s zObjectIdDescriptor.__get__c C s t || j|� d S r? )rC rs r� r@ r@ rC r� E s zObjectIdDescriptor.__set__)N)rE rF rG r$ r r< r� r@ r@ r@ rC rr : s rr c s� e Zd Zd� fdd� Z� fdd�Zdd� Zdd � Zd d � Zd d � Zdd� Z � fdd�Z � fdd�Z dd� Z dd� Z dd� Z� ZS )r Nc sr |dkr(t |ttf�s(t|t�s(td��|| _|| _t |ttf�| _|| _ || _ || _ || _ t t| �j||� d S )NrA z[Unexpected value for `rel_model`. Expected `Model`, `Proxy`, `DeferredRelation`, or "self")rS r3 r � issubclassr* � TypeErrorr- � _related_name�deferred� on_delete� on_update�extrar� r r r ) rA r- � related_namerz 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- r} rz r{ r| r� ) r r rJ r- �_get_related_namerz r{ r| r� r r" r@ rC rJ Y s ��zForeignKeyField.clone_basec C s t | | j�S r? )rj r- r r@ r@ rC �_get_descriptorc s z ForeignKeyField._get_descriptorc C s t | �S r? )rr r r@ r@ rC �_get_id_descriptorf s z"ForeignKeyField._get_id_descriptorc C s t | �S r? )ro 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)rx �callabler� r� r� r r@ r@ rC r~ l s z!ForeignKeyField._get_related_namec s� t �jt�r,� ��fdd�}�j�|� d S t �jt�rL�j�� ��� d S ��_� �_�jpfd�j �_���jkr��d7 ��j s�t � dd��� � �_ � j ��� ��� �_�jdkrj�_�jd k r�t �jt�s�t�j�j��_n �jj j�_� j j�rt� ���fdd �}�j�jj jk�r.|d � n�j�jj jk�rH|d � �� j jk�r`|d � n�� jk�rt|d � t� ���� � t� ���� � t�j�j�� � � d�_ �� j j �j< ��jj j�j< d S )Nc s | �_ ��� �� d S r? )r- r. �r- )r� r� rA r@ rC r s s z.ForeignKeyField.add_to_class.<locals>.callbackz%s_id�_idr� r� rA c s. |j d� jj�f �j�d� t| | ��d S )Nz%s.%s)r+ �backref� obj_id_name)r r� r� r} r )�msg�context�r� r� r� rA r@ rC �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) rS r- r3 r r r, r� r� r� r� r� r� r� r� r� r~ r} r� r r� r� �validate_backrefsr& r� �__dict__rC r r� r� r� �rel)rA r� r� r r� r@ r� rC r. q sR �z ForeignKeyField.add_to_classc s$ t | jt�s| j�� S tt| ��� S )zr Overridden to ensure Foreign Keys use same column type as the primary key they point to. )rS r� r1 r� r r r r" r@ rC r� � s z ForeignKeyField.get_db_fieldc s$ t | jt�s| j�� S tt| ��� S r? )rS r� r1 r� r r r r" r@ rC r� � s z ForeignKeyField.get_modifiersc C s | j �|�S r? )r� r� r� r@ r@ rC r� � s zForeignKeyField.coercec C s t || j�r|�� }| j�|�S r? )rS r- � _get_pk_valuer� r r� r@ r@ rC r � s zForeignKeyField.db_valuec C s t || j�r|S | j�|�S r? )rS r- r� r r� r@ r@ rC r � s z ForeignKeyField.python_value)NNNNN)rE rF rG r rJ r r� r� r~ r. r� r� r� r r r% r@ r@ r" rC r H s � @ c @ sF e Zd ZdZdZdd� Zdd� Zddd�Zd 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 r� r@ r@ rC r � s zCompositeKey.__init__c C s || _ || _t||| � d S r? )r� r� rC 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� �rK � field_name�r: r@ rC rL � s �z(CompositeKey.__get__.<locals>.<listcomp>)�tupler� r9 r@ r� rC r< � s �zCompositeKey.__get__c C s d S r? r@ r� r@ r@ rC r� � s zCompositeKey.__set__c s& � fdd�t � j|�D �}ttj|�S )Nc s" g | ]\}}� j jj| |k�qS r@ )r� r� r& )rK r+ ra r r@ rC rL � s �z'CompositeKey.__eq__.<locals>.<listcomp>)�zipr� rR �operator�and_)rA �other� expressionsr@ r rC ra � s �zCompositeKey.__eq__c C s t | jj| jf�S r? )r r� rE r� r r@ r@ rC r � s zCompositeKey.__hash__)N) rE rF rG r$ r� r r. r< r� ra r r@ r@ r@ rC r � s c @ sD e Zd ZdZddd�Zdd� Zddd �Zd d � Zd d � Zdd� Z dS )�AliasMaprp 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 )Nz<AliasMap: %s>�r� 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 rO r@ r@ rC r� � s z AliasMap.addc C s || j kr| �|� | j | S r? )r� r� �rA r r@ r@ rC � __getitem__� s zAliasMap.__getitem__c C s || j kS r? r� r� r@ r@ rC � __contains__ s zAliasMap.__contains__c C s. |r*|j �� D ]\}}|| kr|| j |< q| S r? )r� �items)rA � alias_mapr rO r@ r@ rC r s zAliasMap.update)r )N) rE rF rG r� r r� r� r� r� r r@ r@ r@ rC r� � s r� c 2 @ s� e Zd Zddddddddd dd d dd d d d�Zejdejdejdejdej 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&d�d/d0�Z'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� Z9d�dUdV�Z:d�dXdY�Z;d�dZd[�Z<d�d\d]�Z=d^d_� Z>d�d`da�Z?dbdc� Z@eAfddde�ZBdfdg� ZCdhdi� ZDdjdk� ZEdldm� ZFdndo� ZGd�dpdq�ZHeGdr�ZId�dtdu�ZJeGdv�ZKd�dwdx�ZLeGdy�ZMd�dzd{�ZNeGd|�ZOd}d~� ZPdd�� ZQeGd��ZRd�d�d��ZSeGd��ZTd�d�� ZUeGd��ZVd�d�� ZWeGd��ZXd.S )�� QueryCompilerr1 �BIGINT�BLOB�SMALLINT�DATE�DATETIME�DECIMAL�REAL�CHAR�INTEGER�VARCHAR�TEXT�TIME)r r r6 ri rD rz r r r3 r r r� r r- r5 rG 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_map�op_map�_op_map� get_parse_map� _parse_mapr� �_unknown_types)rA r� r� �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) rl r� r� r r� r� r+ r� � select_query�compound_select_queryrw ) �_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 r� Q s �zQueryCompiler.get_parse_mapc C s d| j || j f S )Nz%s%s%s)r� )rA r] r@ r@ rC �quoteb s zQueryCompiler.quotec C s || j kr| j | S |�� S r? )r� �upper)rA �fr@ r@ rC r� e s z QueryCompiler.get_column_typec C s | j | S r? )r� )rA �qr@ r@ rC �get_oph s zQueryCompiler.get_opc C s t |�� dd� d�S )Nc S s | d j S �Nr )r� )r� r@ r@ rC rU l rV z.QueryCompiler._sorted_fields.<locals>.<lambda>�r� )r� r� )rA � field_dictr@ r@ rC �_sorted_fieldsk s z QueryCompiler._sorted_fieldsc C s | j |gfS r? )r� �rA r> r� �convr@ r@ rC �_parse_defaultn s z QueryCompiler._parse_defaultc C s� t |jt�r|j}| �|j||�\}}| �|j||�\}}|jtjkrd|dkrd|sd|jr\dndg fS |jrndnd}||| � |j�|f } | || fS )Nz()z0 = 1z(0 = 1)�%s %s %s� (%s %s %s)) rS r� r � parse_noder[ r^ r` r� r� r� ) rA r> r� r� r� �lparamsr[ �rparams�templater� r@ r@ rC r� q s z QueryCompiler._parse_expressionc C s* |j r | �|� |j�|d �S | j|jgfS r? )r� r� ra r� 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� ra r� r� r@ r@ rC r� � s zQueryCompiler._parse_paramc C s8 |j r |p d }| �|j||�\}}d|jt|�f |fS )Nr )r� �parse_node_listr� r� rw �rA r> r� r� r� r� r@ r@ rC r� � s zQueryCompiler._parse_funcc C s2 | � |j|||j�\}}|jr*dt|� }||fS �N�(%s))r� r� r� r� rw r� r@ r@ rC r� � s � zQueryCompiler._parse_clausec C s d� t| j|j��g fS r )r r r� r� r� r@ r@ rC r� � s zQueryCompiler._parse_entityc C s |j t|j�fS r? )ra r� r� r� r@ r@ rC r� � s zQueryCompiler._parse_sqlc C s< |r(d� | �||j �| �|j�f�}n | �|j�}|g fS r )r r� r� r� )rA r> r� r� r� 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\} }}}|jjr�d| |j |f }nd| |j |f }||| fS )Nr� TFz((%s) %s (%s))r� ) r[ rn r� �alias_map_classr� �generate_select�calculate_alias_mapr� �compound_select_parenthesesr� )rA r> r� r� �csq�first_q�second_qr] �new_map�first�first_pr� �second_pr� �lp�r�rpr� r@ r@ rC r� � s* �z*QueryCompiler._parse_compound_select_queryc C sV |� � }|js6|r$t|t�r$|j}n |jjj}|f|_| � ||�\}}dt |� |fS r� ) r �_explicit_selectionrS r r� r� r� r� �_selectr� rw )rA r> r� r� r � select_fieldr� r� r@ r@ rC r� � s z!QueryCompiler._parse_select_queryc C s | � |j||�\}}t|�|fS r? )r� r> rw r� r@ r@ rC r� � s z!QueryCompiler._parse_strip_parensc C sh t |dd �}d}|| jkrP| j| |||�\}}|| jkoJ|jd koJ|d k}�nt|tttf�r|| �|||�\}}d| }n�t|t �r�| j }|r�t|t �r�|j }t|t �r�|� |�} q�|� t ||j��} n|�� } | g}n�t|�r�t|t �s�t|t��r |�� �|| �} | �| ||�\}}n@|d k �rH|� |�} | �| |d �\}}} n| �||d �\}}d}|||fS )Nrn Fr� T)r� r� r� r� rS r� r� r� r� r* r� r r� r r� r� r rv r� r rO r� �_parser� ) rA r> r� r� � node_typer� r� r� r� ra r� �_r@ r@ rC r� � s@ �� � zQueryCompiler._parsec s� | � ||� �\}}}|r6� d k r6|r6� fdd�|D �}t|t�r||jrNd| }|jrfd�|d|jf�}|jr|d�||jf�}|r�tdd� |D ��r�g }g }t|�D ] \} } t| t�r�| � | �\} } q�||fS )Nc s g | ]}� � |��qS r@ )r rJ �r� r@ rC rL � s z,QueryCompiler.parse_node.<locals>.<listcomp>zNOT %sr� �ASc s s | ]}t |t�V qd S r? )rS r= )rK �pr@ r@ rC rO s z+QueryCompiler.parse_node.<locals>.<genexpr>) r� rS r= r@ r8 r rB �any� enumerater� ) rA r> r� r� r� r� r� � clean_params� clean_sql�idxr� �csql�cparamsr@ r� rC r� � s" zQueryCompiler.parse_noder� c C sF g }g }|D ]*}| � |||�\}} |�|� |�| � q |�|�|fS r? )r� r� rI r ) rA r� r� r� r� r� r� r> �node_sql� 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� r� r� r� � table_alias�_joinsr� rS r� r= rO r )rA r� r� r� r� � joined_models�join_objr@ r@ rC r� s z!QueryCompiler.calculate_alias_mapc C s | � t|� |�S r? )r� r )rA �clausesr� r@ r@ rC � build_query* s zQueryCompiler.build_queryc C sZ g }t � }|g}|�rV|�� }||ks||kr0q|�|� || D �]}|} |j} t|jtttt f�rv|j� � � � } nl|j } | j r�|j} |j}n |j} |j}| j}|r�t| |j�}t||jj�}| j r�|| }}||k} ntd��t| t�r�| }n |�| � | �� � || �}|�� }|| jk�r2t| j| �}nt|�}|�t||td�| �� qBq|S )Nz Missing required join predicate.�ON)r� rQ r� r� rS r� rY r� r r7 r rO r� r� r� r� r� r� r� r| r= r� r r� �join_mapr7 )rA �joinsr� r� r r� r� �currr r� r� � constraintr� �fk_model�pk_model�fkr� r[ �dest_nr� �join_sqlr@ r@ rC �generate_joins- sN �z QueryCompiler.generate_joinsc C s. |j }|jj}| �||�}t|t�r0t|�g}n�|jsBtd�g}n*td�g}|jdkrl|td�t |j� g7 }t |j � }d|_ |� |td�f� |jd kr�|�|�� �|| �� n|�t|j� � |jd k r�|�td�� |�tdd � |jD �� � | �|j||�}|�r|� |� |jd k �r4|� td �|jg� |j�rT|� td �t|j� g� |j�rp|� td �|jg� |j�r�|� td �t|j� g� |jd k �s�|j�r�|j �r�|jd k �r�|jn|j }|�td| �� |jd k �r�|�td|j �� |j \} } | �r"| �rdnd} |�t| �� | � ||�S )N�SELECTzSELECT DISTINCT)TFr r� �FROM�WINDOWc S s& g | ] }t t|j�td �|�� ��qS )r� )r r7 r8 r� )rK r� r@ r@ rC rL } s ��z1QueryCompiler.generate_select.<locals>.<listcomp>�WHEREzGROUP BY�HAVINGr� zLIMIT %sz OFFSET %szFOR UPDATE NOWAITz FOR UPDATE) r� r� r� r� rS �CompoundSelectr� � _distinctr7 r� r r� r� rI �_fromr� r rO r� �_windowsr r �_where� _group_by�_having� _order_by�_limit�_offset� limit_max� _for_updater ) rA r� r� r� �dbr � select_clause� join_clauses�limit� for_update�no_wait�stmtr@ r@ rC r� b sT � z QueryCompiler.generate_selectc C s |j }| �� }|�||jj� |jr0d|j }nd}t|�|�� td�g}g }| �|j �D ]D\}}t |t t f�s~t ||jd�}|�t|jdd�tj|dd�� qZ|�t|� � |jr�|�td �|jg� |jd k r�t|j� } d | _|�td �| g� | �||�S ) Nz UPDATE OR %s�UPDATE�SET�r� F�r T)r� r r� � RETURNING)r� r� r� r� r� � _on_conflictr7 r r� �_updaterS r= r* r. r r� rY r` r� r� r rI � _returningr r� r ) rA r� r� r� � statementr r r+ ra �returning_clauser@ r@ rC �generate_update� s2 � z QueryCompiler.generate_updatec C s |dd� |D �� S )Nc S s g | ]}|j d d��qS )Fr/ �r �rK r+ r@ r@ rC rL � s z3QueryCompiler._get_field_clause.<locals>.<listcomp>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} | s�t| �� t�d�d�}d} g } |D ]4} | | } t| ttf��st| | jd�} | � | � q�|� t| � � q�|�rF|� | �|�t d�t |� g� n |j jj �rf|� |j� |j �� |j �r�|� t d �| j|�!� t d �g� n.|j"d k �r�t#|j"� }d |_$|� t d �|g� | �%||�S ) NzINSERT OR %s INTOz INSERT INTOFr� r� Tr. �VALUESr0 )r9 r� )&r� r� r� r� r� �_upsertr� � upsert_sqlr1 r7 r �_query�_fieldsr� r: r� �_rows� _iter_rowsr� �keysr� � attrgetterrS r= r* r. r r� rI r� �auto_increment�default_insert_clause�is_insert_returning�get_primary_key_fieldsr3 r r� r )rA r� r� rK r� r4 r r& � value_clauses� have_fields�row_dictr� r+ ra r5 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� r0 ) r� r7 r r rI r3 r r� r )rA r� r� r r5 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 d�t|�� �t d�|j�� t|j�� �g}|jrH|�t d|j �� |jrb|�t d|j �� t|� S )Nz FOREIGN KEY� REFERENCESz ON DELETE %sz ON UPDATE %s) r7 r� r r- r� rz 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.<locals>.innerr@ )rQ r r@ rP rC �return_parsed_node s z QueryCompiler.return_parsed_nodec C sN |p d|j j|j|jj jf }| �|�}ttd�|�� td�t|�f|j �� S )Nzfk_%s_%s_refs_%sz ALTER TABLEzADD CONSTRAINT) r� r� r� r- rO r r7 r r7 r� )rA r� r+ r � fk_clauser@ r@ rC �_create_foreign_key' s � ��z!QueryCompiler._create_foreign_keyrT Fc s� |rdnd}|j � g g }}� jrP� fdd�� jjD �}|�ttd�t|� �� � jD ]4}|�| � |�� t |t �rV|j sV|�| � |�� qV|j jr�|j jD ] }t |t�s�t|�}|�|� q�tt|�|�� t|| � �S )NzCREATE TABLE IF NOT EXISTSz CREATE TABLEc s g | ]}� j | �� �qS r@ )r& r �rK r� �rK r@ rC rL ; s �z/QueryCompiler._create_table.<locals>.<listcomp>r )r� � composite_keyr� r� r� r r7 r� �declared_fieldsrM rS r ry rO r� r= r ) rA r� �safer4 �columnsr� �pk_colsr+ r r@ rV rC � _create_table5 s2 � � �zQueryCompiler._create_tabler\ c C s6 |rdnd}t |�|�� g}|r.|�t d�� t|� S )NzDROP TABLE IF EXISTSz DROP TABLE�CASCADE�r7 r r� r )rA r� � fail_silently�cascader4 r r@ r@ rC � _drop_tableP s zQueryCompiler._drop_tablera 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_tablerc c C sR d|d� |�f }t|�dkrNt�|�d���� }d|d d� |d d� f }|S )Nz%s_%sr� �@ r' �7 � )r r� �hashlib�md5r[ � hexdigest)rA �tablerZ r� � 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� rU r@ r@ rC rL l s z/QueryCompiler._create_index.<locals>.<listcomp>r c S s g | ] }|� � �qS r@ r7 r8 r@ r@ rC rL r s )r� r� rl r r7 r7 r r� )rA r� r& r� r| �tbl_namer4 rl r@ r@ rC � _create_indexi s ��zQueryCompiler._create_indexro c C s< |j j}|rdnd}| �|dd� |D ��}tt|�t|��S )NzDROP INDEX IF EXISTSz DROP INDEXc S s g | ] }|j �qS r@ rm rU r@ r@ rC rL y s z-QueryCompiler._drop_index.<locals>.<listcomp>)r� r� rl r r7 r7 )rA r� r& r_ rn r4 rl r@ r@ rC � _drop_indexv s zQueryCompiler._drop_indexrp c C s t td�t|��S )NzCREATE SEQUENCE�r r7 r7 �rA � sequence_namer@ r@ rC �_create_sequence} s z QueryCompiler._create_sequencert c C s t td�t|��S )Nz DROP SEQUENCErq rr r@ r@ rC �_drop_sequence� s z QueryCompiler._drop_sequenceru )r� r� NN)NN)Nr� )N)N)N)N)F)FF)FF)F)YrE rF rG 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� 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� r6 r� r: rK rL rM rO rR rT �create_foreign_keyr\ � create_tablera � drop_tablerc �truncate_tablerl ro � create_indexrp � drop_indexrt �create_sequenceru � drop_sequencer@ r@ r@ rC r� s � � � � ' 5 ? > � r� c @ s e Zd Zddd�ZdS )�SqliteQueryCompilerFc C s |� � �� S r? )�deleter� )rA r� rb r` r@ r@ rC ry � s z"SqliteQueryCompiler.truncate_tableN)FF)rE rF rG ry 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� _populated�iterater� � StopIterationr� r@ r@ rC �next� s zResultIterator.nextN)rE rF rG r r� �__next__r@ r@ r@ rC r� � s r� c @ sd e Zd ZdZddd�Zdd� Zedd� �Zd d � Zd d � Z d d� Z dd� Z dd� Z e Z 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 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� rK r@ r@ rC r � s zQueryResultWrapper.__init__c C s | j rt| j�S 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 r� � s zQueryResultWrapper.countc C s | j S r? )r� 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 �� t�n| 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 rK 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 )N�Infr z"Negative values are not supported.)r r| r� r� r� r� r� �rA �nr@ r@ rC r� � s z QueryResultWrapper.fill_cache)N)N)rE rF rG r$ r r� r� r� 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| j�D ](\}}| �||�s0| �|| d |� q0||krld S nd}t||�D ]}| �|| d |� qvd S r� )r� r� r� r� �_initialize_node�_initialize_by_namer� )rA r� �n_colsr� �n_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� r� rZ r� r� r� r )rA r� r� � model_colsr+ r@ r@ rC r� s z)ExtQueryResultWrapper._initialize_by_namec C s� t |t�r*| j�||jp |j|jf� dS t |t�r�t|j �r�|j d }t |t�r�|jpb|jpb|j}|j rp|jprd }| j�|||f� dS dS )NTr F) rS r r� r� r8 r� r r� r� r� r� )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 � fdd�t|�D ��S )Nc s8 g | ]0\}}� j | d dkr"|n� j | d |��qS )rN Nr� )rK r� �colr r@ rC rL s �z8TuplesQueryResultWrapper.process_row.<locals>.<listcomp>)r� r� r� r@ r rC r� s �z$TuplesQueryResultWrapper.process_rowN�rE 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� rC �_prepare_instance)rA r� r: r� �columnr� r@ r@ rC r� ! s &z#NaiveQueryResultWrapper.process_rowNr� r@ r@ r@ rC r� s r� c @ s e Zd Zdd� ZdS )�DictQueryResultWrapperc C s: i }| j D ]*\}}}|d k r(||| �n|| ||< q |S r? r� )rA r� r� r� r� r� r@ r@ rC r� , s "z"DictQueryResultWrapper.process_rowNr� r@ r@ r@ rC r� + s r� 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 |t�r|V qd S r? )rS r )rK r� r@ r@ rC rO 8 s �z5ModelQueryResultWrapper.initialize.<locals>.<genexpr>)�generate_column_map� column_mapr� r� �_col_set�generate_join_list� join_list)rA r� � model_setr@ r@ rC r 6 s z"ModelQueryResultWrapper.initializec C s� g }t | jg�}t| j�D ]�\}}d }}t|t�rrt|t�rT|j}|j}|jj }n|j }}|j }|j pn|j }nL|j d kr�| j }}n |j }}t|t�r�|j r�|j }nt|t�r�|jd }|�||||f� |�|� q||fS �N���)r� r� r� r� rS r r� � _model_alias�field_instancer r� r8 r� rA r= r7 r� r� r� ) rA r� r� r� r> r� r� r� � constructorr@ r@ rC r� < 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� rQ r� r� r� r� r� r� r� r� ) rA r� r� r �stack�currentr r� � fk_present� pk_present�checkr@ r@ rC r� X s2 � �z*ModelQueryResultWrapper.generate_join_listc C s. | � |�}| �|�}|D ] }|�� q|d S r� )�construct_instances� follow_joinsr� )rA r� � collected� instancesr� r@ r@ rC r� u s z#ModelQueryResultWrapper.process_rowNc C s� i }t | j�D ]x\}\}}}}|d k r0||kr0q|| } ||krJ|� ||< || } |d krj| jj| d }t| ||d kr|| n|| �� q|S r� )r� r� r� r� rC ) rA r� rB �collected_modelsr� r� r� r� r� ra r: r@ r@ rC r� | 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 �} | s�q|j d k } | o�|j |jko�t ||j j �d k} | r�t ||j j |j|j � |j�r0| �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� rC r� r� ) rA r� �preparedr� � check_nullr� r� rK � joined_inst�has_fk�mpk�can_populate_joined_pk�can_populate_joined_fkr@ r@ rC r� � sX �� ����� z$ModelQueryResultWrapper.follow_joins)N) rE rF rG r r� r� r� r� r� r@ r@ r@ rC r� 5 s r� � JoinCache�r� r� c s8 e Zd Z� fdd�Z� fdd�Zdd� Zdd� Z� ZS ) �AggregateQueryResultWrapperc s g | _ tt| �j||� d S r? )�_rowr r� 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| j�D ]^\}\}} } }|| jk�rD| j�|g � | j| �|| f� |�|g � || �|| f� �q| jD ]D} | | jk�r��ql| j �| d�} | D ]} | j| �|| � �q��qld S )Nr� r@ )r r� r r� � all_modelsr� r� �models_with_aggregate�back_references�source_to_dest�dest_to_sourcer� r� r� r} r� r� r� rX r� r� rO �columns_to_comparer� r� r� rI )rA r� r� r� r� r� r� �key_to_columnsr� r� �col_namer� �sources� joined_modelr" r@ rC r � sL � �z&AggregateQueryResultWrapper.initializec C sD i }| j �� D ]0\}}g ||< |D ]\}}|| �|| � q"q|S r? )r� r� r� )rA r� r� r� � column_datar� r� r@ r@ rC �read_model_data� s z+AggregateQueryResultWrapper.read_model_datac C s8 | j r| j �� }n | j�� }|sDd| _t| jdd �s>| j�� t�n| 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 |�� }|| jk�r��q�| j| D �]d}z| j| |j \}}W n tk �r� Y �q�Y nX |j�s |j�r�|| �� D ]}t ||g � �q |j|k�r0�q�||j � � D ]j\}}|d k�rT�q>z|| |j|j j }W n tk �r� Y �q>Y nX t||��|� |�|� �q>n`|�r |j|k�rq�|| � � D ]<\}}||j |j|j j }t ||j j |� |�|� �q�|�|j� �q��q�|D ]}|� � �q$|S )NTr� c s, � j jr$t� fdd�� j jjD ��S � �� S )Nc s g | ]}� j | �qS r@ )r� r� r� r@ rC rL s �zHAggregateQueryResultWrapper.iterate.<locals>._get_pk.<locals>.<listcomp>)r� rW r� r� r� r� r� r@ r� rC �_get_pk s �z4AggregateQueryResultWrapper.iterate.<locals>._get_pkF) r� rQ 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� rC r� r� r� )rA r� r� � identity_map� _constructed�primary_instancer� r: � model_data�cur_row�duplicate_models� cur_row_datar� �data�different_models� new_instances�all_nonera r� r� r� r r� r� �pkrK r� r@ r@ rC r� s� � �� z#AggregateQueryResultWrapper.iterate)rE rF rG r r r� r� r% r@ r@ r" rC r� � s 6r� c s� e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d � Zd d � Z d+dd�Z dd� Z e dd� �Z e dd� �Ze d,dd��Ze d-dd��Zd.dd�Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd/d)d*�Z� ZS )0�Queryz?Base class representing a database query on one or more tables.Tc s@ t t| ��� || _|jj| _d| _|| _| jg i| _d | _ d S r� ) r r� r r� r� r� r� � _query_ctxr r �rA r� r" r@ rC r v s zQuery.__init__c C s | � � \}}d| j||f S )Nr� )r� r� �rA r� r� r@ r@ rC r� � 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 r� 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� )rK �mc�jr@ r@ rC rO � s z%Query._clone_joins.<locals>.<genexpr>)�dictr r� r r@ r@ rC r� � s �zQuery._clone_joinsNc C s, t tj|�}|d kr|S |p tj}|||�S r? )rR r� r� )rA �initialr� � conjunction�reducedr@ r@ rC �_add_query_clauses� s zQuery._add_query_clausesc C sx g }|D ]j}t |t�r"|�|� qt |t�r8|�|� qt |t�rR|�|�� � qt|�rt|t �r|�|j j � q|S r? ) rS r= r� r� r� rI �get_proxy_fieldsr rv r* r� rX )rA r �accumr� r@ r@ rC �_model_shorthand� s zQuery._model_shorthandc G s | � | j|�| _d S r? )r� r �rA r� r@ r@ rC rq � s z Query.wherec G s | � | j|tj�| _d S r? )r� r r� �or_r� r@ r@ rC �orwhere� s �z Query.orwherec C s� | j }|s8t|t�p(t|�o(|j�|� }|rNtd��nt|t�rN|jj| }| j � |g � | j | � t ||||�� t|t�s�|| _ d S )Nz#A join condition must be specified.) r� rS r� r r� � rel_existsr| r: r& r rX r� r� )rA r� r� r� r� �require_join_conditionr@ r@ rC r � s � z Query.joinc C s |p| j | _dS )z"Change or reset the query context.N)r� r� r� r@ r@ rC �switch� s z Query.switchc K sN | j }| j�|g �D ]}|j|kr| S q| �|�j|fd|i|���|�S )Nr� )r� r r� r� r� r )rA �lm�rmr� � join_kwargs�ctxr r@ r@ rC � ensure_join� s zQuery.ensure_joinc C s� g }g }t tf}t|�� �D ]�\}}| j}d|kr`|�dd�d tkr`|�dd�\}}t| }ntj}|� d�D ](} t || �} t | |�rp| j }|� | � qp|� t| ||�� q ||fS )N�__r� )r ro r� r� r� �rsplit� DJANGO_MAPr` r� �splitr� rS r- r� rY ) rA �qdictr� r � relationshipr� ra r r^ �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|t�r�| � |j �\} } |�| � ttj| �} |j| _|j| _t||| � qr|�|� qrqF|j }| �� } |D ]T} t| t��r | j| j } }| }n"t| t��r.| jj| j } }| j}| �| ||�} q�| �|�S )Nc S s g | ] }|� � �qS r@ r )rK rN r@ r@ rC rL � s z Query.filter.<locals>.<listcomp>r� r[ )r= rR r� r� r r r� �popleftrS rY r� r[ r r� r r@ r8 rC r� r r r� r- ro r+ r rq )rA r r �dq_noder� �dq_joinsr �sider r� r rl r+ r� r� � field_objr@ r@ rC r� � 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 r� s z Query.sqlc C s | � � \}}| j�||| j�S r? )r� r� � execute_sql�require_commitr� r@ r@ rC �_execute s zQuery._executec C s t �d S r? r r r@ r@ rC �execute s z Query.executeFc C s6 |r| � � �� }n | �� �� }|r.|s.|d S |S d S r� )�tuplesr� r r� )rA �as_tuple�convertr� r@ r@ rC �scalar s z Query.scalar)N)NN)N)N)FF)rE rF rG r$ r r r� r r� r� r� r� r rq r� r r� r r r� r� r� r r r r% r@ r@ r" rC r� r s2 (r� c st e Zd ZdZ� fdd�Zdd� Zed�Zed�Zed�Z 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� r� 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.clone�joiningrq r� 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 r� ? s z RawQuery.sqlc C sX | j d krR| jr | j�t�}n | jr2| j�t�}n | j�t�}|| j| � � d �| _ | j S r? ) r r r� �get_result_wrapper�RESULTS_TUPLESr � RESULTS_DICTS� RESULTS_NAIVEr� r )rA �QRWr@ 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 rq r� r r r$ r� 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 )NrI Fz,"extend" is the only valid keyword argument.)rQ r| )�orig�new_valr rI r@ r@ rC � allow_extendP s r- c s� e Zd ZdZ� fdd�Z� fdd�Zdd� Zee�Zed�Z ed �Z ed �Z d d � Z d d� Z dd� Zee�Zedd� �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 edad1d2��Z dbd3d4�Z dcd5d6�Z!ddd7d8�Z"ded9d:�Z#dfd;d<�Z$d=d>� Z%d?d@� Z&dgdBdC�Z'dhdDdE�Z(dFdG� Z)dHdI� Z*dJdK� Z+dLdM� Z,dNdO� Z-dPdQ� Z.dRdS� Z/dTdU� Z0dVdW� Z1e2�r�dXdY� Z3� Z4S )ir� r� 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_selectr �_SelectQuery__selectr r r r r r! r" r r$ �_naiver r �_aggregate_rowsr8 r )rA r� � selectionr" r@ rC r \ s$ zSelectQuery.__init__c s t t| ��|�}| j|_t| j�|_| jd k rhg |_| jD ],}t|t�rZ|j� |� � � q:|j� |� q:| j d k r~t| j �|_ | j r�| j � � |_ | j d k r�t| j �|_ | jd k r�t| j�|_| j|_| j|_| j|_| j|_| j|_| j|_| j|_| j|_| j|_|S r? )r r� r� r� r� r� r rS r= r� r r r r r r! r" r r$ r0 r r r1 r8 )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� r� r� �compound_operationsr| r )rA r� � supported_ops�r� r@ rC r � s �z&SelectQuery.compound_op.<locals>.innerr@ )r� r r@ r5 rC � compound_op� s zSelectQuery.compound_op�UNION� INTERSECT�EXCEPTc C s4 | j �td���t| |@ ��d���� }| |B | S )Nr� r� )r� rp r7 �from_r� rO r� )rA r[ � wrapped_rhsr@ r@ rC rw � s �zSelectQuery.__xor__c C s t �d�| |�S )N� UNION ALL)r� �_compound_op_staticrZ r@ r@ rC � union_all� s zSelectQuery.union_allc G s, t |�dk| _|p| jjj}| �|�| _d S r� )r� r� r� r� rX r� r� �rA r2 r@ r@ rC �__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? )r� r r r@ r@ rC �group_by� s zSelectQuery.group_byc G s | � | j|�| _d S r? )r� r r� r@ r@ rC �having� s zSelectQuery.havingc O s t | jt|�f|�| _d S r? )r- r r� r r@ r@ rC r� � s zSelectQuery.order_byc O s t | jt|�f|�| _d S r? )r- r r� )rA �windowsr r@ r@ rC r� � s zSelectQuery.windowc C s || _ d S r? )r! )rA �limr@ r@ rC r( � s zSelectQuery.limitc C s || _ d S r? )r" )rA �offr@ r@ rC �offset� s zSelectQuery.offset� c C s$ |dkr|d8 }|| _ || | _d S �Nr r� )r! r" )rA �page� paginate_byr@ r@ rC �paginate� s zSelectQuery.paginateTc C s || _ d S r? )r )rA � is_distinctr@ r@ rC �distinct� s zSelectQuery.distinctFc C s ||f| _ d S r? )r$ )rA r) �nowaitr@ r@ rC r) � s zSelectQuery.for_updatec C s || _ d S r? )r0 )rA �naiver@ r@ rC rO � 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? )r1 )rA �aggregate_rowsr@ r@ rC rP � s zSelectQuery.aggregate_rowsNc C s || _ d S r? rM )rA rO r@ r@ rC rO � s zSelectQuery.aliasc C sx |d kr t �|jj��d�}| j|kr4| �| j�}n| �� }|� |j|�}|j sbdd� |j D �|_ t |j �|f |_ |S )Nr� c S s g | ] }|� � �qS r@ )rO )rK ry r@ r@ rC rL � s z(SelectQuery.annotate.<locals>.<listcomp>) r �Countr� r� rO r� r� r� r r r r� r� )rA r- � annotationr� r@ r@ rC �annotate� s zSelectQuery.annotatec C s* |d krt �td��}| �� }|g|_|S )Nr� )r rQ r7 r� r� )rA � aggregationr� r@ r@ rC � _aggregate� s zSelectQuery._aggregatec C s | � |�j|d�S )N�r )rU r )rA rT r r@ r@ rC � aggregate� s zSelectQuery.aggregatec C s4 | j s| js| js| jr$| j|d�S | jdd�p2dS )N�� clear_limitFrV r )r r r! r" � wrapped_countrW �rA rY r@ r@ rC r� s zSelectQuery.countc C sJ | � � }|rd |_|_|�� \}}d| }| jj|f|�� }|�� pHdS )Nz+SELECT COUNT(1) FROM (%s) AS wrapped_selectr )r� r! r" r� r� �rawr )rA rY r r� r� �wrapped�rqr@ r@ rC rZ s zSelectQuery.wrapped_countc C s$ | � dd�}td�g|_t|�� �S )Nr� �1)rK r7 r� ri r r� r@ r@ rC �exists s zSelectQuery.existsc C sH | � dd�}zt|�� �W S tk rB | j�d| �� ��Y nX d S )Nr� z:Instance matching query does not exist: SQL: %s PARAMS: %s)rK r� r r� r� r r� 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� r� r� r@ r@ rC �peek s zSelectQuery.peekc C s" | j |kr|| _ d| _| j|d�S )NT)r� )r! r� rb r� r@ r@ rC r� & s zSelectQuery.firstc C s | � � �| �S r? )r� r� r r@ r@ rC r� , s zSelectQuery.sqlc C sT | j }| jD ]B}t|t�r*|j |kr* dS t|t�r |jd k r |j|kr dS q dS )NFT)r� r� rS r r= rA )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| j�t�S | jr$| j�t�S | js8| jr8| �� rD| j�t �S | j rV| j�t �S | j�t �S d S r? ) r r� r% r& r r' r0 r rc r( r1 �RESULTS_AGGREGATE_MODELS�RESULTS_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� rd rg r )rA r� � 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|t�r|j}n|}|d k r:|dkr6|d nd }|�|� |j| S rH )r rS �slice�stopr� r� )rA ra r� r� r@ r@ rC r� 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__)rG )T)TF)T)T)T)T)N)N)N)NT)F)F)r� )r� )5rE rF rG rn r r� r6 r5 r= rq rp rs rw r> r/ r rp r: rA rB r� r� r( rF rK rM r) rO r r$ rP rO rS rU rW r� rZ r` r� rb r� r� rc rd rg r r� r� r� 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 r� l s zNoopSelectQuery.sqlc C s dS )Nr� r@ r r@ r@ rC rd o s z NoopSelectQuery.get_query_metac C s | j �t�S r? )r� r% r& r r@ r@ rC rg r s z#NoopSelectQuery._get_result_wrapperN)rE rF rG r� rd rg r@ r@ r@ rC ro k s ro c sP e Zd ZdZd� fdd� Z� fdd�Zddd �Zd d � Zd d � Zdd� Z � Z S )r r� Nc s( || _ || _|| _tt| ��|g � d S r? )r� r� r[ r r r )rA r� r� r� r[ r" r@ rC r x s zCompoundSelect.__init__c s, t t| ��|�}| j|_| j|_| j|_|S r? )r r r� r� r� r[ r� r" r@ rC r� ~ s z CompoundSelect._clone_attributesFc C s | j |d�S )NrX )rZ r[ r@ r@ rC r� � s zCompoundSelect.countc C s | j �� S r? )r� rd r r@ r@ rC rd � s z CompoundSelect.get_query_metac C s | j �� o| j�� S r? )r� rc r[ r r@ r@ rC rc � s zCompoundSelect.verify_naivec C s� | j r| j�t�S | jr$| j�t�S | jr6| j�t�S | jj pD| j j }| jj pZ| j j pZ| j }|sl|rl| � � rx| j�t �S | j�t�S d S r? )r r� r% r& r r' r1 re r� r r[ r0 rc r( rf )rA � has_joins�is_naiver@ r@ rC rg � s z"CompoundSelect._get_result_wrapper)NNN)F) rE rF rG rn r r� r� rd rc rg r% r@ r@ r" rC r u s r c st e Zd Z� fdd�Z� fdd�Zdd� Zeedd� ��Zeedd d ���Zeedd d ���Z dd� Z dd� Z � Z S )� _WriteQueryc s, d | _ d| _d| _d | _tt| ��|� d S r? )r3 r r r r rt r r� r" r@ rC r � s z_WriteQuery.__init__c s6 t t| ��|�}| jr2t| j�|_| j|_| j|_|S r? )r rt r� r3 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� r� r� r5 r| r )rA r r r% �rD r@ rC r � s �z-_WriteQuery.requires_returning.<locals>.innerr@ )rD r r@ ru rC �requires_returning� s z _WriteQuery.requires_returningc G s> t |�dkr |d d kr d | _n|s.| jjj}| �|�| _d S ra )r� r3 r� r� rX r� 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 | j�t�S | jr.| j�t�S | j�t�S r? )r3 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% r3 r� r r )rA ri rK r@ r@ rC � _execute_with_result_wrapper� s z(_WriteQuery._execute_with_result_wrapper)T)T) rE rF rG r r� rv r rw r r$ r% rx r% r@ r@ r" rC rt � s rt c s` e Zd Zd� fdd� Z� fdd�Zeddd��Zed�Zd d � Z d d � Z d d� Z dd� Z � Z S )� UpdateQueryNc s || _ d | _tt| ��|� d S r? )r2 r1 r ry r )rA r� r r" r@ rC r � s zUpdateQuery.__init__c s( t t| ��|�}t| j�|_| j|_|S r? )r ry r� r� r2 r1 r� r" r@ rC r� � s z UpdateQuery._clone_attributesc C s || _ d S r? �r1 �rA �actionr@ r@ rC � on_conflict� s zUpdateQuery.on_conflictr c C s | � � �| �S r? )r� r6 r r@ r@ rC r� � s zUpdateQuery.sqlc C s@ | j d k r | jd kr | �� S | jd k r,| jS | j�| �� �S d S r? �r3 r rx r� � rows_affectedr r r@ r@ rC r � s zUpdateQuery.executec C s | j jjjstd��t| �� �S )NzwUPDATE queries cannot be iterated over unless they specify a RETURNING clause, which is not supported by your database.)r� r� r� r5 r| r� r r r@ r@ rC r� � s zUpdateQuery.__iter__c C s t | �� �� �S r? rj r r@ r@ rC r� � s zUpdateQuery.iterator)N)N)rE rF rG r r� r r} r r r� r r� r� r% r@ r@ r" rC ry � s ry c s� e Zd Zd� fdd� Zdd� Z� fdd�Zed �Zed �Ze dd d ��Z e d dd��Z e d dd��Z 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_fieldsr1 )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.)r� )r+ �� valid_fieldsr@ rC �validate_field s z.InsertQuery._iter_rows.<locals>.validate_field) r� r� r� r� � _default_dict�_default_callablesr@ r r� r& r� ) rA � model_metar� �defaults� callablesrJ � field_rowr� r� r+ r@ r� rC rA 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� r1 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? rz 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 rF J s z InsertQuery.is_insert_returningc C s | � � �| �S r? )r� rK r r@ r@ rC r� 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 �id_list�last_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 | jr�tt�d�|�� �S dS d S )Nc S s g | ]\}}|� |��qS r@ )r )rK r+ r� r@ r@ rC rL w s �z'InsertQuery.execute.<locals>.<listcomp>r T)r� r> r3 r� � insert_manyr� r rx r r� r� r� r� r� rG rW �last_insert_idr� r r� � itemgetter�fetchall)rA �insert_with_loopr� �pk_rowrK � clean_datar@ r@ rC r d s8 ��� � zInsertQuery.execute)NNNNF)T)N)T)rE rF rG r rA r� r r rq r r� r} r� r� rF r� r� r r% r@ r@ r" rC r� s( � r� c @ s$ e Zd Zed�Zdd� Zdd� ZdS )� DeleteQueryr c C s | � � �| �S r? )r� rL r r@ r@ rC r� � 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 r� r r@ r@ r@ rC r� � s r� � IndexMetadata)r� r� rZ r� rj �ColumnMetadata)r� � data_typer� r� rj �ForeignKeyMetadata)r� � dest_table� dest_columnrj c @ s e Zd ZdS )�PeeweeExceptionN�rE 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__� rV zExceptionWrapper.__enter__c C sH |d kr d S |j | jkrD| j|j }tr.|}n|j}t|||� |� d S r? )rE r� �PY26r rc )rA �exc_type� exc_value� traceback�new_type�exc_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� fdd�Z� ZS )�_BaseConnectionLocalc s4 t t| �jf |� d | _d| _d | _g | _g | _d S r� )r r� r � autocommit�closed�conn� 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/d0�Z7d1d2� Z8d3d4� Z9d5d6� Z:d7d8� Z;d9d:� Z<d;d<� Z=d=d>� Z>d?d@� Z?ddAdB�Z@dCdD� ZAdEdF� ZBdGdH� ZCdIdJ� ZDdKdL� ZEd�dMdN�ZFdOdP� ZGd�dQdR�ZHd�dSdT�ZId�dUdV�ZJd�dWdX�ZKd�dYdZ�ZLd[d\� ZMd�d]d^�ZNd�d_d`�ZOd�dadb�ZPd�dcdd�ZQd�dedf�ZRdgdh� ZSd�didj�ZTd�dkdl�ZUd�dmdn�ZVd�dodp�ZWdqdr� ZXdsdt� ZYdudv� ZZdwdx� Z[dydz� Z\d{d|� Z]dS )��DatabaseFr7 r8 r9 r< Tr� Nr� ) �ConstraintErrorr 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� threading�Lock� _conn_lockr� � autorollback� use_speedupsr r� r� ) rA r� � threadlocalsr� r& �opsr� r� r� r@ r@ rC r � s zDatabase.__init__c K s0 | � � s| �� |d k| _|| _| j�|� d S r? )� is_closedr� ry r� r� r )rA r� r� r@ r@ rC r� � s z Database.initc C s t | j�S 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� ry � 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� ry 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? r )rA r� r r@ r@ rC r� 2 s zDatabase._connectc C s t | j|�| _d S r? )r r� )rG r& r@ r@ rC �register_fields5 s zDatabase.register_fieldsc C s t | j|�| _d S r? )r r� )rG 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� r� rf r� r& r� r� r' r� r� re r� )rA � wrapper_typer@ r@ rC r% = s& � � � �zDatabase.get_result_wrapperc C s |j jr|jS d S r? )r� rD � lastrowid)rA r� r� r@ r@ rC r� O s zDatabase.last_insert_idc C s |j S r? )�rowcount)rA r� r@ r@ rC r S s zDatabase.rows_affectedc C s | � | j| j| j| j�S r? )�compiler_classr� r� r� r� r r@ r@ rC r� V s �zDatabase.compilerc C s | j | �� �|�� S r? )r 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 rp ) �logger�debugr� r� r r� �get_autocommitr� �rollback�commit)rA r� r� r r� r@ r@ rC r ^ 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� rQ r r@ r@ rC �pop_execution_context� s z Database.pop_execution_contextc C s t | jj�S r? )r� r� r� r r@ r@ rC �execution_context_depth� s z Database.execution_context_depthc C s t | |d�S )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� rQ r r@ r@ rC �pop_transaction� s zDatabase.pop_transactionc C s t | jj�S 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.<locals>.innerr )rA r r r@ r� rC �commit_on_success� s zDatabase.commit_on_successc C s | j s t�t| |�S r? )� savepointsr � savepoint�rA �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? r �rA r� r@ r@ rC � get_tables� s zDatabase.get_tablesc C s t �d S r? r �rA rj r� r@ r@ rC � get_indexes� s zDatabase.get_indexesc C s t �d S r? r r� r@ r@ rC � get_columns� s zDatabase.get_columnsc C s t �d S r? r r� r@ r@ rC �get_primary_keys� s zDatabase.get_primary_keysc C s t �d S r? r r� r@ r@ rC �get_foreign_keys� s zDatabase.get_foreign_keysc C s t �d S r? r )rA �seqr@ r@ rC �sequence_exists� s zDatabase.sequence_existsc C s | � � }| j|�||�� S r? )r� r rw )rA r� rY �qcr@ r@ rC rw � s zDatabase.create_tablec C s t ||d� d S )N)r_ )�create_model_tables)rA r� rY r@ r@ rC � create_tables� s zDatabase.create_tablesc sH | � � }t|ttf�s"td| ��� fdd�|D �}| j|�� ||�� S )Nz=Fields passed to "create_index" must be a list or tuple: "%s"c s& g | ] }t |t�r � jj| n|�qS r@ �rS r: r� r& rU �r� r@ rC rL � s �z)Database.create_index.<locals>.<listcomp>)r� rS r� r� r| r rz )rA r� r& r� r� �fobjsr@ r� rC rz � s � �zDatabase.create_indexc sH | � � }t|ttf�s"td| ��� fdd�|D �}| j|�� ||�� S )Nz;Fields passed to "drop_index" must be a list or tuple: "%s"c s& g | ] }t |t�r � jj| n|�qS r@ r� rU r� r@ rC rL � s �z'Database.drop_index.<locals>.<listcomp>)r� rS r� r� r| r r{ )rA r� r& rY r� r� r@ r� rC r{ � s � �zDatabase.drop_indexc C s | � � }| j|�|||�� S r? )r� r rv )rA r� r+ r r� r@ r@ rC rv � s �zDatabase.create_foreign_keyc C s" | j r | �� }| j|�|�� S d S r? )� sequencesr� r r| �rA r� r� r@ r@ rC r| � s zDatabase.create_sequencec C s | � � }| j|�|||�� S r? )r� r rx )rA r� r_ r` r� r@ r@ rC rx � s �zDatabase.drop_tablec C s t |||d� d S )N)r_ r` )�drop_model_tables)rA r� rY r` r@ r@ rC � drop_tables� s zDatabase.drop_tablesc C s | � � }| j|�|||�� S r? )r� r ry )rA r� rb r` r� r@ r@ rC ry � s �zDatabase.truncate_tablec C s" t t|��D ]}|�||� q d S r? )r� rv ry )rA r� rb r` r� r@ r@ rC �truncate_tables� s zDatabase.truncate_tablesc C s" | j r | �� }| j|�|�� S d S r? )r� r� r r} r� r@ r@ rC r} � s zDatabase.drop_sequencec C s t �t|td�|��S �Nr �r �EXTRACTr r4 �rA rA � date_fieldr@ r@ rC r? � s zDatabase.extract_datec C s t �||�S r? )r � DATE_TRUNCr r@ r@ rC � truncate_date� s zDatabase.truncate_datec C s t d�S )NzDEFAULT VALUESr� r� r@ r@ rC rE � s z Database.default_insert_clausec C s dS )NzSELECT 0 WHERE 0r@ r r@ r@ rC rq � s zDatabase.get_noop_sqlc C s t S r? )r; r r@ r@ rC r8 � 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. r� r� r3 r� � distinct_on� drop_cascader� � foreign_keysr) �for_update_nowaitr� r� r� r# r� r� �reserved_tablesr5 r� r� �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� ro 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� rw r� rz r{ rv r| rx r ry r r} r? r rE rq r8 r@ r@ r@ rC r� � s� � � � r� c s� e Zd ZeZdddd�ZdZeo(ejdkZ dZ e j de j diZd Zd*� fd d � Zd d� Zdd� Zdd� Zd+dd�Zd,dd�Zd-dd�Zd.dd�Zd/d d �Zd0d d �Zd1d d!�Zd2d"d#�Zd$d%� Z d&d'� Z d(d)� Z � Z S )3r6 r� r� )ri r r< F)rO rM � r r� �GLOBr� zINSERT OR REPLACE INTONc sF |pg | _ |�dd �}|r*| j �d|f� tt| �j|f|�|� d S )N� journal_mode)�_pragmasrQ r� r r6 r )rA r� �pragmasr r r r" r@ rC r s zSqliteDatabase.__init__c K sJ t s td��t 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 )NrA rN � date_truncr� )� _set_pragmas�create_functionr� r� r� r� r@ r@ rC r & s z SqliteDatabase._add_conn_hooksc C s< | j r8|�� }| j D ]\}}|�d||f � q|�� d S )NzPRAGMA %s = %s;)r r� r r� )rA r� r� �pragmara r@ r@ rC r , s zSqliteDatabase._set_pragmas�DEFERREDc C s | j d| dd� d S )NzBEGIN %sF�r �r )rA � lock_typer@ r@ rC r� 3 s zSqliteDatabase.beginc C s t d��d S )NzBSQLite does not support ALTER TABLE statements to add constraints.)r- )rA r� r+ r r@ r@ rC rv 6 s z!SqliteDatabase.create_foreign_keyc C s | � dd�}dd� |�� D �S )Nz<SELECT name FROM sqlite_master WHERE type = ? ORDER BY name;�rj c S s g | ] }|d �qS )r r@ �rK r� r@ r@ rC rL = s z-SqliteDatabase.get_tables.<locals>.<listcomp>�r r� )rA r� r� r@ r@ rC r� : s �zSqliteDatabase.get_tablesc s� d}| � |�df�}t|�� ��t� �| � d� �}|�� D ]*}|d }t|d �dk}|r<��|� q<i � t��D ](}| � d| �}dd� |�� D �� |< qt� ���fd d�t��D �S ) NzQSELECT name, sql FROM sqlite_master WHERE tbl_name = ? AND type = ? ORDER BY namer� zPRAGMA index_list("%s")r� rN zPRAGMA index_info("%s")c S s g | ] }|d �qS )rN r@ r" r@ r@ rC rL R s z.SqliteDatabase.get_indexes.<locals>.<listcomp>c s( g | ] }t |�| � | |�k���qS r@ �r� �rK r� �� index_columns� index_to_sqlrj �unique_indexesr@ rC rL T s ��)r r� r� r� r r� r� ) rA rj r� r� r� r� r� � is_uniquerl r@ r& rC r� ? s" �zSqliteDatabase.get_indexesc s$ | � d� �}� fdd�|�� D �S )N�PRAGMA table_info("%s")c s2 g | ]*}t |d |d |d t|d �� ��qS )r� rN rO r� )r� ri r" r! r@ rC rL _ s �z.SqliteDatabase.get_columns.<locals>.<listcomp>r# �rA rj r� r� r@ r! rC r� ] s �zSqliteDatabase.get_columnsc C s | � d| �}dd� |�� D �S )Nr+ c S s g | ]}|d r|d �qS )r� r� r@ r" r@ r@ rC rL d s z3SqliteDatabase.get_primary_keys.<locals>.<listcomp>r# r, r@ r@ rC r� b s z SqliteDatabase.get_primary_keysc s$ | � d� �}� fdd�|�� D �S )Nz PRAGMA foreign_key_list("%s")c s& g | ] }t |d |d |d � ��qS )rO rN r� �r� r" r! r@ rC rL h s �z3SqliteDatabase.get_foreign_keys.<locals>.<listcomp>r# r, r@ r! rC r� f s �z SqliteDatabase.get_foreign_keysc C s t | |�S r? )�savepoint_sqliter� r@ r@ rC r� k s zSqliteDatabase.savepointc C s t �||�S r? )r rA r r@ r@ rC r? n s zSqliteDatabase.extract_datec C s t �t| |�S r? )r r� r� r r@ r@ rC r q s z SqliteDatabase.truncate_datec C s t jS r? )r �Binaryr r@ r@ rC r8 t s z SqliteDatabase.get_binary_type)N)r )N)N)N)N)N)N)N) rE rF rG r~ r� r� r r �sqlite_version_infor� r# r` r� r� r� r= r r� r r r� rv r� r� r� r� r� r� r? r r8 r% r@ r@ r" rC r6 s: � � c @ s� e Zd ZdZdZdZdZdddddddd �ZdZdZ dZ d Z e j d iZd gZdZdZdZdZd'dd�Zdd� Zdd� Zd(dd�Zd)dd�Zd*dd�Zd+dd �Zd,d d �Zd d � Z d!d"� Z d#d$� Z d%d&� Z d S )-r/ T�BYTEA�BOOLEAN� TIMESTAMP�NUMERIC�DOUBLE PRECISION�SERIALr= )r6 ri rz r r r� r< �%s�~�userNc K sT t s td��t jf d|i|��}| jrBt�tj|� t�tj|� |rP|�|� |S )Nzpsycopg2 must be installed.r� ) �psycopg2r! r� �register_unicode� pg_extensions� register_type�UNICODE� UNICODEARRAY�set_client_encoding)rA r� r+ r r� r@ r@ rC r� � s zPostgresqlDatabase._connectc C s< |j }|jdk r |jjr |jjS |jr8d|j|jjf S d S )NFz %s_%s_seq)r� r� r� rD r� r� )rA r� rK r@ r@ rC �_get_pk_sequence� s z#PostgresqlDatabase._get_pk_sequencec C s` | � |�}|sd S |j}|jr*d|j }nd}|�d||f � |�� d }| �� r\| �� |S )Nz%s.r1 zSELECT CURRVAL('%s"%s"')r )rA r� r� r r� r� r� )rA r� r� r� rK r� �resultr@ r@ rC r� � s z!PostgresqlDatabase.last_insert_id�publicc C s d}dd� | � ||f��� D �S )NzSSELECT tablename FROM pg_catalog.pg_tables WHERE schemaname = %s ORDER BY tablenamec S s g | ] \}|�qS r@ r@ )rK r� r@ r@ rC rL � s z1PostgresqlDatabase.get_tables.<locals>.<listcomp>r# )rA r� r� r@ r@ rC r� � s z PostgresqlDatabase.get_tablesc s, d}| � |� d|f�}� fdd�|�� 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� c s2 g | ]*}t |d |d |d �d�|d � ��qS )r r� rO �,rN )r� r r" r! r@ rC rL � s �z2PostgresqlDatabase.get_indexes.<locals>.<listcomp>r# �rA rj r� r� r� r@ r! rC r� � s �z PostgresqlDatabase.get_indexesc s< d}| � |�|f�}t| ��|��� � �fdd�|�� 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_positionc s* g | ]"\}}}t |||d k|� k���qS ��YES�r� �rK r� r� r� ��pksrj r@ rC rL � s �z2PostgresqlDatabase.get_columns.<locals>.<listcomp>�r r� r� r� rE r@ rJ rC r� � s �z PostgresqlDatabase.get_columnsc C s( d}| � |d||f�}dd� |�� 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 = %sr c S s g | ] \}|�qS r@ r@ r" r@ r@ rC rL � s z7PostgresqlDatabase.get_primary_keys.<locals>.<listcomp>r# rE r@ r@ rC r� � s z#PostgresqlDatabase.get_primary_keysc s* d}| � |� |f�}� fdd�|�� 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 = %sc s& g | ] }t |d |d |d � ��qS )r r� rN r- r" r! r@ rC rL � s �z7PostgresqlDatabase.get_foreign_keys.<locals>.<listcomp>r# �rA rj r� r� r� r@ r! rC r� � s �z#PostgresqlDatabase.get_foreign_keysc C s | � d|f�}t|�� d �S )Nz� SELECT COUNT(*) FROM pg_class, pg_namespace WHERE relkind='S' AND pg_class.relnamespace = pg_namespace.oid AND relname=%sr )r ri r� )rA r� r� r@ r@ rC r� � s �z"PostgresqlDatabase.sequence_existsc G s( d� dgt|� �}| �d| |� d S )NrD r7 zSET search_path TO %s)r r� r )rA � search_path� path_paramsr@ r@ rC �set_search_path s z"PostgresqlDatabase.set_search_pathc C s dS )NzSELECT 0 WHERE falser@ r r@ r@ rC rq s z PostgresqlDatabase.get_noop_sqlc C s t jS r? )r: r/ r r@ r@ rC r8 s z"PostgresqlDatabase.get_binary_type)N)rC )rC )rC )rC )rC ) rE rF rG r. r� r r r� r) r r� r� r` r� r� r r5 r� r r; r� rA r� r� r� r� r� r� r� rP rq r8 r@ r@ r@ rC r/ w sH � � c @ s� e Zd ZdZddgZdddddd d d �ZdZd Zd Ze j de j de j diZ dZdZdZdd� Zd+dd�Zd,dd�Zd-dd �Zd.d d �Zd/d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� ZdS )0r+ Tr7 r< �BOOLr4 r5 �FLOATzINTEGER AUTO_INCREMENT�LONGTEXTz VARCHAR(40))ri r r r r� r5 r< r7 l ���� z LIKE BINARYr� r� �`Fz REPLACE INTOc K sJ t s td��ddd�}|�|� d|kr6|�d�|d< t jf d|i|��S )Nz%MySQLdb or PyMySQL must be installed.�utf8T)�charset� use_unicode�password�passwdr% )�mysqlr! r rQ r� )rA r� r � conn_kwargsr@ r@ rC r� % s � zMySQLDatabase._connectNc C s dd� | � d�D �S )Nc S s g | ] \}|�qS r@ r@ r" r@ r@ rC rL 2 s z,MySQLDatabase.get_tables.<locals>.<listcomp>z SHOW TABLESr r� r@ r@ rC r� 1 s zMySQLDatabase.get_tablesc sx | � d� �}t� �i � |�� D ]@}|d s:��|d � � �|d g � � |d �|d � q � ��fdd�� D �S )N�SHOW INDEX FROM `%s`r� rN r� c s$ g | ] }t |d � | |�k���qS r? r$ r% ��indexesrj r� r@ rC rL = s �z-MySQLDatabase.get_indexes.<locals>.<listcomp>)r r� r� r� rX r� )rA rj r� r� r� r@ r] rC r� 4 s �zMySQLDatabase.get_indexesc s8 d}| � |�f�}t| ����� � �fdd�|�� D �S )Nz� SELECT column_name, is_nullable, data_type FROM information_schema.columns WHERE table_name = %s AND table_schema = DATABASE()c s* g | ]"\}}}t |||d k|� k���qS rF rH rI rJ r@ rC rL G s �z-MySQLDatabase.get_columns.<locals>.<listcomp>rL rM r@ rJ rC r� @ s �zMySQLDatabase.get_columnsc C s | � d| �}dd� |�� D �S )Nr\ c S s g | ]}|d dkr|d �qS )rN �PRIMARYr� r@ r" r@ r@ rC rL L s z2MySQLDatabase.get_primary_keys.<locals>.<listcomp>r# r, r@ r@ rC r� J s z MySQLDatabase.get_primary_keysc s( d}| � |� f�}� fdd�|�� 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 NULLc s g | ]\}}}t |||� ��qS r@ r- )rK r� r� r� r! r@ rC rL W s �z2MySQLDatabase.get_foreign_keys.<locals>.<listcomp>r# rE r@ r! rC r� N s �z MySQLDatabase.get_foreign_keysc C s t �tt|�td�|��S r r r r@ r@ rC r? [ s zMySQLDatabase.extract_datec C s t �|t| �S r? )r � DATE_FORMAT�MYSQL_DATE_TRUNC_MAPPINGr r@ r@ rC r ^ s zMySQLDatabase.truncate_datec C s t t|jj�td��S )NzVALUES (DEFAULT))r r� r� r� r7 r� r@ r@ rC rE a s �z#MySQLDatabase.default_insert_clausec C s dS )NzDO 0r@ r r@ r@ rC rq f s zMySQLDatabase.get_noop_sqlc C s t jS r? )rZ r/ r r@ r@ rC r8 i s z MySQLDatabase.get_binary_type)N)N)N)N)N) rE rF rG r. r3 r� r) r� r# r` r� r� r� r� r� r r= r� r� r� r� r� r� r? r rE rq r8 r@ r@ r@ rC r+ sD � � c @ s e Zd Zdd� ZdS )�_callable_context_managerc s t � �� �fdd��}|S )Nc s& �� � | |�W 5 Q R � S Q R X d S r? r@ r� �r rA r@ rC r o s z1_callable_context_manager.__call__.<locals>.innerr )rA r r r@ rc rC �__call__n s z"_callable_context_manager.__call__N)rE rF rG rd r@ r@ r@ rC rb m s rb c @ s& e Zd Zd dd�Zdd� Zdd� ZdS ) r� Tc C s || _ || _d | _d S r? )r� r� r� )rA r� r� r@ r@ rC r v s zExecutionContext.__init__c C sZ | j j�H | j �| � | j j| j j f| j j�| _| jrL| j �� | _| j� � W 5 Q R X | S r? ) r� r� r� r� r� r� r� r� �txnr� r r@ r@ rC r� { s �� zExecutionContext.__enter__c C st | j j�b | jd kr | j �� nFz*| jrH|s8| j�d� | j�|||� W 5 | j �� | j �| j� X W 5 Q R X d S r? ) r� r� r� r� r� r� re r� r� �rA r� �exc_val�exc_tbr@ r@ rC r� � s zExecutionContext.__exit__N)T�rE rF rG r r� r� r@ r@ r@ rC r� u s r� c s6 e Zd Zd� fdd� Z� fdd�Z� fdd�Z� ZS ) r; Tc s t t| ��||� || _d S r? )r r; r r� )rA r� r� r� r" r@ rC r � s zUsing.__init__c s: g | _ | jD ] }| j �|jj� | j|j_q tt| ��� S r? )�_origr� r� r� r� r r; r� �rA r� r" r@ rC r� � s zUsing.__enter__c s: t t| ��|||� t| j�D ]\}}| j| |j_q d S r? )r r; r� r� r� rj r� r� )rA r� rg rh r� r� r" r@ rC r� � s zUsing.__exit__)T)rE rF rG r r� r� r% r@ r@ r" rC r; � s c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r� c C s || _ d S r? �r% �rA r% r@ r@ rC r � s z_atomic.__init__c C s2 | j �� dkr | j �� | _n | j �� | _| j�� S r� )r% r� r� �_helperr� r� r r@ r@ rC r� � s z_atomic.__enter__c C s | j �|||�S r? )rn r� rf r@ r@ rC r� � s z_atomic.__exit__Nri r@ r@ r@ rC r� � s r� c @ s@ e Zd Zdd� Zdd� Zddd�Zddd �Zd d � Zd d � ZdS )r� c C s || _ d S r? rl rm r@ r@ rC r � s ztransaction.__init__c C s | j �� d S r? )r% r� r r@ r@ rC �_begin� s ztransaction._beginTc C s | j �� |r| �� d S r? )r% r� ro �rA r� r@ r@ rC r� � s ztransaction.commitc C s | j �� |r| �� d S r? )r% r� ro rp r@ r@ rC r� � s ztransaction.rollbackc C s> | j �� | _| j �d� | j �� dkr.| �� | j �| � | S )NFr )r% r� rj r� r� ro r� r r@ r@ rC r� � s ztransaction.__enter__c C sj zJ|r| �d� n6| j �� dkrHz| �d� W n | �d� � Y nX W 5 | j �| j� | j �� X d S �NFr� )r% r� rj r� r� r� r� rf r@ r@ rC r� � s ztransaction.__exit__N)T)T) rE rF rG r ro r� r� r� r� r@ r@ r@ rC r� � s r� c @ s> e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � ZdS )r� Nc C s4 || _ |�� }|p dt�� j | _|�| j�| _d S )Nr] )r% r� r< �uuid4r> r� r� � quoted_sid)rA r% r� � _compilerr@ r@ rC r � s zsavepoint.__init__c C s | j j|dd� d S )NFr )r% r r� r@ r@ rC r � s zsavepoint._executec C s | � d| j � d S )NzRELEASE SAVEPOINT %s;�r rs r r@ r@ rC r� � s zsavepoint.commitc C s | � d| j � d S )NzROLLBACK TO SAVEPOINT %s;ru r r@ r@ rC r� � s zsavepoint.rollbackc C s, | j �� | _| j �d� | �d| j � | S )NFz SAVEPOINT %s;)r% r� �_orig_autocommitr� r rs r r@ r@ rC r� � s zsavepoint.__enter__c C sL z6|r| �� n$z | �� W n | �� � Y nX W 5 | j �| j� X d S r? )r% r� rv r� r� rf r@ r@ rC r� � s zsavepoint.__exit__)N) rE rF rG r r r� r� r� r� r@ r@ r@ rC r� � s r� c s( e Zd Z� fdd�Z� fdd�Z� ZS )r. c s8 | j �� }|jd k r$|j| _d |_nd | _tt| ��� S r? )r% r� r �_orig_isolation_levelr r. r� r� r" r@ rC r� � s zsavepoint_sqlite.__enter__c s8 ztt| ��|||�W �S | j d k r2| j | j�� _X d S r? )rw r% r� r r r. r� rf r" r@ rC r� s � zsavepoint_sqlite.__exit__)rE rF rG r� r� r% r@ r@ r" rC r. � s r. c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )r� c C s || _ | j j| _|| _d S r? )r� r� r� r� )rA rO r� r@ r@ rC r s zFieldProxy.__init__c C s t | j| j�S r? )r� r� r� r r@ r@ rC rJ s zFieldProxy.clone_basec C s | j �|�S r? )r� r� r� r@ r@ rC r� s zFieldProxy.coercec C s | j �|�S r? )r� r r� r@ r@ rC r s zFieldProxy.python_valuec C s | j �|�S r? )r� r r� r@ r@ rC r ! s zFieldProxy.db_valuec C s |dkr| j S t| j|�S �Nr� )r� r� r� r� r@ r@ rC r� $ s zFieldProxy.__getattr__N) rE rF rG r rJ r� r r r� r@ r@ r@ rC r� s r� c @ s> e Zd Zdd� Zdd� Zdd� Zddd �Zd d � Zd d � ZdS )r� c C s || j d< d S rx )r� r� r@ r@ rC r * s zModelAlias.__init__c C s$ t | j|�}t|t�r t| |�S |S r? )r� r� rS r r� )rA r� r r@ r@ rC r� - s zModelAlias.__getattr__c C s t d��d S )Nz-Cannot set attributes on ModelAlias instances)r r! r@ r@ rC r 3 s zModelAlias.__setattr__Fc s* � j j}|r|jn|j}� fdd�|D �S )Nc s g | ]}t � |��qS r@ )r� rU r r@ rC rL 9 s z/ModelAlias.get_proxy_fields.<locals>.<listcomp>)r� r� rX � sorted_fields)rA rX �mmr& r@ r rC r� 6 s zModelAlias.get_proxy_fieldsc G s4 |s | � � }t| f|�� }| jjr0|j| jj� }|S r? )r� r� r� r� )rA r2 r� r@ r@ rC rp ; s zModelAlias.selectc K s | j f |�S r? r� r r@ r@ rC rd C s zModelAlias.__call__N)F) rE rF rG r r� r r� rp rd r@ r@ r@ rC r� ) s r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )r� ��_keys�_itemsc C s g | _ g | _d S r? r{ r r@ r@ rC r J s z_SortedFieldList.__init__c C s | j | S r? )r} )rA r� r@ r@ rC r� N s z _SortedFieldList.__getitem__c C s t | j�S r? )r� r} r r@ r@ rC r� Q s z_SortedFieldList.__iter__c C s0 |j }t| j|�}t| j|�}|| j||� kS r? )r� r r| r r} )rA �item�kr� r� r@ r@ rC r� T s z _SortedFieldList.__contains__c C s | j �|j�S r? )r| r� r� r� r@ r@ rC r� Z s z_SortedFieldList.indexc C s2 |j }t| j|�}| j�||� | j�||� d S r? )r� r r| �insertr} )rA r~ r r� r@ r@ rC r� ] s z_SortedFieldList.insertc C s | � |�}| j|= | j|= d S r? )r� r} r| )rA r~ r� r@ r@ rC �removec s z_SortedFieldList.removeN) rE rF rG r r r� r� r� r� r� r� r@ r@ r@ rC r� G s r� c @ s e Zd ZdS )r Nr� r@ r@ r@ rC r h s z peewee.dbc @ s| e Zd Zd dd�Zdd� Zdd � Zd d � Zd d � Zdd� Zdd� Z dd� Z dd� Z d dd�Z d dd�Z dd� Zd!d d �ZdS )"� ModelOptionsNTFc K s || _ |j�� | _i | _i | _i | _i | _i | _i | _ g | _ t � | _ g | _ g | _t� | _g | _|d k rj|nt| _|| _|| _t|p�g �| _|| _|| _|| _| | _| | _ | | _ | | _ d | _ d| _ i | _!i | _"| �#� D ]\}}t$| ||� q�t| �%� �| _&| j�r| j�s| �|�| _d S r? )'r� rE r' r� r& rZ r� �_default_by_namer� r� �_default_callable_listr� �_sorted_field_listry �sorted_field_namesr� r� rX �default_databaser� r� � db_table_funcr� r^ r� r� r r� r� r� �only_save_dirtyrD rW r� r� r� rC rB �_additional_keys)rA rG r� r� r� r^ r� r� r r� r� r� r� r r� ra r@ r@ rC r p sD zModelOptions.__init__c C s d| j j| jf S )Nz<%s: %s>)r# rE r� r r@ r@ rC r� � s zModelOptions.__repr__c C s~ | j rzg }| j D ]b}t|t�r:|jdkr,dnd}||j }| j|�d� }|�d�rd|�|� � � q|�|� � � q|| _ d S )NrU r� r1 ) r� rS r rB r� r& �lstriprh r� rV rT )rA � norm_order_byr~ r� r+ r@ r@ rC r� � s zModelOptions.preparedc C s^ t | j�| _dd� | jD �| _t| j�� �t| j�� �B t| jf�B | _ dd� | jD �| _ d S )Nc S s g | ] }|j �qS r@ �r� rU r@ r@ rC rL � s z4ModelOptions._update_field_lists.<locals>.<listcomp>c S s g | ]}t |t�s|�qS r@ )rS r r8 r@ r@ rC rL � s �) r� r� ry r� r� r& rB r� r� r� rX r r@ r@ rC �_update_field_lists� s � �z ModelOptions._update_field_listsc C s� | � |j� || j|j< || j|j< | j�|� | �� |jd k r�|j| j |< t |j�rz|j| j |< | j � |j|jf� n|j| j|< |j| j|j< d S r? )� remove_fieldr� r& rZ r� r� r� r� r� r� r� r� r� r� r� r� r� r@ r@ rC r� � s zModelOptions.add_fieldc C s� || j krd S | j �|�}| j|j= | j�|� | �� |jd k r�| j|= | j �|d �r�t | j �D ]$\}\}}||krb| j �|� q�qbn | j �|d � | j �|jd � d S r? )r& rQ rZ r� r� r� r� r� r� r� r� r� r� r� r� )rA r� �originalr� r� r� r@ r@ rC r� � s zModelOptions.remove_fieldc C s( | j �� }| jD ]\}}|� ||< q|S r? )r� r r� )rA �ddr� r� r@ r@ rC �get_default_dict� s z ModelOptions.get_default_dictc C s* z| j �|�W S tk r$ Y dS X d S r� )r� r� r| r� r@ r@ rC �get_field_index� s z ModelOptions.get_field_indexc s$ � j r � fdd�� jjD �S � jgS )Nc s g | ]}� j | �qS r@ )r& r� r r@ rC rL � s �z7ModelOptions.get_primary_key_fields.<locals>.<listcomp>)rW r� r� r r@ r rC rG � s �z#ModelOptions.get_primary_key_fieldsc C s� t |t�}| ot |t�}|r"g }| jD ]Z}t |t�r(|j|kr(|d kpf|rX|j|jkpf|of|j|jk}|r(|sx| S |�|� q(|r�|S d S r? ) rS r r= ry r r- r� r8 r� ) rA r� r �multi�is_field�is_noder� r+ �is_matchr@ r@ rC r� � s" �� zModelOptions.rel_for_modelc C s |j �| j||�S r? )r� r� r� )rA r� r r� r@ r@ rC r� � s z"ModelOptions.reverse_rel_for_modelc C s | � |�p| �|�S r? )r� r� rk r@ r@ rC r� s zModelOptions.rel_existsc C sr g }| j g}|rn|�� }||kr"q |�|� |jj�� D ]}|�|j� q8|r |jj�� D ]}|�|j � qZq |S r? )r� rQ r� r� r� r� r- r� )rA �backrefsr� r� r� r r@ r@ rC �related_models s zModelOptions.related_models) NNNNNNNNNTF)NF)NF)F)rE rF rG r r� r� r� r� r� r� r� rG r� r� r� r� r@ r@ r@ rC r� o s, � + r� c s> e Zd Zeddddddddd g �Z� fd d �Zd d � Z� ZS )� BaseModelr� r� r� r^ r� r� r� r� r� c s� |t ks|d jt kr,tt� ��� |||�S i }|�dd �}|rf|j�� D ]\}}|�d�sJ|||< qJt |dd �}d } |D ]�} t | d�s�qzt | d�} | d kr�t | j �} � j | jB } | j�� D ] \}}|| kr�||kr�|||< q�| j�� D ]2\}}||kr�q�t|t�r�|jj s�t |j�||< q�qztt� ��� |||�� |�dt�} | � f|�� _d � _t� jj�� j_� jj�s�t�dd� j�� �� j_g }� j�� D ]P\}}t|t ��r�|j �r�|�r�t d��n"|j �r�|| }}n|� ||f� �q�d }|d k�r| �r| | j }}nt d d �d }}nt|t!��r.d }d }|d k �rn|�"� |� |� j_ t|t ��p`t#|j$�� j_%|� j_&|D ]\}}|�"� |� �qrt � d��r�t'� d� fdd�� d� j }d� j(i}t)|t*f|�}|� _*� j�+� t � d��r� �,� t-�.� � � S )Nr �Metar� r� r� �model_options_basez[^\w]+z primary key is overdetermined.FT)r� rm �_composite_key� __unicode__r� c s d� j | �� f S )Nz<%s: %r>)rE r� r �rG r@ rC rU i s �z#BaseModel.__new__.<locals>.<lambda>z%sDoesNotExistrF �validate_model)/rJ rE r r� �__new__rQ r� r� rh r� r� r r� � inheritabler� rS r� r+ r� r� r� r� r� r^ r� r� r� r' r r| r� r� r1 r r. ri r� rD rW rC rF r r r� r� r r4 )rG r� �bases�attrs� meta_optionsrK r �v�model_pk� parent_pk�b� base_meta�all_inheritable�ModelOptionsBaser& r� �pk_namerW r+ �exc_name� exc_attrs�exception_classr" r� rC r� s� � zBaseModel.__new__c C s t | �� �S r? )r� rp r r@ r@ rC r� y s zBaseModel.__iter__)rE rF rG r� r� r� r� r% r@ r@ r" rC r� s � `r� c @ s� e Zd Zdd� Zedd� �Zedd� �ZedTdd ��ZedUd d ��ZedVd d��Z edd� �Z edd� �Z edd� �Z edd� �Z edd� �Zedd� �Zedd � �Zed d � �Zed d � �ZedWd"d#��Zed$d%� �Zed&d'� �Zed(d)� �ZedXd*d+��Zed,d-� �ZedYd.d/��ZedZd0d1��Zed2d3� �Zed4d5� �Z d6d7� Z e Z d8d9� Z e Z d:d;� Z!d<d=� Z"d>d?� Z#d@dA� Z$dBdC� Z%d[dDdE�Z&dFdG� Z'e(dHdI� �Z)d\dJdK�Z*d]dLdM�Z+dNdO� Z,dPdQ� Z-dRdS� Z.dS )^r* c O s@ | j �� | _t| j�| _i | _|�� D ]\}}t| ||� q&d S r? )r� r� r� r� r� rk r� rC )rA r r r r� r@ r@ rC r } s zModel.__init__c C s t | �S r? )r� r� r@ r@ rC rO � s z Model.aliasc G s( t | f|�� }| jjr$|j| jj� }|S r? )r� r� r� )rG r2 r� r@ r@ rC rp � s z Model.selectNc s, |pi }|� � �fdd��D �� t� |�S )Nc s g | ]}� j j| �| f�qS r@ �r� r& rU �rG r r@ rC rL � s z Model.update.<locals>.<listcomp>)r ry )rG � _Model__datar �fdictr@ r� rC r � s z Model.updatec s, |pi }|� � �fdd��D �� t� |�S )Nc s g | ]}� j j| �| f�qS r@ r� rU �rG r� r@ rC rL � s z Model.insert.<locals>.<listcomp>)r r� )rG r� r� r� r@ r� rC r� � s z Model.insertTc C s t | ||d�S )N)r� r� �r� )rG r� r� r@ r@ rC r� � s zModel.insert_manyc C s t | ||d�S )N)r& r� r� )rG r& r� r@ r@ rC � insert_from� s zModel.insert_fromc C s t | �S r? )r� r� r@ r@ rC r � s z Model.deletec G s t | |f|�� S r? )r )rG r� r� r@ r@ rC r\ � s z Model.rawc K s" | f |�}|j dd� |�� |S )NT)� force_insert)�saver� )rG r� rK r@ r@ rC �create� s z Model.createc O s2 | � � �� }|r|j|� }|r*|jf |�}|�� S r? )rp rO rq r� r� )rG r� r �sqr@ r@ rC r� � s z Model.getc K s8 |� di �}| �� }|�� D ]6\}}d|kr>|jf ||i�}q |�t| |�|k�}q z|�� dfW S | jk �r2 zZtdd� |�� D ��}|� |� | j j � � �$ | j f |�dfW 5 Q R � W Y S Q R X W n\ tk �r, } z<z|�� dfW W Y �& Y S | jk �r |�Y nX W 5 d }~X Y nX Y nX d S )Nr� r Fc s s" | ]\}}d |kr||fV qdS )r Nr@ )rK r r� r@ r@ rC rO � s �z&Model.get_or_create.<locals>.<genexpr>T)rQ rp r� r� rq r� r� r r� r r� r� r� r� r# )rG r r� r� r+ ra r� �excr@ r@ rC � get_or_create� s& ,zModel.get_or_createc K s� z6| j j�� � | jf |�dfW 5 Q R � W S Q R X W n\ tk r� g }|�� D ],\}}t| |�}|jsp|jrR|� ||k� qR| j |� df Y S X d S r� ) r� r� r� r� r# r� r� r� r� r� r� )rG r r� r� ra r+ r@ r@ rC � create_or_get� s ( zModel.create_or_getc O s | � � j||�S r? )rp r� )rG �dqr� r@ r@ rC r� � s z Model.filterc C s0 i }| j jr| j j|d< | j j| j jjf |�kS )Nr� )r� r� r� r� r� )rG r r@ r@ rC � table_exists� s zModel.table_existsFc C sb |r| � � rd S | jj}| jj}|jrL|dk rL|jrL|�|j�sL|�|j� |�| � | � � d S r? ) r� r� r� r� r� r� r� r| rw �_create_indexes)rG r_ r% r� r@ r@ rC rw � s zModel.create_tablec C sD g }| j jD ]2}|jrq t|j|jt|t�f�}|r |�|� q |S r? ) r� ry r� r� r� r� rS r r� )rG r& r+ �requires_indexr@ r@ rC �_fields_to_index� s � zModel._fields_to_indexc C s" t �dd� | �� D �| jjp d�S )Nc S s g | ]}|f|j f�qS r@ )r� r8 r@ r@ rC rL s z%Model._index_data.<locals>.<listcomp>r@ )� itertools�chainr� r� r^ r� r@ r@ rC � _index_data s �zModel._index_datac C s( | � � D ]\}}| jj�| ||� qd S r? )r� r� r� rz )rG � field_listr* r@ r@ rC r� s zModel._create_indexesc C s( | � � D ]\}}| jj�| ||� qd S r? )r� r� r� r{ )rG rY r� r* r@ r@ rC � _drop_indexes s zModel._drop_indexesc s� g }� j j�� }� j j}� j jjr:|jr:|�|�|j�� |�|�� �� � � � D ] }|�|� � |g|j �� qR� j j r�� j j D ].\}}� fdd�|D �}|�|� � ||�� q�dd� |D �S )Nc s g | ]}� j j| �qS r@ r� rU r� r@ rC rL % s z Model.sqlall.<locals>.<listcomp>c S s g | ] \}}|�qS r@ r@ )rK r� r� r@ r@ rC rL ' s ) r� r� r� r� r� r� r� r| rw r� rz r� r^ )rG �queriesr� r� r+ r� r� r& r@ r� rC �sqlall s z Model.sqlallc C s | j j�| ||� d S r? )r� r� rx )rG r_ r` r@ r@ rC rx ) s zModel.drop_tablec C s | j j�| ||� d S r? )r� r� ry )rG rb r` r@ r@ rC ry - s zModel.truncate_tablec C s& | j jrt| j j| j j�S t| j j�S r? )r� r� r7 r� r� r@ r@ rC r 1 s zModel.as_entityc O s t | f|�|�S r? )ro )rG r r r@ r@ rC �noop7 s z Model.noopc C s t | | jjj�S r? )r� r� r� r� r r@ r@ rC r� ; s zModel._get_pk_valuec C s | j jst| | j jj|� d S r? )r� rW rC r� r� r� r@ r@ rC � _set_pk_value? s zModel._set_pk_valuec C s | j j| �� kS r? )r� r� r� r r@ r@ rC �_pk_exprD s zModel._pk_exprc C s | j �� | �� d S r? )r� �clearr� r r@ r@ rC r� G s zModel._prepare_instancec C s d S r? r@ r r@ r@ rC r� K s zModel.preparedc C s, i }|D ] }|j |kr||j ||j < q|S r? r� )rA r� �only�new_datar+ r@ r@ rC � _prune_fieldsN s zModel._prune_fieldsc C sd | j jD ]V}|| jko8||ko8|| d ko8| j�|�d k }|rt| |t| |�� | j| ||< qd S r? )r� r� r� rk r� rC r� r� )rA r� r� � conditionsr@ r@ rC �_populate_unsaved_relationsU s � ��z!Model._populate_unsaved_relationsc C s( t | j�}| jjdk r(| jj}| �� }nd }}|rB| �||�}n,| jjrn|sn| �|| j�}|sn| j� � dS | � |� |d k r�|s�| jj r�|j D ]}|� |d � q�n|� |jd � | jf |��| �� ��� }nJ|d kr�| jf |��� d}n,| jf |��� }|d k �r |}| �|� d}| j� � |S rq )r� r� r� r� r� r� r� � dirty_fieldsr� r� r� rW r� rQ r� r rq r� r r� r� ) rA r� r� r� �pk_field�pk_value� pk_part_namer� �pk_from_cursorr@ r@ rC r� ` s@ � z Model.savec C s t | j�S r? )ri r� r r@ r@ rC �is_dirty� s zModel.is_dirtyc s � fdd�� j jD �S )Nc s g | ]}|j � jkr|�qS r@ )r� r� rU r r@ rC rL � s z&Model.dirty_fields.<locals>.<listcomp>)r� ry r r@ r rC r� � s zModel.dirty_fieldsc c s� t | �}| �� �| �� �}t | �|fg}t� }|r�|�� \}}||krHq.|�|� |jj� � D ]r\}}|j } |j |kr�|| j |j j k} | �� �| �} n||> } | �� �| �} |jr�|r�|�| | f� | |fV q^q.d S r? )r rp rq r� r� rQ r� r� r� r� r� r- r� r� r� r� r� ) rA �search_nullabler� r� r� r� �klass�rel_namer r- r> �subqueryr@ r@ rC � dependencies� s& zModel.dependenciesc C sz |rd| � |�}tt|��D ]H\}}|j}|jrP|sP|jf |jd i��|��� q|� � �|��� q| � � �| � � ��� S r? ) r� r� r� r� r� r r� rq r r r� )rA � recursive�delete_nullabler� r� r r� r@ r@ rC �delete_instance� s zModel.delete_instancec C s t | j| �� f�S r? )r r# r� r r@ r@ rC r � s zModel.__hash__c C s( |j | j ko&| �� d k o&|�� | �� kS r? )r# r� �rA r� r@ r@ rC ra � s ��z Model.__eq__c C s | |k S r? r@ r� r@ r@ rC rb � s z Model.__ne__)N)N)T)F)F)FF)FF)FN)F)FF)/rE rF rG r ro rO rp r r� r� r� r r\ r� r� r� r� r� r� rw r� r� r� r� r� rx ry r r� r� �get_idr� �set_idr� r� r� r� r� r� r� r� r� r� r� r ra rb r@ r@ r@ rC r* | s� % c C s | � � } d | _| _| S r? )r r r )r� r@ r@ rC �clean_prefetch_subquery� s r� c s� t | �g}t|�D �]�\}}t|t�r0|\}}nd }t|t�sPt|t�rP|�� }|j�d }}t t |d ��D ]~}|| } | j } | j ��j j�dd�} | r��fdd�| D �}�fdd�| D �} n�j j�dd�}|s�|rn|�ks�|d krn q�qn|�s |�s |�rd| nd} td || f ��|�rht| �� ttj� fd d�t|| �D ��}|�|�}|�t ||d �� q|rt| �� ttj� fd d�|D ��}|�|�}|�t ||d�� q|S ) Nr� T)r� c s g | ]}t � |j��qS r@ )r� r� �rK r )�subquery_modelr@ rC rL � s z)prefetch_add_subquery.<locals>.<listcomp>c s g | ]}t � |jj��qS r@ )r� r� r� r� )� last_modelr@ rC rL � s z using %sr1 z1Error: unable to find foreign key for query: %s%sc s g | ]\}}|� � |�> �qS r@ )rp )rK r r� ��cleanedr@ rC rL � s �Fc s g | ]}|j � �|�> �qS r@ )r� rp )rK r� r� r@ rC rL � s �)�PrefetchResultr� rS r� r� rv r* rp r� r� r� r� r� r� r� r r� rR r� r� r� rq r� )r� � subqueries� fixed_queriesr� r� � target_model�fksr� r� �prefetch_result� last_query�relsrK �tgt_err�exprr@ )r� r� r� rC �prefetch_add_subquery� sZ �� �� � r� � __prefetched)r� r& r� � rel_models� field_to_namer� c s. e Zd Zd� fdd� Zdd� Zdd� Z� ZS ) r� Nc sr |rP|r&dd� |D �}dd� |D �}n dd� |D �}dd� |D �}t t||��}|j}tt| ��| ||||||�S )Nc S s g | ] }|j �qS r@ r� r8 r@ r@ rC rL � s z*PrefetchResult.__new__.<locals>.<listcomp>c S s g | ] }|j j�qS r@ )r� r� r8 r@ r@ rC rL � s c S s g | ] }|j �qS r@ r� r8 r@ r@ rC rL � s c S s g | ] }|j �qS r@ r� r8 r@ r@ rC rL � s )r� r� r� r r� r� )rG r� r& r� r� r� r� �foreign_key_attrsr" r@ rC r� � s" �zPrefetchResult.__new__c C s� | j rB| jD ]2}|j|j }||f}||kr t||j|| � q n^| jD ]V\}}|j|jj }||f}|�|g �}d|j }|D ]} t| ||� q�t|||� qHd S )Nz %s_prefetch) r� r& r� r� rC r� r� r� r} ) rA r: �id_mapr+ � identifierr� �attname� rel_instancesr� rK r@ r@ rC �populate_instance s z PrefetchResult.populate_instancec C sX | j D ]L\}}|j�|j| �}||f}| jr8|||< q|�|g � || �|� qd S r? )r� r� r r� r� rX r� )rA r: r� r+ r� �identityr� r@ r@ rC �store_instance s z PrefetchResult.store_instance)NNNNN)rE rF rG r� r� r r% r@ r@ r" rC r� � s �r� c G s� |s| S t | |�}i }i }t|�D ]�}|j}|jrX|jD ] }|�|g � || �|� q8i ||< || }t|�|��} |j D ]:} |jr�|� | |� | r||| D ]} | � | || j � q�q|q"|j S r? ) r� r� r� r& r� rX r� ri r� r� r r� ) r� r� r� �deps�rel_mapr� � query_modelr- r� � has_relationsr: r� r@ r@ rC r0 s* c K s t | �D ]}|jf |� qdS )z8Create tables for all given models (in the right order).N)rv rw )r� �create_table_kwargsr� r@ r@ rC r� ; s r� c K s" t t| ��D ]}|jf |� q dS )z6Drop tables for all given models (in the right order).N)r� rv rx )r� �drop_table_kwargsr� r@ r@ rC r @ s r )N)N)r' )�r] rz r rg r� �loggingr� r� re r� rG r< rt �bisectr r � collectionsr r r � ImportErrorr� r r � functoolsr �inspectr � __version__�__all__r> �Handler� getLoggerr� � addHandlerrJ �objectrM � version_info�PY2rn r� �builtinsrQ rR r� r r( rZ r) r: r� rh r; r rd rc �unicode�buffer�exec� RuntimeErrorrn rq Z pysqlite2rr �pysq3r r0 Z psycopg2cffirs �registerr: rt r< ZMySQLdbrZ ZpymysqlZplayhouse._speedupsru rv rw r� r� r� r� �register_adapterr# rD �DATETIME_PARTSr� r� r� r� r� ra r� r� r� r` r& r� r( r� r) r� r' r( rf r& r' re r� r� r� r� r� r� r� r� r� r� r� r r r r r3 r r6 r= r7 r4 r7 r� r rY r. r� r r� r� r= r r r� r� r� r� r r r" r r5 r1 r r r r r, r r r8 r r< rB rC r r r9 r: r rj ro rr r r r� r� r~ r� r� r� r� r� r� r� r� r� r� r r- r� ro r rt ry r� r� r� r� r� r� r� r! r r r# r$ r% r, r- r2 r� r� �localr� r� r6 r/ r+ rb r� r; r� r� r� r. r� r� r r� r� r r� r* r� r� r� r� r0 r� r r@ r@ r@ rC �<module> s� �8 " � ��� �� % M{ & 9" R' 4 /0 (<' ��� <u ` '# ! &h @ 2+
💾 Save Changes
❌ Cancel