o
    ÕèFhD!  ã                   @   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.Ú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/html/loop/nvenv/lib/python3.10/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 r-d}tƒ j||fd|i|¤Ž\}}|dkr!d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_typeÚ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_postgresql4   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.ÚCOALESCEc                    ó*   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   N   ó   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   r>   S   s   ÿzCoalesce.empty_result_set_valuec                    s\   | j  ¡ dkr$|  ¡ }| dd„ |  ¡ D ƒ¡ tt|ƒj||fi |¤ŽS | j||fi |¤ŽS )Nr   c                 S   s   g | ]}t |d d�‘qS )ÚTO_NCLOB)r/   )r   )Ú.0r   r   r   r   Ú
<listcomp>a   s    
ÿÿz&Coalesce.as_oracle.<locals>.<listcomp>)r
   r$   ÚcopyÚset_source_expressionsr=   r   r1   r   )r   r   r   r   Úcloner   r   r   r*   [   s   þÿzCoalesce.as_oracle)
r+   r,   r-   r.   r/   r   Úpropertyr>   r*   r0   r   r   r   r   r1   I   s    
r1   c                       s<   e Zd ZdZdZdZedƒZ‡ fdd„Z‡ fdd„Z	‡  Z
S )	ÚCollateÚCOLLATEz*%(expressions)s %(function)s %(collation)sFz^[\w-]+$c                    s2   |r| j  |¡std| ƒ‚|| _tƒ  |¡ d S )NzInvalid collation name: %r.)Úcollation_reÚmatchr7   Ú	collationr   r   )r   r   rL   r   r   r   r   r   s   zCollate.__init__c                    s,   |  d|j | j¡¡ tƒ j||fi |¤ŽS )NrL   )Ú
setdefaultÚopsÚ
quote_namerL   r   r   r   r   r   r   r   x   s   zCollate.as_sql)r+   r,   r-   r/   r   Úallowed_defaultr   rJ   r   r   r0   r   r   r   r   rH   j   s    rH   c                       ó0   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.
    ÚGREATESTc                    r3   )Nr4   z+Greatest must take at least two expressionsr5   r8   r   r   r   r   ˆ   r;   zGreatest.__init__c                    ó   t ƒ j||fddi|¤ŽS )zUse the MAX function on SQLite.r/   ÚMAX©r   r"   r   r   r   r   r"   �   ó   zGreatest.as_sqlite©r+   r,   r-   r.   r/   r   r"   r0   r   r   r   r   rR   }   ó
    rR   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 )
Ú
JSONObjectÚJSON_OBJECTc                    s8   g }|  ¡ D ]\}}| t|ƒ|f¡ qtƒ j|Ž  d S r<   )ÚitemsÚextendr   r   r   )r   Úfieldsr9   Ú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.)ÚfeaturesÚ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 )r4   r   )r   r   )rB   Úindexr   r   r   r   rC   ¦   s    ÿÿz,JSONObject.as_postgresql.<locals>.<listcomp>r/   ÚJSONB_BUILD_OBJECT)rD   rE   Ú	enumerater=   r   rZ   r   )r   r   r   r   rD   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)rB   Úargr   r   r   rC   µ   s    z@JSONObject.as_oracle.<locals>.ArgJoiner.join.<locals>.<listcomp>r4   é   z, )Úziprf   )r   Úargsr   r   r   rf   ´   s   (
z,JSONObject.as_oracle.<locals>.ArgJoiner.joinN)r+   r,   r-   rf   r   r   r   r   Ú	ArgJoiner³   s    rk   z,%(function)s(%(expressions)s RETURNING CLOB))Ú
arg_joinerr   r(   )r   r   r   r   rk   r   r   r   r*   ²   s   þüûzJSONObject.as_oracle)r+   r,   r-   r/   r   r
   r   r   r)   r*   r0   r   r   r   r   rZ   ’   s    rZ   c                       rQ   )Ú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.
    ÚLEASTc                    r3   )Nr4   z(Least must take at least two expressionsr5   r8   r   r   r   r   Ì   r;   zLeast.__init__c                    rT   )zUse the MIN function on SQLite.r/   ÚMINrV   r   r   r   r   r"   Ñ   rW   zLeast.as_sqliterX   r   r   r   r   rm   Á   rY   rm   c                       s$   e Zd ZdZdZ‡ fdd„Z‡  ZS )ÚNullIfÚNULLIFr4   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.)r=   Ú
isinstancer   r`   r7   r   r   )r   r   r   r   Úexpression1r   r   r   r*   Ú   s   zNullIf.as_oracle)r+   r,   r-   r/   Úarityr*   r0   r   r   r   r   rp   Ö   s    rp   N)r.   Ú	django.dbr   Údjango.db.models.expressionsr   r   Údjango.db.models.fieldsr   Údjango.db.models.fields.jsonr   Údjango.utils.regex_helperr   r   r1   rH   rR   rZ   rm   rp   r   r   r   r   Ú<module>   s    ?!/