3 (h@sdZddlmZmZmZmZGdddeZGdddeZGdddeZGd d d eZ Gd d d eZ Gd ddeZ GdddeZ GdddeZ GdddeZGdddeZGdddeZdS)z, Classes that represent database functions. )Func TransformValuefieldscsZeZdZdZdZdZejdejdej diZ fddZ fd d Z d d Z d dZZS)Castz3 Coerce an expression to a new field type. ZCASTz,%(function)s(%(expressions)s AS %(db_type)s)charzsigned integersignedcstt|j||ddS)N) output_field)superr__init__)self expressionr ) __class__C/tmp/pip-install-q3hcpn_q/Django/django/db/models/functions/base.pyr sz Cast.__init__c s.d|kr|jj||d<tt|j||f|S)Ndb_type) _output_fieldrr ras_sql)r compiler connection extra_context)rrrrsz Cast.as_sqlcCs6i}t|j}||jkr&|j||d<|j||f|S)Nr)typer mysql_typesr)r rrrZoutput_field_classrrras_mysqls   z Cast.as_mysqlcCs|j||ddS)Nz%(expressions)s::%(db_type)s)template)r)r rrrrr as_postgresql#szCast.as_postgresql)__name__ __module__ __qualname____doc__functionrrZ CharField IntegerFieldZ FloatFieldrr rrr __classcell__rr)rrrs   rcs0eZdZdZdZfddZfddZZS)CoalescezT Chooses, from left to right, the first non-null expression and returns it. ZCOALESCEcs*t|dkrtdtt|j||dS)Nz+Coalesce must take at least two expressions)len ValueErrorr r#r )r expressionsextra)rrrr .s zCoalesce.__init__csd|jjdkrXGdddtfdd|jD}|j}|j|tt|j||S|j||S)NZ TextFieldc@seZdZdZdS)z#Coalesce.as_oracle..ToNCLOBZTO_NCLOBN)rrrr rrrrToNCLOB7sr)csg|] }|qSrr).0r )r)rr ;sz&Coalesce.as_oracle..) r Zget_internal_typerget_source_expressionscopyset_source_expressionsr r#r)r rrr'clone)r)r)r as_oracle3s zCoalesce.as_oracle)rrrrr r r0r"rr)rrr#(s r#csDeZdZdZdZfddZfddZfddZd d ZZ S) ConcatPairz A helper class that concatenates two arguments together. This is used by `Concat` because not all backend databases support more than two arguments. ZCONCATc stt|j||f|dS)N)r r1r )r leftrightr()rrrr JszConcatPair.__init__cs |j}tt|j||dddS)Nz%(expressions)sz || )rZ arg_joiner)coalescer r1r)r rrZ coalesced)rrr as_sqliteMs zConcatPair.as_sqlitecstt|j||dddS)NZ CONCAT_WSz!%(function)s('', %(expressions)s))r r)r r1r)r rr)rrrrSs zConcatPair.as_mysqlcCs(|j}dd|jD}|j||S)NcSsg|]}t|tdqS))r#r)r*r rrrr+]sz'ConcatPair.coalesce..)r-r,r.)r cr'rrrr4Ys zConcatPair.coalesce) rrrrr r r5rr4r"rr)rrr1Bs    r1cs0eZdZdZdZdZfddZddZZS)Concatz Concatenates text fields together. Backends that result in an entire null expression when any arguments are null will wrap each argument in coalesce functions to ensure we always get a non-null result. Nz%(expressions)scs6t|dkrtd|j|}tt|j|f|dS)Nr$z)Concat must take at least two expressions)r%r&_pairedr r8r )r r'r(Zpaired)rrrr ls  zConcat.__init__cCs0t|dkrt|St|d|j|ddS)Nr$r)r%r1r9)r r'rrrr9rs zConcat._paired) rrrrr rr r9r"rr)rrr8cs  r8cs0eZdZdZdZfddZfddZZS)Greatesta  Chooses the maximum expression and returns it. If any expression is null the return value is database-specific: On Postgres, the maximum not-null expression is returned. On MySQL, Oracle, and SQLite, if any expression is null, null is returned. ZGREATESTcs*t|dkrtdtt|j||dS)Nr$z+Greatest must take at least two expressions)r%r&r r;r )r r'r()rrrr s zGreatest.__init__cstt|j||ddS)zUse the MAX function on SQLite.MAX)r )r r;r)r rr)rrrr5szGreatest.as_sqlite)rrrrr r r5r"rr)rrr;{s r;cs0eZdZdZdZfddZfddZZS)Leasta  Chooses the minimum expression and returns it. If any expression is null the return value is database-specific: On Postgres, the minimum not-null expression is returned. On MySQL, Oracle, and SQLite, if any expression is null, null is returned. ZLEASTcs*t|dkrtdtt|j||dS)Nr$z(Least must take at least two expressions)r%r&r r=r )r r'r()rrrr s zLeast.__init__cstt|j||ddS)zUse the MIN function on SQLite.ZMIN)r )r r=r)r rr)rrrr5szLeast.as_sqlite)rrrrr r r5r"rr)rrr=s r=cs4eZdZdZdZdZfddZfddZZS)Lengthz2Returns the number of characters in the expressionZLENGTHlengthc s0|jdtj}tt|j|fd|i|dS)Nr )poprr!r r>r )r r r(r )rrrr szLength.__init__cstt|j||ddS)NZ CHAR_LENGTH)r )r r>r)r rr)rrrrszLength.as_mysql) rrrrr lookup_namer rr"rr)rrr>s  r>c@seZdZdZdZdS)LowerZLOWERlowerN)rrrr rArrrrrBsrBcs*eZdZdZdfdd ZddZZS)NowZCURRENT_TIMESTAMPNc s.|dkrtj}tt|jfd|i|dS)Nr )rZ DateTimeFieldr rDr )r r r()rrrr sz Now.__init__cCs|j||ddS)NzSTATEMENT_TIMESTAMP())r)r)r rrrrrrszNow.as_postgresql)N)rrrrr rr"rr)rrrDsrDcs:eZdZdZd fdd ZfddZfddZZS) SubstrZ SUBSTRINGNc sdt|ds"|dkrtdt|}||g}|dk rNt|dsDt|}|j|tt|j||dS)z expression: the name of a field, or an expression returning a string pos: an integer > 0, or an expression returning an integer length: an optional number of characters to return Zresolve_expressionr:z'pos' must be greater than 0N)hasattrr&rappendr rEr )r r posr?r(r')rrrr s   zSubstr.__init__cstt|j||ddS)NSUBSTR)r )r rEr)r rr)rrrr5szSubstr.as_sqlitecstt|j||ddS)NrI)r )r rEr)r rr)rrrr0szSubstr.as_oracle)N)rrrr r r5r0r"rr)rrrEs rEc@seZdZdZdZdS)UpperZUPPERupperN)rrrr rArrrrrJsrJN)rZdjango.db.modelsrrrrrr#r1r8r;r=r>rBrDrErJrrrrs!!