U Ha@sdZddlmZddlmZmZddlmZddlm Z GdddeZ Gdd d eZ Gd d d eZ Gd d d eZ GdddeZGdddeZGdddeZdS)z;Database functions that do comparisons or type conversions.)NotSupportedError)FuncValue) JSONField)_lazy_re_compilecsTeZdZdZdZdZfddZfddZfdd Zd d Z fd d Z Z S)Castz)Coerce an expression to a new field type.ZCASTz,%(function)s(%(expressions)s AS %(db_type)s)cstj||ddS)N) output_field)super__init__)self expressionr __class__I/tmp/pip-unpacked-wheel-3jxiddxt/django/db/models/functions/comparison.pyr sz Cast.__init__c s"|j||d<tj||f|S)Ndb_type)rZ cast_db_typer as_sqlr compiler connection extra_contextr rrrsz Cast.as_sqlc  s|j|}|dkrZd}tj||fd|i|\}}|dkrBdnd}|d|||fS|dkrd }tj||fd|i|S|j||f|S) N>timedatetimezstrftime(%%s, %(expressions)s)templaterz%H:%M:%fz%Y-%m-%d %H:%M:%frdatezdate(%(expressions)s))rrr rinsert) r rrrrrZsqlparams format_stringr rr as_sqlites  zCast.as_sqlitecKsFd}|j}|dkrd}n|dkr.|jr.d}|j||fd|i|S)NZ FloatFieldz(%(expressions)s + 0.0)rz"JSON_EXTRACT(%(expressions)s, '$')r)rget_internal_typeZmysql_is_mariadbr)r rrrr output_typerrras_mysql"s z Cast.as_mysqlc s<|jdkr,d}tj||fd|i|S|j||f|S)Nrz JSON_QUERY(%(expressions)s, '$')r)rrr r)r rrrrr rr as_oracle-szCast.as_oracle) __name__ __module__ __qualname____doc__functionrr rrr!r" __classcell__rrr rrs    rcs0eZdZdZdZfddZfddZZS)Coalescez:Return, from left to right, the first non-null expression.ZCOALESCEcs&t|dkrtdtj||dS)Nz+Coalesce must take at least two expressionslen ValueErrorr r r expressionsextrar rrr 9s zCoalesce.__init__c sT|jdkrD|}|dd|Dtt|j||f|S|j||f|S)NZ TextFieldcSsg|]}t|ddqS)ZTO_NCLOB)r')r).0r rrr Csz&Coalesce.as_oracle..)rrcopyZset_source_expressionsget_source_expressionsr r)r)r rrrcloner rrr">s zCoalesce.as_oracle)r#r$r%r&r'r r"r(rrr rr)5s r)cs8eZdZdZdZedZfddZfddZZ S)CollateZCOLLATEz*%(expressions)s %(function)s %(collation)sz ^[\w\-]+$cs2|r|j|std|||_t|dS)NzInvalid collation name: %r.) collation_rematchr- collationr r )r r r9r rrr Ps zCollate.__init__c s(|d|j|jtj||f|S)Nr9) setdefaultopsZ quote_namer9r rrr rrrVszCollate.as_sql) r#r$r%r'rrr7r rr(rrr rr6Js  r6cs0eZdZdZdZfddZfddZZS)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. ZGREATESTcs&t|dkrtdtj||dS)Nr*z+Greatest must take at least two expressionsr+r.r rrr es zGreatest.__init__c stj||fddi|S)zUse the MAX function on SQLite.r'MAXr rrr rrrjszGreatest.as_sqliter#r$r%r&r'r rr(rrr rr<[s r<csBeZdZdZeZfddZfddZddZdd Z Z S) JSONObjectZ JSON_OBJECTc s8g}|D]\}}|t||fq tj|dS)N)itemsextendrr r )r fieldsr/keyvaluer rrr sszJSONObject.__init__c s"|jjstdtj||f|S)Nz7JSONObject() is not supported on this database backend.)featuresZhas_json_object_functionrr rrr rrrys zJSONObject.as_sqlcKs|j||fddi|S)Nr'ZJSONB_BUILD_OBJECTrrrrr as_postgresqlszJSONObject.as_postgresqlcKs*Gddd}|j||f|dd|S)Nc@seZdZddZdS)z'JSONObject.as_oracle..ArgJoinercSs2ddt|ddd|dddD}d|S)NcSsg|]}d|qS)z VALUE )join)r1argrrrr2sz@JSONObject.as_oracle..ArgJoiner.join..r*z, )ziprI)r argsrrrrIs(z,JSONObject.as_oracle..ArgJoiner.joinN)r#r$r%rIrrrr ArgJoinersrNz,%(function)s(%(expressions)s RETURNING CLOB))Z arg_joinerrrG)r rrrrNrrrr"szJSONObject.as_oracle) r#r$r%r'rrr rrHr"r(rrr rr@os   r@cs0eZdZdZdZfddZfddZZS)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. ZLEASTcs&t|dkrtdtj||dS)Nr*z(Least must take at least two expressionsr+r.r rrr s zLeast.__init__c stj||fddi|S)zUse the MIN function on SQLite.r'ZMINr>rr rrrszLeast.as_sqliter?rrr rrOs rOcs$eZdZdZdZfddZZS)NullIfZNULLIFr*c s:|d}t|tr(|jdkr(tdtj||f|S)Nrz2Oracle does not allow Value(None) for expression1.)r4 isinstancerrEr-r r)r rrrZ expression1r rrr"s zNullIf.as_oracle)r#r$r%r'arityr"r(rrr rrPsrPN)r&Z django.dbrZdjango.db.models.expressionsrrZdjango.db.models.fields.jsonrZdjango.utils.regex_helperrrr)r6r<r@rOrPrrrrs   -(