3
g                 @   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	 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)	JSONField)_lazy_re_compilec                   sT   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	 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__ c/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django/db/models/functions/comparison.pyr
      s    zCast.__init__c                s"   | j j||d< t j||f|S )Ndb_type)r   Zcast_db_typer	   as_sql)r   compiler
connectionextra_context)r   r   r   r      s    zCast.as_sqlc       	         s   | j j|}|d
krZd}t j||fd|i|\}}|dkrBdnd}|jd| ||fS |dkrd	}t j||fd|i|S | j||f|S )Ndatetimetimezstrftime(%%s, %(expressions)s)templatez%H:%M:%fz%Y-%m-%d %H:%M:%fr   datezdate(%(expressions)s)>   r   r   )r   r   r	   r   insert)	r   r   r   r   r   r   Zsqlparamsformat_string)r   r   r   	as_sqlite   s    zCast.as_sqlitec             K   sF   d }| j j }|dkrd}n|dkr.|jr.d}| j||fd|i|S )NZ
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                s<   | j j dkr,d}t j||fd|i|S | j||f|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    __classcell__r   r   )r   r   r      s   r   c                   s0   e Zd ZdZdZ f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|| d S )N   z+Coalesce must take at least two expressions)len
ValueErrorr	   r
   )r   expressionsextra)r   r   r   r
   9   s    zCoalesce.__init__c                sT   | j j dkrD| j }|jdd | j D  tt|j||f|S | j||f|S )NZ	TextFieldc             S   s   g | ]}t |d dqS )ZTO_NCLOB)r%   )r   ).0r   r   r   r   
<listcomp>D   s    z&Coalesce.as_oracle.<locals>.<listcomp>)r   r   copyZset_source_expressionsget_source_expressionsr	   r'   r   )r   r   r   r   clone)r   r   r   r    >   s    zCoalesce.as_oracle)r!   r"   r#   r$   r%   r
   r    r&   r   r   )r   r   r'   5   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   |o| j j|std| || _t j| d S )NzInvalid collation name: %r.)collation_rematchr*   	collationr	   r
   )r   r   r5   )r   r   r   r
   P   s    zCollate.__init__c                s(   |j d|jj| j t j||f|S )Nr5   )
setdefaultopsZ
quote_namer5   r	   r   )r   r   r   r   )r   r   r   r   V   s    zCollate.as_sql)
r!   r"   r#   r%   r   r   r3   r
   r   r&   r   r   )r   r   r2   J   s
   r2   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|| d S )Nr(   z+Greatest must take at least two expressions)r)   r*   r	   r
   )r   r+   r,   )r   r   r   r
   e   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   r   r   r   j   s    zGreatest.as_sqlite)r!   r"   r#   r$   r%   r
   r   r&   r   r   )r   r   r8   [   s   r8   c                   sB   e Zd ZdZe Z fddZ fddZdd Zdd	 Z	  Z
S )

JSONObjectZJSON_OBJECTc                s<   g }x&|j  D ]\}}|jt||f qW t j|  d S )N)itemsextendr   r	   r
   )r   fieldsr+   keyvalue)r   r   r   r
   s   s    zJSONObject.__init__c                s"   |j jstdt j||f|S )Nz7JSONObject() is not supported on this database backend.)featuresZhas_json_object_functionr   r	   r   )r   r   r   r   )r   r   r   r   y   s    zJSONObject.as_sqlc             K   s   | j ||fddi|S )Nr%   ZJSONB_BUILD_OBJECT)r   )r   r   r   r   r   r   r   as_postgresql   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j|S )Nc             S   s   g | ]}d j |qS )z VALUE )join)r-   argr   r   r   r.      s    z@JSONObject.as_oracle.<locals>.ArgJoiner.join.<locals>.<listcomp>r(      z, )ziprB   )r   argsr   r   r   rB      s    (z,JSONObject.as_oracle.<locals>.ArgJoiner.joinN)r!   r"   r#   rB   r   r   r   r   	ArgJoiner   s   rG   z,%(function)s(%(expressions)s RETURNING CLOB))Z
arg_joinerr   )r   )r   r   r   r   rG   r   r   r   r       s    zJSONObject.as_oracle)r!   r"   r#   r%   r   r   r
   r   rA   r    r&   r   r   )r   r   r:   o   s   r:   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|| d S )Nr(   z(Least must take at least two expressions)r)   r*   r	   r
   )r   r+   r,   )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%   ZMIN)r	   r   )r   r   r   r   )r   r   r   r      s    zLeast.as_sqlite)r!   r"   r#   r$   r%   r
   r   r&   r   r   )r   r   rH      s   rH   c                   s$   e Zd ZdZdZ fddZ  ZS )NullIfZNULLIFr(   c                s:   | j  d }t|tr(|jd kr(tdt j||f|S )Nr   z2Oracle does not allow Value(None) for expression1.)r0   
isinstancer   r?   r*   r	   r   )r   r   r   r   Zexpression1)r   r   r   r       s    zNullIf.as_oracle)r!   r"   r#   r%   arityr    r&   r   r   )r   r   rI      s   rI   N)r$   Z	django.dbr   Zdjango.db.models.expressionsr   r   Zdjango.db.models.fields.jsonr   Zdjango.utils.regex_helperr   r   r'   r2   r8   r:   rH   rI   r   r   r   r   <module>   s   -(