B CVvg@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-install-o3oxmrkh/Django/django/db/models/functions/comparison.pyr sz Cast.__init__c s"|j||d<tj||f|S)Ndb_type)rZ cast_db_typer as_sql)r compiler connection extra_context)r 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_string)r 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 rrrr)r rr as_oracle-szCast.as_oracle) __name__ __module__ __qualname____doc__functionrr rrrr __classcell__rr)r rrs    rcs0eZdZdZdZfddZfddZZS)Coalescez:Return, from left to right, the first non-null expression.ZCOALESCEcs&t|dkrtdtj||dS)Nz+Coalesce must take at least two expressions)len ValueErrorr r )r expressionsextra)r 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 Dsz&Coalesce.as_oracle..)rrcopyZset_source_expressionsget_source_expressionsr r'r)r rrrclone)r rrr >s zCoalesce.as_oracle)r!r"r#r$r%r r r&rr)r 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 r5)r rrr Ps zCollate.__init__c s(|d|j|jtj||f|S)Nr5) setdefaultopsZ quote_namer5r r)r rrr)r rrrVszCollate.as_sql) r!r"r#r%rrr3r rr&rr)r rr2Js  r2cs0eZdZdZdZfddZfddZZS)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 expressions)r)r*r r )r r+r,)r rrr es zGreatest.__init__c stj||fddi|S)zUse the MAX function on SQLite.r%MAX)r r)r rrr)r rrrjszGreatest.as_sqlite)r!r"r#r$r%r rr&rr)r rr8[s r8csBeZdZdZeZfddZfddZddZdd Z Z S) JSONObjectZ JSON_OBJECTc s<g}x&|D]\}}|t||fqWtj|dS)N)itemsextendrr r )r fieldsr+keyvalue)r rrr sszJSONObject.__init__c s"|jjstdtj||f|S)Nz7JSONObject() is not supported on this database backend.)featuresZhas_json_object_functionrr r)r rrr)r rrryszJSONObject.as_sqlcKs|j||fddi|S)Nr%ZJSONB_BUILD_OBJECT)r)r rrrrrr as_postgresqls zJSONObject.as_postgresqlcKs*Gddd}|j||f|dd|S)Nc@seZdZddZdS)z'JSONObject.as_oracle..ArgJoinercSs2ddt|ddd|dddD}d|S)NcSsg|]}d|qS)z VALUE )join)r-argrrrr.sz@JSONObject.as_oracle..ArgJoiner.join..r(z, )ziprB)r argsrrrrBs(z,JSONObject.as_oracle..ArgJoiner.joinN)r!r"r#rBrrrr ArgJoinersrGz,%(function)s(%(expressions)s RETURNING CLOB))Z arg_joinerr)r)r rrrrGrrrr szJSONObject.as_oracle) r!r"r#r%rrr rrAr r&rr)r 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 expressions)r)r*r r )r r+r,)r rrr s zLeast.__init__c stj||fddi|S)zUse the MIN function on SQLite.r%ZMIN)r r)r rrr)r rrrszLeast.as_sqlite)r!r"r#r$r%r rr&rr)r rrHs rHcs$eZdZdZdZfddZZS)NullIfZNULLIFr(c s:|d}t|tr(|jdkr(tdtj||f|S)Nrz2Oracle does not allow Value(None) for expression1.)r0 isinstancerr?r*r r)r rrrZ expression1)r rrr s zNullIf.as_oracle)r!r"r#r%arityr r&rr)r rrIsrIN)r$Z django.dbrZdjango.db.models.expressionsrrZdjango.db.models.fields.jsonrZdjango.utils.regex_helperrrr'r2r8r:rHrIrrrrs   -(