a
    ¥Q•h(!  ã                   @   s¸   d Z ddlmZ ddlmZmZ ddlmZ ddlm	Z	 ddl
m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dS )z;Database functions that do comparisons or type conversions.é    )ÚNotSupportedError)ÚFuncÚValue)Ú	TextField)Ú	JSONField)Ú_lazy_re_compilec                       s\   e Zd ZdZdZdZ‡ fdd„Z‡ fdd„Z‡ fdd	„Zd
d„ Z	dd„ Z
‡ fdd„Z‡  ZS )ÚCastz)Coerce an expression to a new field type.ZCASTz,%(function)s(%(expressions)s AS %(db_type)s)c                    s   t ƒ j||d� d S )N)Úoutput_field)ÚsuperÚ__init__)ÚselfÚ
expressionr	   ©Ú	__class__© ú^/var/www/sistema_ama/venv/lib/python3.9/site-packages/django/db/models/functions/comparison.pyr      s    zCast.__init__c                    s&   | j  |¡|d< tƒ j||fi |¤ŽS )NÚdb_type)r	   Úcast_db_typer
   Úas_sql©r   ÚcompilerÚ
connectionÚextra_contextr   r   r   r      s    zCast.as_sqlc           	         s”   | j  |¡}|dv rZd}tƒ j||fd|i|¤Ž\}}|dkrBdnd}| d|¡ ||fS |dkr€d	}tƒ j||fd|i|¤ŽS | j||fi |¤ŽS )
N>   ÚtimeÚdatetimezstrftime(%%s, %(expressions)s)Útemplater   z%H:%M:%fz%Y-%m-%d %H:%M:%fr   Údatezdate(%(expressions)s))r	   r   r
   r   Úinsert)	r   r   r   r   r   r   ÚsqlÚparamsÚformat_stringr   r   r   Ú	as_sqlite   s.    ÿÿÿ
ÿÿÿzCast.as_sqlitec                 K   sF   d }| j  ¡ }|dkrd}n|dkr.|jr.d}| j||fd|i|¤ŽS )NÚ
FloatFieldz(%(expressions)s + 0.0)r   z"JSON_EXTRACT(%(expressions)s, '$')r   )r	   Úget_internal_typeZmysql_is_mariadbr   )r   r   r   r   r   Úoutput_typer   r   r   Úas_mysql(   s    
zCast.as_mysqlc                 K   s   | j ||fddi|¤ŽS )Nr   z(%(expressions)s)::%(db_type)s©r   r   r   r   r   Úas_postgresql3   s    þýüzCast.as_postgresqlc                    s@   | j  ¡ dkr,d}tƒ j||fd|i|¤ŽS | j||fi |¤ŽS )Nr   z JSON_QUERY(%(expressions)s, '$')r   )r	   r#   r
   r   )r   r   r   r   r   r   r   r   Ú	as_oracle>   s    ÿÿÿzCast.as_oracle)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úfunctionr   r   r   r!   r%   r'   r(   Ú__classcell__r   r   r   r   r   	   s   r   c                       s<   e Zd ZdZdZ‡ fdd„Zedd„ ƒZ‡ fdd„Z‡  Z	S )	ÚCoalescez:Return, from left to right, the first non-null expression.ZCOALESCEc                    s*   t |ƒdk rtdƒ‚tƒ j|i |¤Ž d S )Né   z+Coalesce must take at least two expressions©ÚlenÚ
ValueErrorr
   r   ©r   ÚexpressionsÚextrar   r   r   r   M   s    zCoalesce.__init__c                 C   s0   |   ¡ D ]"}|j}|tu s"|d ur|  S qd S ©N)Úget_source_expressionsÚempty_result_set_valueÚNotImplemented)r   r   Úresultr   r   r   r9   R   s
    
zCoalesce.empty_result_set_valuec                    s\   | j  ¡ dkrH|  ¡ }| dd„ |  ¡ D ƒ¡ tt|ƒj||fi |¤ŽS | j||fi |¤ŽS )Nr   c                 S   s   g | ]}t |d d�‘qS )ZTO_NCLOB)r-   )r   )Ú.0r   r   r   r   Ú
<listcomp>`   s   ÿz&Coalesce.as_oracle.<locals>.<listcomp>)r	   r#   ÚcopyÚset_source_expressionsr8   r
   r/   r   )r   r   r   r   Úcloner   r   r   r(   Z   s    þÿzCoalesce.as_oracle)
r)   r*   r+   r,   r-   r   Úpropertyr9   r(   r.   r   r   r   r   r/   H   s   
r/   c                       s8   e Zd ZdZdZedƒZ‡ fdd„Z‡ fdd„Z‡  Z	S )ÚCollateZCOLLATEz*%(expressions)s %(function)s %(collation)sz	^[\w\-]+$c                    s2   |r| j  |¡std| ƒ‚|| _tƒ  |¡ d S )NzInvalid collation name: %r.)Úcollation_reÚmatchr3   Ú	collationr
   r   )r   r   rE   r   r   r   r   p   s    zCollate.__init__c                    s,   |  d|j | j¡¡ tƒ j||fi |¤ŽS )NrE   )Ú
setdefaultÚopsÚ
quote_namerE   r
   r   r   r   r   r   r   v   s    zCollate.as_sql)
r)   r*   r+   r-   r   r   rC   r   r   r.   r   r   r   r   rB   i   s
   rB   c                       s0   e Zd ZdZdZ‡ fdd„Z‡ fdd„Z‡  ZS )ÚGreatestzý
    Return the maximum expression.

    If any expression is null the return value is database-specific:
    On PostgreSQL, the maximum not-null expression is returned.
    On MySQL, Oracle, and SQLite, if any expression is null, null is returned.
    ZGREATESTc                    s*   t |ƒdk rtdƒ‚tƒ j|i |¤Ž d S )Nr0   z+Greatest must take at least two expressionsr1   r4   r   r   r   r   †   s    zGreatest.__init__c                    s   t ƒ j||fddi|¤ŽS )zUse the MAX function on SQLite.r-   ÚMAX©r
   r!   r   r   r   r   r!   ‹   s    zGreatest.as_sqlite©r)   r*   r+   r,   r-   r   r!   r.   r   r   r   r   rI   {   s   rI   c                       sF   e Zd ZdZeƒ Z‡ fdd„Z‡ fdd„Z‡ fdd„Zdd	„ Z	‡  Z
S )
Ú
JSONObjectZJSON_OBJECTc                    s8   g }|  ¡ D ]\}}| t|ƒ|f¡ qtƒ j|Ž  d S r7   )ÚitemsÚextendr   r
   r   )r   Úfieldsr5   ÚkeyÚvaluer   r   r   r   ”   s    zJSONObject.__init__c                    s&   |j jstdƒ‚tƒ j||fi |¤ŽS )Nz7JSONObject() is not supported on this database backend.)ÚfeaturesZhas_json_object_functionr   r
   r   r   r   r   r   r   š   s
    ÿzJSONObject.as_sqlc                    sB   |   ¡ }| dd„ t| ¡ ƒD ƒ¡ tt|ƒj||fddi|¤ŽS )Nc                 S   s,   g | ]$\}}|d  dkr$t |tƒ ƒn|‘qS )r0   r   )r   r   )r<   Úindexr   r   r   r   r=   ¤   s   ÿz,JSONObject.as_postgresql.<locals>.<listcomp>r-   ZJSONB_BUILD_OBJECT)r>   r?   Ú	enumerater8   r
   rM   r   )r   r   r   r   r>   r   r   r   r'   ¡   s    
þÿ
þýüzJSONObject.as_postgresqlc                 K   s*   G dd„ dƒ}| j ||f|ƒ ddœ|¤ŽS )Nc                   @   s   e Zd Zdd„ ZdS )z'JSONObject.as_oracle.<locals>.ArgJoinerc                 S   s2   dd„ t |d d d… |dd d… ƒD ƒ}d |¡S )Nc                 S   s   g | ]}d   |¡‘qS )z VALUE )Újoin)r<   Úargr   r   r   r=   ³   ó    z@JSONObject.as_oracle.<locals>.ArgJoiner.join.<locals>.<listcomp>r0   é   z, )ÚziprV   )r   Úargsr   r   r   rV   ²   s    (z,JSONObject.as_oracle.<locals>.ArgJoiner.joinN)r)   r*   r+   rV   r   r   r   r   Ú	ArgJoiner±   s   r\   z,%(function)s(%(expressions)s RETURNING CLOB))Ú
arg_joinerr   r&   )r   r   r   r   r\   r   r   r   r(   °   s    þüûzJSONObject.as_oracle)r)   r*   r+   r-   r   r	   r   r   r'   r(   r.   r   r   r   r   rM   �   s   rM   c                       s0   e Zd ZdZdZ‡ fdd„Z‡ fdd„Z‡  ZS )ÚLeastzó
    Return the minimum expression.

    If any expression is null the return value is database-specific:
    On PostgreSQL, return the minimum not-null expression.
    On MySQL, Oracle, and SQLite, if any expression is null, return null.
    ZLEASTc                    s*   t |ƒdk rtdƒ‚tƒ j|i |¤Ž d S )Nr0   z(Least must take at least two expressionsr1   r4   r   r   r   r   Ê   s    zLeast.__init__c                    s   t ƒ j||fddi|¤ŽS )zUse the MIN function on SQLite.r-   ÚMINrK   r   r   r   r   r!   Ï   s    zLeast.as_sqliterL   r   r   r   r   r^   ¿   s   r^   c                       s$   e Zd ZdZdZ‡ fdd„Z‡  ZS )ÚNullIfZNULLIFr0   c                    s>   |   ¡ d }t|tƒr(|jd u r(tdƒ‚tƒ j||fi |¤ŽS )Nr   z2Oracle does not allow Value(None) for expression1.)r8   Ú
isinstancer   rR   r3   r
   r   )r   r   r   r   Zexpression1r   r   r   r(   Ø   s    zNullIf.as_oracle)r)   r*   r+   r-   Úarityr(   r.   r   r   r   r   r`   Ô   s   r`   N)r,   Ú	django.dbr   Údjango.db.models.expressionsr   r   Údjango.db.models.fieldsr   Údjango.db.models.fields.jsonr   Údjango.utils.regex_helperr   r   r/   rB   rI   rM   r^   r`   r   r   r   r   Ú<module>   s   ?!/