U
    h                  
   @  s  d dl mZ d dlmZ d dlZd dlZd dlZd dlmZ d dlm	Z	 d dlm
Z
 d dlmZ d dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddl&m'Z' ejrdd lm(Z( dd!lm)Z) dd"lm*Z* dd#lm+Z+ dd$lm,Z, dd%lm-Z- dd&lm.Z. dd'lm/Z/ dd(lm0Z0 dd)l1m2Z2 dd*l(m3Z3 dd+l(m4Z4 dd,l(m5Z5 dd-l(m6Z6 dd.l(m7Z7 dd/l(m8Z8 dd0l(m9Z9 dd1l*m:Z: dd2l+m;Z; dd3l+m<Z< dd4l+m=Z= dd5l+m>Z> dd6l+m?Z? dd7l+m@Z@ dd8lmAZA ed9ejBd:ZCed;e	d<ef d:ZDed=ejEd:ZFed>ed:ZGd?d@dAdBdCZHdDdE ZIdFdFdFdGdHdIdJZJdFdFdKdLdMdNdOZKedPd?d?dFdQdRdSZLed<dTdUd?dVd?dFdWdXdSZLedYd?d?dZdQd[dSZLed\d?d?d]dQd^dSZLed_d?d?d]dQd`dSZLedad?d?dbdQdcdSZLeddded?dfdQdgdSZLedhded?didQdjdSZLedkd?d?dldQdmdSZLedndod?dpdQdqdSZLedrdsd?dsdQdtdSZLedudvd?dvdQdwdSZLedxdyd?dzdQd{dSZLed<d|dxdvdVd?d}d~ddSZLeddd?ddQddSZLeddd?ddQddSZLedd?d?d?dQddSZLddddddd?ddd@d@d?d?dddSZLdUd?d?dFdQddZMd_dddddZNG dd dZOG dd dZPG dd dZQG dd deOZRG dd deOZSde#jTddfd?ddddddddZUG dd deOZVG dd deOZWG dd deOZXG dd deOZYG dd deOZZG dd deOZ[G dd deSeOZ\G dd de\eOZ]G dd deOZ^G dd deSeOZ_G dd deWeSeOZ`G dd deWeOZaG dd deVeOZbG dd deReOZcG ddÄ deWeOZdG ddń deWeSeOejeZfG ddǄ defeOZgG ddɄ defeOZhG dd˄ deReOZiG dd̈́ deOZjG ddτ deQeOZkG ddф dePeWeOZlG ddӄ dePeReOZmG ddՄ demZnG ddׄ deOZoG ddل de\ZpG ddۄ deYeWeOZqG dd݄ deOZrG dd߄ deWeOZsG dd deVeOZtG dd derZuG dd deOZvG dd deOZwG dd deYeVeOZxG dd dexZyG dd deyZzG dd deYeVeOZ{G dd deVeOZ|G dd deVeOZ}i Z~eeD ]JZeeeZedrBeddZee krBe e eZee~e< qBese~ej ZeeeefD ]Zee~eje < qdS )    )annotationsN)Any)Callable)cast)Dict)Iterable)Iterator)List)NoReturn)Optional)overload)Sequence)Tuple)Type)TYPE_CHECKING)TypeVar)Union   )roles)visitors)is_from_clause)ExecutableOption)Options)HasCacheKey)	Visitable   )exc)
inspection)util)Literal)elements)lambdas)schema)
selectable)_ColumnExpressionArgument)_ColumnsClauseArgument)_DDLColumnArgument)_DMLTableArgument)_FromClauseArgument)_DMLTableElement)BindParameter)ClauseElement)ColumnClause)ColumnElement)NamedColumn)SQLCoreOperations)
TextClause)Column)_ColumnsClauseElement)_JoinTargetProtocol)
FromClause)HasCTE)
SelectBase)Subquery)_TraverseCallableType_SR)bound_F._StringOnlyR_Tr   bool)elementreturnc                 C  s   t | ttjf ot| d S )zhReturn whether or not the element is a "literal" in the context
    of a SQL expression construct.

    __clause_element__)
isinstancer   r"   SchemaEventTargethasattr)r?    rE   </tmp/pip-unpacked-wheel-y9ln43lq/sqlalchemy/sql/coercions.py_is_literalN   s    
rG   c                 C  sr   t | tjr2t | ts2| D ]}t|s dS qdS t | ttjtt	t
jjf opt| d opt | t ppt| t S )a  Return whether or not the element is a "literal" in the context
    of a SQL expression construct.

    does a deeper more esoteric check than _is_literal.   is used
    for lambda elements that have to distinguish values that would
    be bound vs. not without any context.

    FTrA   )rB   collections_abcr   str_deep_is_literalr   r"   rC   r   r   r   ZlanghelperssymbolrD   type
issubclass)r?   elemrE   rE   rF   rJ   Z   s,    
 


rJ   rI   zCallable[[_F], _F])	paramnamemeth_rst	param_rstr@   c                 C  s   t | d||f S )Nz.. warning:: The %s argument to %s can be passed as a Python string argument, which will be treated as **trusted SQL text** and rendered as given.  **DO NOT PASS UNTRUSTED INPUT TO THIS PARAMETER**.)r   Zadd_parameter_text)rO   rP   rQ   rE   rE   rF   _document_text_coercion   s    rR   z+Union[Sequence[_T], Sequence[Sequence[_T]]]Sequence[_T])attrnamefnnameargsr@   c                 C  sb   |rXt |d tttfrXt|dkrXt |d trJtd|  d| dtd|d S td|S )Nr   r   zThe "z" argument to zt(), when referring to a sequence of items, is now passed as a series of positional elements, rather than as a list. rS   )rB   listsetdictlenr   ArgumentErrorr   )rT   rU   rV   rE   rE   rF   !_expression_collection_was_a_list   s    $r\   zType[roles.TruncatedLabelRole])roler?   kwr@   c                 K  s   d S NrE   r]   r?   r^   rE   rE   rF   expect   s    ra   as_keyzType[roles.DMLColumnRole]zLiteral[True])r]   r?   rc   r^   r@   c                K  s   d S r_   rE   )r]   r?   rc   r^   rE   rE   rF   ra      s    zType[roles.LiteralValueRole]zBindParameter[Any]c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    z!Type[roles.DDLReferredColumnRole]zUnion[Column[Any], str]c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    z#Type[roles.DDLConstraintColumnRole]c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    zType[roles.StatementOptionRole]z%Union[ColumnElement[Any], TextClause]c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    z&Type[roles.LabeledColumnExprRole[Any]]z_ColumnExpressionArgument[_T]zNamedColumn[_T]c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    zgUnion[Type[roles.ExpressionElementRole[Any]], Type[roles.LimitOffsetRole], Type[roles.WhereHavingRole]]zColumnElement[_T]c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    	zUnion[Type[roles.ExpressionElementRole[Any]], Type[roles.LimitOffsetRole], Type[roles.WhereHavingRole], Type[roles.OnClauseRole], Type[roles.ColumnArgumentRole]]zColumnElement[Any]c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    zType[roles.DMLTableRole]r'   r)   c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    zType[roles.HasCTERole]r5   c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    zType[roles.SelectStatementRole]r6   c                 K  s   d S r_   rE   r`   rE   rE   rF   ra     s    zType[roles.FromClauseRole]r(   r4   c                 K  s   d S r_   rE   r`   rE   rE   rF   ra     s    )explicit_subqueryr7   )r]   r?   rd   r^   r@   c                K  s   d S r_   rE   )r]   r?   rd   r^   rE   rE   rF   ra     s    zType[roles.ColumnsClauseRole]z_ColumnsClauseArgument[Any]r2   c                 K  s   d S r_   rE   r`   rE   rE   rF   ra      s    zType[roles.JoinTargetRole]r3   c                 K  s   d S r_   rE   r`   rE   rE   rF   ra   (  s    z	Type[_SR]c                 K  s   d S r_   rE   r`   rE   rE   rF   ra   1  s    F)apply_propagate_attrsargnamepost_inspectdisable_inspectionzOptional[ClauseElement]Optional[str])r]   r?   re   rf   rg   rh   r^   r@   c                K  s  | j r2t|r2t|dr2tj|| tjf ||dS t|  }|}t|tj	t
jt
jtjfsRd }	|jrt|j|f|}	n|}d}
|jrt|| rt|drd}
n(t|drd}
t|dds| }qqq|
sL|jr*|s*tj|dd}|d k	r*|r|j z| }	W n" tk
r(   ||| Y nX |	d krP|j|fd|i|}	n|}	n t|tjrn|jf |}	n|}	|d k	rtjrt|	ttfst|jst|	d	d r|	j|_|j |	j!j"kr|j#r|j#|	f||d
|}	|	S |j$||	fd|i|S d S )N__code__)re   FrA   Tis_clause_element)Zraiseerrrf   _propagate_attrs)rf   original_element)%allows_lambdacallablerD   r!   ZLambdaElementZLambdaOptions_impl_lookuprB   r    ZCompilerElementr"   Z
SchemaItemZFetchedValueZ	PyWrapper_resolve_literal_only_literal_coercion$_skip_clauseelement_for_target_matchgetattrrA   _use_inspectionr   inspectZ_post_inspectAttributeError_raise_for_expectedZ_sa__py_wrapper_literaltypingr   r/   r+   AssertionErrorrl   _role_class	__class____mro___post_coercion_implicit_coercions)r]   r?   re   rf   rg   rh   r^   implrm   resolvedrk   ZinsprE   rE   rF   ra   9  s    	
		





   c                 K  s"   | dd  t| |fddi|S )Nrc   T)popra   r`   rE   rE   rF   expect_as_key  s    r   zIterable[_DDLColumnArgument]zwIterator[Tuple[Union[str, Column[Any]], Optional[ColumnClause[Any]], Optional[str], Optional[Union[Column[Any], str]]]])r]   expressionsr@   c           	      c  s   |D ]~}d }d }t | |}t|tr<t|ts2t| }}n(g }|j}t|i d|i |rd|d }|d k	rp|n|}||||fV  qd S )Ncolumnr   )ra   rB   rI   rz   appendr   Ztraverse)	r]   r   exprZstrnamer   r   colsZ
col_appendZadd_elementrE   rE   rF    expect_col_expression_collection  s    


r   c                   @  sv   e Zd ZU dZdd ZdZded< dZdZdd	 Z	dddd
dddddZ
ddddddd
dd
d
ddddddZdS )RoleImpl)r{   nameru   c                 K  s
   t  d S r_   )NotImplementedErrorselfr?   r^   rE   rE   rF   rr     s    zRoleImpl._literal_coercionNr   r~   Fc                 C  s    || _ |j| _t|tj| _d S r_   )r{   Z
_role_namer   rM   r   ZUsesInspectionru   )r   Z
role_classrE   rE   rF   __init__  s    zRoleImpl.__init__ri   r?   r   rf   r^   r@   c                 K  s   |  ||| d S r_   rx   r   r?   r   rf   r^   rE   rE   rF   r     s    zRoleImpl._implicit_coercionsadvicecodeerrOptional[Any]Optional[Exception]r
   r?   rf   r   r   r   r   r^   r@   c          
      K  sn   |d k	r||k	rd||f }nt |}|r<d| j||f }	nd| j|f }	|rZ|	d| 7 }	tj|	|d|d S )Nz!%r object resolved from %r objectz$%s expected for argument %r; got %s.z%s expected, got %s. )r   )reprr   r   r[   )
r   r?   rf   r   r   r   r   r^   gotmsgrE   rE   rF   rx     s    zRoleImpl._raise_for_expected)N)NN)__name__
__module____qualname__	__slots__rr   r~   __annotations__rq   rs   r   r   rx   rE   rE   rE   rF   r     s   
	   r   c                   @  s   e Zd ZdZdd ZdS )_DeannotaterE   c                 K  s   ddl m} ||S )Nr   )_deep_deannotate)r   r   )r   r   r^   r   rE   rE   rF   r~     s    z_Deannotate._post_coercionNr   r   r   r   r~   rE   rE   rE   rF   r   	  s   r   c                   @  s   e Zd ZdZdZdS )_StringOnlyrE   TN)r   r   r   r   rq   rE   rE   rE   rF   r     s   r   c                   @  s"   e Zd ZdZdddZdd ZdS )_ReturnsStringKeyrE   Nc                 K  s    t |tr|S | ||| d S r_   rB   rI   rx   r   rE   rE   rF   r     s    
z%_ReturnsStringKey._implicit_coercionsc                 K  s   |S r_   rE   r   rE   rE   rF   rr   !  s    z#_ReturnsStringKey._literal_coercion)Nr   r   r   r   r   rr   rE   rE   rE   rF   r     s   
r   c                   @  s"   e Zd ZdZdd ZdddZdS )_ColumnCoercionsrE   c                 C  s   t d d S )Nz|implicitly coercing SELECT object to scalar subquery; please use the .scalar_subquery() method to produce a scalar subquery.)r   warn)r   rE   rE   rF   "_warn_for_scalar_subquery_coercion(  s    z3_ColumnCoercions._warn_for_scalar_subquery_coercionNc                 K  s~   |}t |dds | ||| nZ|jr6|   | S |jrZt|tjrZ|   |j	 S | j
jrl|jrl|S | ||| d S )Nrk   F)rt   rx   _is_select_baser   scalar_subquery_is_from_clauserB   r#   r7   r?   r{   rn   _is_lambda_element)r   r?   r   rf   r^   rm   rE   rE   rF   r   /  s     
z$_ColumnCoercions._implicit_coercions)N)r   r   r   r   r   r   rE   rE   rE   rF   r   %  s   r   zType[exc.SQLAlchemyError]r   r
   )r?   rf   exc_clsextrar   r@   c                 C  s:   |dt | |rd|f nd|r(d| ndd |d S )Nzc%(extra)sTextual SQL expression %(expr)r %(argname)sshould be explicitly declared as text(%(expr)r)for argument %s z%s )r   rf   r   )r   ellipses_string)r?   rf   r   r   r   rE   rE   rF   _no_text_coercionA  s    r   c                   @  s   e Zd ZdZddddZdS )_NoTextCoercionrE   Nrf   c                K  s4   t |tr$ttj| jr$t|| n| || d S r_   )rB   rI   rM   r    r0   r{   r   rx   r   r?   rf   r^   rE   rE   rF   rr   V  s     z!_NoTextCoercion._literal_coercionr   r   r   r   rr   rE   rE   rE   rF   r   S  s   r   c                   @  s4   e Zd ZdZdZdZdZd	ddZddddZdS )
_CoerceLiteralsrE   FNc                 C  s
   t ||S r_   r   r   r?   rf   rE   rE   rF   _text_coercione  s    z_CoerceLiterals._text_coercionr   c                K  s   t |tr6| jr&|dkr&tjdddS | j||f|S | jrl|d krLt S |dkr\t S |dkrlt	 S | j
rt |tjrtjt|ddS | || d S )N*T)Z
is_literalF)rB   rI   _coerce_starr    r,   r   _coerce_constsNullFalse_True__coerce_numericsnumbersNumberrx   r   rE   rE   rF   rr   h  s    
z!_CoerceLiterals._literal_coercion)N)	r   r   r   r   r   r   r   r   rr   rE   rE   rE   rF   r   _  s   
r   c                   @  s*   e Zd ZdZd	dddddZdd ZdS )
LiteralValueImplTNF)type_literal_executec                K  s4   t |s | j|f||d| tjd ||d|dS )N)r   rf   T)r   uniquer   )rG   rx   r    r*   )r   r?   r   rf   r   r   r^   rE   rE   rF   r     s     
 z$LiteralValueImpl._implicit_coercionsc                 K  s   |S r_   rE   r   rE   rE   rF   rr     s    z"LiteralValueImpl._literal_coercion)N)r   r   r   rq   r   rr   rE   rE   rE   rF   r   }  s    r   c                      s@   e Zd ZdZdddddddddddddd	 fd
dZ  ZS )_SelectIsNotFromrE   Nr   r   ri   r   r   r
   r   c                  sj   |st |tjst |tjr:d|d k	r,|jn|f }d}nd }t j|f|||||d| dsftd S )NzETo create a FROM clause from a %s object, use the .subquery() method.Z89ve)rf   r   r   r   r   F)rB   r   ZSelectStatementRoler|   superrx   rz   )r   r?   rf   r   r   r   r   r^   r|   rE   rF   rx     s0    


z$_SelectIsNotFrom._raise_for_expected)NNr   r   r   r   rx   __classcell__rE   rE   r   rF   r     s     r   c                   @  s0   e Zd ZdZd
ddddddddZdd	 ZdS )HasCacheKeyImplrE   Nr   ri   r   c                 K  s    t |tr|S | ||| d S r_   )rB   r   rx   r   rE   rE   rF   r     s    
z#HasCacheKeyImpl._implicit_coercionsc                 K  s   |S r_   rE   r   rE   rE   rF   rr     s    z!HasCacheKeyImpl._literal_coercion)Nr   rE   rE   rE   rF   r     s    r   c                   @  s0   e Zd ZdZd
ddddddddZdd	 ZdS )ExecutableOptionImplrE   Nr   ri   r   c                 K  s    t |tr|S | ||| d S r_   )rB   r   rx   r   rE   rE   rF   r     s    
z(ExecutableOptionImpl._implicit_coercionsc                 K  s   |S r_   rE   r   rE   rE   rF   rr     s    z&ExecutableOptionImpl._literal_coercion)Nr   rE   rE   rE   rF   r     s    r   c                      s4   e Zd ZdZddddddZd	 fdd	Z  ZS )
ExpressionElementImplrE   NF)r   r   is_crudc             
   K  sp   |d kr"|s"|d ks|j s"t S ztj|||d|dW S  tjk
rj } z| j||d W 5 d }~X Y nX d S )NT)r   Z_is_crudr   )Zshould_evaluate_noner    r   r*   r   r[   rx   )r   r?   r   r   r   r^   r   rE   rE   rF   rr     s&        z'ExpressionElementImpl._literal_coercionc                   sD   t |tjrd}nt |tjr$d}nd }t j|f|||d|S )NzTTo create a column expression from a VALUES clause, use the .scalar_values() method.z`To create a column expression from a FROM clause row as a whole, use the .table_valued() method.rf   r   r   )rB   r#   Valuesr   ZAnonymizedFromClauseRoler   rx   r   r?   rf   r   r^   r   r   rE   rF   rx     s       z)ExpressionElementImpl._raise_for_expected)NN)r   r   r   r   rr   rx   r   rE   rE   r   rF   r     s     r   c                   @  s.   e Zd ZdZdddddZddddZdS )	BinaryElementImplrE   N)bindparam_typerf   c             
   K  sJ   z|j |||dW S  tjk
rD } z| j||d W 5 d }~X Y nX d S )N)r   r   )_bind_paramr   r[   rx   )r   r?   r   operatorr   rf   r^   r   rE   rE   rF   rr     s    
z#BinaryElementImpl._literal_coercion)r   c                K  s,   |j jr(|j js(||d k	r |n|j }|S r_   )rL   Z_isnullZ_with_binary_element_type)r   r   r   r   r^   rE   rE   rF   r~   '  s
    z BinaryElementImpl._post_coercion)r   r   r   r   rr   r~   rE   rE   rE   rF   r     s
   r   c                   @  sJ   e Zd ZdZdddddddddZdd	 Zed
dd Zdd Z	dS )InElementImplrE   Nr   ri   r   c                 K  sd   |j rRt|tjr4|jjr4| | | j|jf|S | | | j| f|S n| 	||| d S r_   )
r   rB   r#   Aliasr?   r   _warn_for_implicit_coercionr~   selectrx   r   rE   rE   rF   r   2  s    


z!InElementImpl._implicit_coercionsc                 C  s   t d|jj  d S )Nz_Coercing %s object into a select() for use in IN(); please pass a select() construct explicitly)r   r   r|   r   )r   rN   rE   rE   rF   r   F  s
    z)InElementImpl._warn_for_implicit_coercionzsqlalchemy.sql.elementsc                  s   t |ri t|}|D ]>}t|st|t jjjsPt|dsP| j	|f| q||< qrzt
j fdd|D  S  j|ddS n| j	|f| d S )NrA   c                   s(   g | ] }|kr| n
  |qS rE   )r   ).0or   Znon_literal_expressionsr   rE   rF   
<listcomp>a  s   
z3InElementImpl._literal_coercion.<locals>.<listcomp>T)	expanding)r   Zis_non_string_iterablerW   rG   rB   Z	preloadedZsql_elementsr-   rD   rx   r    
ClauseListr   )r   r?   r   r   r^   r   rE   r   rF   rr   M  s,    
 
zInElementImpl._literal_coercionc                K  s|   |j r| S t|tjr8t|jdkr,t|j|dS t|tj	r`|j
dd}d|_||_|S t|tjrt| S |S d S )Nr   )ZagainstT)Zmaintain_key)r   r   rB   r    r   rZ   Zclausesrz   Z
self_groupr*   Z_cloner   Z	expand_opr#   r   Zscalar_values)r   r?   r   r   r^   rE   rE   rF   r~   p  s    zInElementImpl._post_coercion)N)
r   r   r   r   r   r   r   Zpreload_modulerr   r~   rE   rE   rE   rF   r   /  s    
"r   c                   @  s*   e Zd ZdZdZdd ZddddZdS )	OnClauseImplrE   Tc                 K  s   |  | d S r_   r   r   rE   rE   rF   rr     s    zOnClauseImpl._literal_coercionN)rm   c                K  s   t |tjr|S |S r_   )rB   r   JoinTargetRole)r   r   rm   r^   rE   rE   rF   r~     s    zOnClauseImpl._post_coercion)r   r   r   r   r   rr   r~   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZdZdddZdS )WhereHavingImplrE   TNc                 C  s
   t ||S r_   r   r   rE   rE   rF   r     s    zWhereHavingImpl._text_coercion)Nr   r   r   r   r   r   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZdZdddZdS )StatementOptionImplrE   TNc                 C  s
   t |S r_   r    r0   r   rE   rE   rF   r     s    z"StatementOptionImpl._text_coercion)Nr   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZdS )ColumnArgumentImplrE   Nr   r   r   r   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZdS )ColumnArgumentOrKeyImplrE   Nr   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZdddZdS )StrAsPlainColumnImplrE   Nc                 C  s
   t |S r_   )r    r,   r   rE   rE   rF   r     s    z#StrAsPlainColumnImpl._text_coercion)N)r   r   r   r   r   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZdZdddZdS )ByOfImplrE   TNc                 C  s
   t |S r_   )r    Z_textual_label_referencer   rE   rE   rF   r     s    zByOfImpl._text_coercion)Nr   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZdd ZdS )OrderByImplrE   c                 K  s(   t || jr |jd k	r t|S |S d S r_   )rB   r{   Z_order_by_label_elementr    Z_label_reference)r   r   r^   rE   rE   rF   r~     s    

zOrderByImpl._post_coercionNr   rE   rE   rE   rF   r     s   r   c                   @  s(   e Zd ZdZdddddddddZdS )	GroupByImplrE   Nr   ri   r   c                 K  s   t |rtj|j S |S d S r_   )r   r    r   cr   rE   rE   rF   r     s    zGroupByImpl._implicit_coercions)Nr   r   r   r   r   rE   rE   rE   rF   r     s    r   c                   @  s   e Zd ZdZddddZdS )DMLColumnImplrE   Frb   c                K  s   |r
|j S |S d S r_   )key)r   r?   rc   r^   rE   rE   rF   r~     s    zDMLColumnImpl._post_coercionNr   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZddddZdS )ConstExprImplrE   Nr   c                K  s@   |d krt  S |dkr t  S |dkr0t  S | || d S )NFT)r    r   r   r   rx   r   rE   rE   rF   rr     s    zConstExprImpl._literal_coercionr   rE   rE   rE   rF   r     s   r   c                   @  s0   e Zd ZdZd
ddddddddZdd	 ZdS )TruncatedLabelImplrE   Nr   ri   r   c                 K  s    t |tr|S | ||| d S r_   r   r   rE   rE   rF   r     s    
z&TruncatedLabelImpl._implicit_coercionsc                 K  s   t |tjr|S t|S dS )zcoerce the given value to :class:`._truncated_label`.

        Existing :class:`._truncated_label` and
        :class:`._anonymous_label` objects are passed
        unchanged.
        N)rB   r    Z_truncated_labelr   rE   rE   rF   rr     s    z$TruncatedLabelImpl._literal_coercion)Nr   rE   rE   rE   rF   r     s    r   c                   @  s   e Zd ZdZdZdddZdS )DDLExpressionImplrE   TNc                 C  s
   t |S r_   r   r   rE   rE   rF   r     s    z DDLExpressionImpl._text_coercion)Nr   rE   rE   rE   rF   r     s   r   c                   @  s   e Zd ZdZdS )DDLConstraintColumnImplrE   Nr   rE   rE   rE   rF   r   !  s   r   c                   @  s   e Zd ZdZdS )DDLReferredColumnImplrE   Nr   rE   rE   rE   rF   r   %  s   r   c                   @  s0   e Zd ZdZd
ddddddddZdd	 ZdS )LimitOffsetImplrE   Nr   ri   r   c                 K  s   |d krd S |  ||| d S r_   r   r   rE   rE   rF   r   ,  s    z#LimitOffsetImpl._implicit_coercionsc                K  s,   |d krd S t |}tj|||ddS d S )NT)r   r   )r   Zasintr#   Z_OffsetLimitParam)r   r?   r   r   r^   valuerE   rE   rF   rr   8  s    
   z!LimitOffsetImpl._literal_coercion)Nr   rE   rE   rE   rF   r   )  s    r   c                      s0   e Zd ZdZddddddd fddZ  ZS )	LabeledColumnExprImplrE   Nr   ri   r   c                   sX   t |tjr|d S t j||fd|i|}t |tjrF|d S | ||| d S Nrf   )rB   r   ExpressionElementRolelabelr   r   rx   )r   r?   r   rf   r^   newr   rE   rF   r   G  s    
 
z)LabeledColumnExprImpl._implicit_coercions)N)r   r   r   r   r   r   rE   rE   r   rF   r   D  s    r   c                      sL   e Zd ZdZdZdZdZedej	Z
d
dd fddZddd	Z  ZS )ColumnsClauseImplrE   Tz^\w\S*$N)r   c                  sF   |s*t |tr*dddd |D  d}t j|f|||d|S )NzDid you mean to say select(z, c                 s  s   | ]}t |V  qd S r_   )r   )r   erE   rE   rF   	<genexpr>h  s     z8ColumnsClauseImpl._raise_for_expected.<locals>.<genexpr>z)?r   )rB   rW   joinr   rx   )r   r?   rf   r   r   r^   r   rE   rF   rx   c  s      z%ColumnsClauseImpl._raise_for_expectedc                 C  sL   t |}| j| }tdt||r2d|f nd|r<dndd d S )NzTextual column expression %(column)r %(argname)sshould be explicitly declared with text(%(column)r), or use %(literal_column)s(%(column)r) for more specificityr   r   literal_columnr   )r   rf   r  )rI   _guess_straight_columnmatchr   r[   r   r   )r   r?   rf   Zguess_is_literalrE   rE   rF   r   p  s    
z ColumnsClauseImpl._text_coercion)NN)N)r   r   r   r   r   r   r   recompileIr	  rx   r   r   rE   rE   r   rF   r  Z  s      r  c                   @  s   e Zd ZdZdS )ReturnsRowsImplrE   Nr   rE   rE   rE   rF   r    s   r  c                      s>   e Zd ZdZddddZddddddd fd	d
Z  ZS )StatementImplrE   Nr   c                K  sF   ||k	rBt |tsBz
|j W n$ tk
r@   td| d Y nX |S )Na  Object %r should not be used directly in a SQL statement context, such as passing to methods such as session.execute().  This usage will be disallowed in a future release.  Please use Core select() / update() / delete() etc. with Session.execute() and other statement execution methods.1.4)rB   rI   Z_execute_on_connectionrw   r   warn_deprecated)r   r   rm   rf   r^   rE   rE   rF   r~     s    
 

zStatementImpl._post_coercionr   ri   r   c                   s(   |j r
|S t j||fd|i|S d S r   )r   r   r   r   r   rE   rF   r     s     z!StatementImpl._implicit_coercions)N)r   r   r   r   r~   r   r   rE   rE   r   rF   r    s
    r  c                   @  s(   e Zd ZdZdddddddddZdS )	SelectStatementImplrE   Nr   ri   r   c                 K  s    |j r| S | ||| d S r_   )_is_text_clausecolumnsrx   r   rE   rE   rF   r     s    z'SelectStatementImpl._implicit_coercions)Nr   rE   rE   rE   rF   r    s    r  c                   @  s   e Zd ZdZdS )
HasCTEImplrE   Nr   rE   rE   rE   rF   r    s   r  c                   @  s   e Zd ZdZdS )	IsCTEImplrE   Nr   rE   rE   rE   rF   r    s   r  c                	   @  sB   e Zd ZdZdZddddZdddd	d	d
dd	d	dddZdS )JoinTargetImplrE   TNr   c                K  s   |  || d S r_   r   r   rE   rE   rF   rr     s    z JoinTargetImpl._literal_coercionF)legacyr   ri   r>   )r?   r   rf   r  r^   r@   c                K  s>   t |tjr|S |r,|jr,tjddd |S | ||| d S NzImplicit coercion of SELECT and textual SELECT constructs into FROM clauses is deprecated; please call .subquery() on any Core select or ORM Query object in order to produce a subquery object.r  )version)rB   r   r   r   r   r  rx   )r   r?   r   rf   r  r^   rE   rE   rF   r     s    	
z"JoinTargetImpl._implicit_coercions)N)r   r   r   r   rs   rr   r   rE   rE   rE   rF   r    s    r  c                
   @  sB   e Zd ZdZdddddddddddd	d
dZddddZdS )FromClauseImplrE   NFT)rd   allow_selectr   ri   r>   )r?   r   rf   rd   r  r^   r@   c                K  sH   |j r,|r| S |rDtjddd |jS n|jr6|S | ||| d S r  )r   Zsubqueryr   r  _implicit_subqueryr  rx   )r   r?   r   rf   rd   r  r^   rE   rE   rF   r     s    
z"FromClauseImpl._implicit_coercions)
deannotatec                K  s   |r|  S |S d S r_   )Z_deannotate)r   r?   r  r^   rE   rE   rF   r~     s    zFromClauseImpl._post_coercion)N)r   r   r   r   r   r~   rE   rE   rE   rF   r    s    r  c                	   @  s0   e Zd ZdZddddddddddd	d
ZdS )StrictFromClauseImplrE   NF)r  r   ri   r>   )r?   r   rf   r  r^   r@   c                K  s0   |j r|rtjddd |jS | ||| d S r  )r   r   r  r  rx   )r   r?   r   rf   r  r^   rE   rE   rF   r     s    	
z(StrictFromClauseImpl._implicit_coercions)Nr   rE   rE   rE   rF   r    s
    r  c                   @  s    e Zd ZdZdddddZdS )AnonymizedFromClauseImplrE   FN)flatr   c                K  s   |d kst |j|dS )N)r!  )rz   Z_anonymous_fromclause)r   r?   r!  r   r^   rE   rE   rF   r~   1  s    z'AnonymizedFromClauseImpl._post_coercionr   rE   rE   rE   rF   r   .  s   r   c                   @  s   e Zd ZdZdd ZdS )DMLTableImplrE   c                 K  s   d|j kr|j d S |S d S )NZ	dml_table)Z_annotationsr   rE   rE   rF   r~   :  s    

zDMLTableImpl._post_coercionNr   rE   rE   rE   rF   r"  7  s   r"  c                   @  s(   e Zd ZdZdddddddddZdS )	DMLSelectImplrE   Nr   ri   r   c                 K  s<   |j r*t|tjr |jjr |jS | S n| ||| d S r_   )r   rB   r#   r   r?   r   r   rx   r   rE   rE   rF   r   D  s    

z!DMLSelectImpl._implicit_coercions)Nr   rE   rE   rE   rF   r#  A  s    r#  c                      s"   e Zd ZdZd fdd	Z  ZS )CompoundElementImplrE   Nc                   s>   t |tjr|jrd}q"d}nd }t j|f|||d|S )NzFUse the plain select() object without calling .subquery() or .alias().z9To SELECT from any FROM clause, use the .select() method.r   )rB   r   ZFromClauseRoleZ_is_subqueryr   rx   r   r   rE   rF   rx   Z  s       z'CompoundElementImpl._raise_for_expected)NNr   rE   rE   r   rF   r$  W  s   r$  ZRoleZImpl)
__future__r   collections.abcabcrH   r   r  ry   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   _typingr   baser   r   	cache_keyr   r   r   r   r   Zutil.typingr   r    r!   r"   r#   r$   r%   r&   r'   r(   Zdmlr)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   ZSQLRoler9   r;   Z
StringRoler<   r=   rG   rJ   rR   r\   ra   r   r   r   r   r   r   r   r[   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   ZByOfRoler   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r  r  r   r"  r#  r$  rp   dirr   rt   clsendswithreplaceglobalsr   r  Zee_implintr>   rI   floatZpy_typerE   rE   rE   rF   <module>	   sR  &		 t7	)+V			
)*'%	


