U e(!@sdZddlmZddlmZmZddlmZddlm Z ddl m Z GdddeZ Gd d d eZ Gd d d eZGd ddeZGdddeZGdddeZGdddeZdS)z;Database functions that do comparisons or type conversions.)NotSupportedError)FuncValue) TextField) JSONField)_lazy_re_compilecs\eZdZdZdZdZfddZfddZfdd Zd d Z d d Z fddZ 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-lctamlir/django/db/models/functions/comparison.pyr sz Cast.__init__c s"|j||d<tj||f|S)Ndb_type)r Z cast_db_typer as_sqlr compiler connection extra_contextrrrrsz 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))r rr rinsert) r rrrrrZsqlparams format_stringrrr as_sqlites2   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)r get_internal_typeZmysql_is_mariadbr)r rrrr output_typerrras_mysql(s z Cast.as_mysqlcKs|j||fddi|S)Nrz(%(expressions)s)::%(db_type)srrrrr as_postgresql3szCast.as_postgresqlc s<|jdkr,d}tj||fd|i|S|j||f|S)Nrz JSON_QUERY(%(expressions)s, '$')r)r r r r)r rrrrrrr as_oracle>szCast.as_oracle) __name__ __module__ __qualname____doc__functionrr rrr"r$r% __classcell__rrrrr s     rcs<eZdZdZdZfddZeddZfddZZ S) 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 expressionsextrarrrr Ms zCoalesce.__init__cCs0|D]"}|j}|tks"|dk r|SqdSN)get_source_expressionsempty_result_set_valueNotImplemented)r r resultrrrr6Rs   zCoalesce.empty_result_set_valuec sT|jdkrD|}|dd|Dtt|j||f|S|j||f|S)NrcSsg|]}t|ddqS)ZTO_NCLOB)r*)r).0r rrr `sz&Coalesce.as_oracle..)r r copyset_source_expressionsr5r r,r)r rrrclonerrrr%ZszCoalesce.as_oracle) r&r'r(r)r*r propertyr6r%r+rrrrr,Hs   r,cs8eZdZdZdZedZfddZfddZZ S)CollateZCOLLATEz*%(expressions)s %(function)s %(collation)sz ^[\w\-]+$cs2|r|j|std|||_t|dS)NzInvalid collation name: %r.) collation_rematchr0 collationr r )r r rBrrrr ps zCollate.__init__c s(|d|j|jtj||f|S)NrB) setdefaultopsZ quote_namerBr rrrrrrvszCollate.as_sql) r&r'r(r*rrr@r rr+rrrrr?is  r?cs0eZdZdZdZfddZfddZZS)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.r1rrrr s zGreatest.__init__c stj||fddi|S)zUse the MAX function on SQLite.r*MAXr rrrrrrszGreatest.as_sqliter&r'r(r)r*r rr+rrrrrE{s rEcsFeZdZdZeZfddZfddZfddZdd Z Z S) JSONObjectZ JSON_OBJECTc s8g}|D]\}}|t||fq tj|dSr4)itemsextendrr r )r fieldsr2keyvaluerrrr szJSONObject.__init__c s"|jjstdtj||f|S)Nz7JSONObject() is not supported on this database backend.)featuresZhas_json_object_functionrr rrrrrrs zJSONObject.as_sqlc sB|}|ddt|Dtt|j||fddi|S)NcSs,g|]$\}}|ddkr$t|tn|qS)r-r)rr)r9indexr rrrr:sz,JSONObject.as_postgresql..r*ZJSONB_BUILD_OBJECT)r;r< enumerater5r rIr)r rrrr;rrrr$s  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)r9argrrrr:sz@JSONObject.as_oracle..ArgJoiner.join..r-z, )ziprR)r argsrrrrRs(z,JSONObject.as_oracle..ArgJoiner.joinN)r&r'r(rRrrrr ArgJoinersrWz,%(function)s(%(expressions)s RETURNING CLOB))Z arg_joinerrr#)r rrrrWrrrr%szJSONObject.as_oracle) r&r'r(r*rr r rr$r%r+rrrrrIs    rIcs0eZdZdZdZfddZfddZZS)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.r1rrrr s zLeast.__init__c stj||fddi|S)zUse the MIN function on SQLite.r*ZMINrGrrrrrszLeast.as_sqliterHrrrrrXs rXcs$eZdZdZdZfddZZS)NullIfZNULLIFr-c s:|d}t|tr(|jdkr(tdtj||f|S)Nrz2Oracle does not allow Value(None) for expression1.)r5 isinstancerrNr0r r)r rrrZ expression1rrrr%s zNullIf.as_oracle)r&r'r(r*arityr%r+rrrrrYsrYN)r)Z django.dbrZdjango.db.models.expressionsrrZdjango.db.models.fieldsrZdjango.db.models.fields.jsonrZdjango.utils.regex_helperrrr,r?rErIrXrYrrrrs    ?!/