B CVvgt@spddlmZddlmZddlmZdddgZGdddZGd ddeZGd ddeZ Gd ddeZ d S) )Enum)Q)QueryCheckConstraint DeferrableUniqueConstraintc@s<eZdZddZddZddZddZd d Zd d Zd S)BaseConstraintcCs ||_dS)N)name)selfr r @/tmp/pip-install-o3oxmrkh/Django/django/db/models/constraints.py__init__ szBaseConstraint.__init__cCs tddS)Nz.This method must be implemented by a subclass.)NotImplementedError)r model schema_editorr r r constraint_sql szBaseConstraint.constraint_sqlcCs tddS)Nz.This method must be implemented by a subclass.)r)r rrr r r create_sqlszBaseConstraint.create_sqlcCs tddS)Nz.This method must be implemented by a subclass.)r)r rrr r r remove_sqlszBaseConstraint.remove_sqlcCs0d|jj|jjf}|dd}|dd|jifS)Nz%s.%szdjango.db.models.constraintszdjango.db.modelsr r ) __class__ __module____name__replacer )r pathr r r deconstructs zBaseConstraint.deconstructcCs|\}}}|j||S)N)rr)r _argskwargsr r r cloneszBaseConstraint.cloneN) rr __qualname__r rrrrrr r r r r s rcs\eZdZfddZddZddZddZd d Zd d Zfd dZ fddZ Z S)rcs*||_t|ddstdt|dS)NZ conditionalFzACheckConstraint.check must be a Q instance or boolean expression.)checkgetattr TypeErrorsuperr )r rr )rr r r !s  zCheckConstraint.__init__csRt|dd}||j}|jjd}||j\}}|tfdd|DS)NF)r alias_cols) connectionc3s|]}|VqdS)N) quote_value).0p)rr r /sz1CheckConstraint._get_check_sql..)r build_wherer get_compilerr$as_sqltuple)r rrquerywherecompilersqlparamsr )rr _get_check_sql*s   zCheckConstraint._get_check_sqlcCs|||}||j|S)N)r2Z _check_sqlr )r rrrr r r r1s zCheckConstraint.constraint_sqlcCs|||}|||j|S)N)r2Z_create_check_sqlr )r rrrr r r r5s zCheckConstraint.create_sqlcCs|||jS)N)Z_delete_check_sqlr )r rrr r r r9szCheckConstraint.remove_sqlcCsd|jj|j|jfS)Nz<%s: check='%s' name=%r>)rrrr )r r r r __repr__<szCheckConstraint.__repr__cs.t|tr"|j|jko |j|jkSt|S)N) isinstancerr rr"__eq__)r other)rr r r5?s zCheckConstraint.__eq__cs$t\}}}|j|d<|||fS)Nr)r"rr)r rrr)rr r rDs zCheckConstraint.deconstruct) rrrr r2rrrr3r5r __classcell__r r )rr r s  c@seZdZdZdZdS)rdeferredZ immediateN)rrrZDEFERREDZ IMMEDIATEr r r r rJscsheZdZdddddfdd ZddZdd Zd d Zd d ZddZfddZ fddZ Z S)rNr ) condition deferrableinclude opclassescs|s tdt|tdtfs&td|r6|r6td|rF|rFtd|rV|rVtdt|tdtfsptdt|tdttfstdt|ttfstd|rt|t|krtd t||_||_ ||_ |rt|nd |_ ||_ t |dS) Nz=At least one field is required to define a unique constraint.z0UniqueConstraint.condition must be a Q instance.z4UniqueConstraint with conditions cannot be deferred.z8UniqueConstraint with include fields cannot be deferred.z3UniqueConstraint with opclasses cannot be deferred.z:UniqueConstraint.deferrable must be a Deferrable instance.z1UniqueConstraint.include must be a list or tuple.z3UniqueConstraint.opclasses must be a list or tuple.z]UniqueConstraint.fields and UniqueConstraint.opclasses must have the same number of elements.r ) ValueErrorr4typerrlistr,lenfieldsr9r:r;r<r"r )r rAr r9r:r;r<)rr r r Ps:  zUniqueConstraint.__init__cs`|jdkrdSt|dd}||j}|jjd}||j\}}|tfdd|DS)NF)rr#)r$c3s|]}|VqdS)N)r%)r&r')rr r r(sz6UniqueConstraint._get_condition_sql..)r9rr)r*r$r+r,)r rrr-r.r/r0r1r )rr _get_condition_sql~s   z#UniqueConstraint._get_condition_sqlc sRfdd|jD}fdd|jD}||}|j||j||j||jdS)Ncsg|]}j|jqSr )_meta get_fieldcolumn)r& field_name)rr r sz3UniqueConstraint.constraint_sql..csg|]}j|jqSr )rCrDrE)r&rF)rr r rGs)r9r:r;r<)rAr;rBZ _unique_sqlr r:r<)r rrrAr;r9r )rr rs  zUniqueConstraint.constraint_sqlc sRfdd|jD}fdd|jD}||}|j||j||j||jdS)Ncsg|]}j|jqSr )rCrDrE)r&rF)rr r rGsz/UniqueConstraint.create_sql..csg|]}j|jqSr )rCrDrE)r&rF)rr r rGs)r9r:r;r<)rAr;rBZ_create_unique_sqlr r:r<)r rrrAr;r9r )rr rs  zUniqueConstraint.create_sqlcs<||}fdd|jD}|j|j||j||jdS)Ncsg|]}j|jqSr )rCrDrE)r&rF)rr r rGsz/UniqueConstraint.remove_sql..)r9r:r;r<)rBr;Z_delete_unique_sqlr r:r<)r rrr9r;r )rr rs   zUniqueConstraint.remove_sqlc Csnd|jj|j|j|jdkrdnd|j|jdkr4dnd|j|jsFdn dt|j|js\dn dt|jfS)Nz<%s: fields=%r name=%r%s%s%s%s>z condition=%sz deferrable=%sz include=%sz opclasses=%s) rrrAr r9r:r;reprr<)r r r r r3s zUniqueConstraint.__repr__cs^t|trR|j|jkoP|j|jkoP|j|jkoP|j|jkoP|j|jkoP|j|jkSt |S)N) r4rr rAr9r:r;r<r"r5)r r6)rr r r5s       zUniqueConstraint.__eq__csdt\}}}|j|d<|jr*|j|d<|jr:|j|d<|jrJ|j|d<|jrZ|j|d<|||fS)NrAr9r:r;r<)r"rrAr9r:r;r<)r rrr)rr r rs     zUniqueConstraint.deconstruct) rrrr rBrrrr3r5rr7r r )rr rOs&    N) enumrZdjango.db.models.query_utilsrZdjango.db.models.sql.queryr__all__rrrrr r r r s    *