o Fh<@sddlZddlmZmZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZdd lmZed d ZGd d d eZdS)N) lru_cachepartial)settings)BaseDatabaseOperations)InetJsonberrors is_psycopg3mogrifysplit_tzname_delta) OnConflict)Cast)_lazy_re_compilecCs|durtjSttj|dS)N)cls)jsondumpsr)encoderra/var/www/html/loop/nvenv/lib/python3.10/site-packages/django/db/backends/postgresql/operations.pyget_json_dumpssrcseZdZdZdZegdZddddZer+dd l m Z e j e j e j e j e j e j d Zd d Zed ZddZd`ddZddZddZddZddZddZddZdd Zd`d!d"Zd#d$Zd%d&Zd`d'd(Zd)d*Zd+d,Z d-d.Z!d/d0Z"d1d2Z#d3d3d4d5d6Z$d7d8Z%dad9d:Z&d;d<Z'd=d>Z(d?d@Z)dAdBZ*erdCdDZ+ndEdDZ+dFdGZ,dHdIZ-erdJdKZ.dLdMZ/dNdOZ0dPdQZ1dbdRdSZ2dTdUZ3dVdWZ4fdXdYZ5d`fdZd[ Z6fd\d]Z7fd^d_Z8Z9S)cDatabaseOperationsvarcharEXPLAIN)ANALYZEBUFFERSCOSTSSETTINGSSUMMARYTIMINGVERBOSEWALintegerbigintsmallint) AutoField BigAutoFieldSmallAutoFieldr)numeric)SmallIntegerField IntegerFieldBigIntegerFieldPositiveSmallIntegerFieldPositiveIntegerFieldPositiveBigIntegerFieldcCs.|}|dvrd||jddSdS)N)GenericIPAddressFieldIPAddressField TimeField UUIDFieldzCAST(%%s AS %s)(r%s)get_internal_typedb_type connectionsplit)self output_field internal_typerrrunification_cast_sql;s z'DatabaseOperations.unification_cast_sqlz[A-Z_]+cCs|dkr d|d|fS|dkrd|d|fS|dkr$d|d|fS|}|j|s5td |d |d |d|fS) Nweek_dayzEXTRACT(DOW FROM z) + 1 iso_week_dayzEXTRACT(ISODOW FROM )iso_yearzEXTRACT(ISOYEAR FROM zInvalid lookup type: zEXTRACT(z FROM )upper_extract_format_re fullmatch ValueErrorr9 lookup_typesqlparamsrrrdate_extract_sqlRs z#DatabaseOperations.date_extract_sqlNcC*||||\}}d|d|g|RfSNDATE_TRUNC(%s, r?_convert_sql_to_tzr9rFrGrHtznamerrrdate_trunc_sqlaz!DatabaseOperations.date_trunc_sqlcCs6t|\}}}|r|dkrdnd}|||S|S)N+-r )r9rPsignoffsetrrr_prepare_tzname_deltafs z(DatabaseOperations._prepare_tzname_deltacCs4|rtjr||}|dg||RfS||fS)Nz AT TIME ZONE %s)rUSE_TZrW)r9rGrHrP tzname_paramrrrrNms  z%DatabaseOperations._convert_sql_to_tzcC"||||\}}d|d|fS)Nr3z)::daterMr9rGrHrPrrrdatetime_cast_date_sqlsz)DatabaseOperations.datetime_cast_date_sqlcCrZ)Nr3)::timerMr[rrrdatetime_cast_time_sqlwr]z)DatabaseOperations.datetime_cast_time_sqlcCs@||||\}}|dkrd|ddg|RfS||||SNsecondz#EXTRACT(SECOND FROM DATE_TRUNC(%s, z)))rNrIrOrrrdatetime_extract_sql{sz'DatabaseOperations.datetime_extract_sqlcCrJrKrMrOrrrdatetime_trunc_sqlrRz%DatabaseOperations.datetime_trunc_sqlcCs.|dkrd|ddg|RfS||||Sr`)rIrErrrtime_extract_sqlsz#DatabaseOperations.time_extract_sqlcCrJ)NrLr^rMrOrrrtime_trunc_sqlsz!DatabaseOperations.time_trunc_sqlcCdS)Nz DEFERRABLE INITIALLY DEFERREDrr9rrrdeferrable_sqlz!DatabaseOperations.deferrable_sqlcCs|S)z Given a cursor object that has just performed an INSERT...RETURNING statement into a table, return the tuple of returned data. )fetchall)r9cursorrrrfetch_returned_insert_rowssz-DatabaseOperations.fetch_returned_insert_rowscCs@d}|dvr|dvr d}n |dvrd}nd}|dvrd |}|S) Nr4) iexactcontains icontains startswith istartswithendswith iendswithregexiregex)r0r/zHOST(%s)) CICharField CIEmailField CITextFieldz %s::citextz%s::text)rmrorqrsz UPPER(%s)r)r9rFr;lookuprrr lookup_casts zDatabaseOperations.lookup_castcCsdSNrrgrrrno_limit_valueriz!DatabaseOperations.no_limit_valuecCs|gSr{r)r9rGrrrprepare_sql_scriptsz%DatabaseOperations.prepare_sql_scriptcCs |dr |dr |Sd|S)N"z"%s")rprr)r9namerrr quote_nameszDatabaseOperations.quote_namecCst|||jSr{)r r7)r9rGrHrrr compose_sqlszDatabaseOperations.compose_sqlcCrf)Nz(SELECT set_config('TimeZone', %s, false)rrgrrrset_time_zone_sqlriz$DatabaseOperations.set_time_zone_sqlF)reset_sequences allow_cascadecsd|sgSddfdd|Dg}|r |d|r*|ddd|gS) NTRUNCATE, c3s |] }|VqdSr{) SQL_FIELDr).0tabler9stylerr sz/DatabaseOperations.sql_flush..zRESTART IDENTITYCASCADEz%s; ) SQL_KEYWORDjoinappend)r9rtablesrr sql_partsrrr sql_flushszDatabaseOperations.sql_flushc CsRg}|D]"}|d}|dpd}|d|d|||||fq|S)Nrcolumnidz7%s setval(pg_get_serial_sequence('%s','%s'), 1, false);SELECT)rr SQL_TABLErr)r9r sequencesrG sequence_info table_name column_namerrrsequence_reset_by_name_sqls z-DatabaseOperations.sequence_reset_by_name_sqlcCs |r d||Sd||S)NzUSING INDEX TABLESPACE %sz TABLESPACE %s)r)r9 tablespaceinlinerrrtablespace_sqlsz!DatabaseOperations.tablespace_sqlcCsddlm}g}|j}|D]G}|jjD]@}t||jrS|d|d| ||jj | |j | ||j | ||j |d|d| ||jj fnqq |S)Nr)modelszZ%s setval(pg_get_serial_sequence('%s','%s'), coalesce(max(%s), 1), max(%s) %s null) %s %s;rzIS NOTFROM) django.dbrr_meta local_fields isinstancer%rrrdb_tablerr)r9r model_listroutputqnmodelfrrrsequence_reset_sqls.    z%DatabaseOperations.sequence_reset_sqlcC|Sr{r)r9xrrrprep_for_iexact_queryriz(DatabaseOperations.prep_for_iexact_querycCrf)a Return the maximum length of an identifier. The maximum length of an identifier is 63 by default, but can be changed by recompiling PostgreSQL after editing the NAMEDATALEN macro in src/include/pg_config_manual.h. This implementation returns 63, but can be overridden by a custom database backend that inherits most of its behavior from this one. ?rrgrrrmax_name_lengths z"DatabaseOperations.max_name_lengthcCs0|rdd|D}dd|g|fSdggfS)NcSsg|] }|D]}|qqSrr)r param_listparamrrr %sz3DatabaseOperations.distinct_sql..zDISTINCT ON (%s)rDISTINCTr)r9fieldsrHrrr distinct_sql#s zDatabaseOperations.distinct_sqlcCs&z|||WStjyYdSwr{)rr DataErrorr9rkrGrHrrrlast_executed_query,s z&DatabaseOperations.last_executed_querycCs|jdur |jSdSr{)querydecoderrrrr4s  cs,|sdSfdd|D}dd|dfS)N)rcs,g|]}d|jjj|jfqS)z%s.%s)rrrrrrfieldrgrrr>s z.z RETURNING %srrr)r9rcolumnsrrgrreturn_insert_columns;s  z(DatabaseOperations.return_insert_columnscCs*dd|D}ddd|D}d|S)Ncss|]}d|VqdS)rNr)rrowrrrrIsz5DatabaseOperations.bulk_insert_sql..rcs|]}d|VqdS)z(%s)Nr)rrGrrrrJzVALUES r)r9rplaceholder_rowsplaceholder_rows_sql values_sqlrrrbulk_insert_sqlHsz"DatabaseOperations.bulk_insert_sqlcCs$|dus t|dr |S|j||S)Nresolve_expression)hasattrintegerfield_type_map)r9valuer;rrradapt_integerfield_valueOsz+DatabaseOperations.adapt_integerfield_valuecCrr{rr9rrrradapt_datefield_valueTriz(DatabaseOperations.adapt_datefield_valuecCrr{rrrrradapt_datetimefield_valueWriz,DatabaseOperations.adapt_datetimefield_valuecCrr{rrrrradapt_timefield_valueZriz(DatabaseOperations.adapt_timefield_valuecCrr{r)r9r max_digitsdecimal_placesrrradapt_decimalfield_value]riz+DatabaseOperations.adapt_decimalfield_valuecCs|rt|SdSr{)rrrrradapt_ipaddressfield_value`sz-DatabaseOperations.adapt_ipaddressfield_valuecCst|t|dS)N)r)rr)r9rrrrradapt_json_valueesz#DatabaseOperations.adapt_json_valuec sF|dkr|\}}|\}}g||R}d||f|fSt|||S)N DateFieldz(interval '1 day' * (%s - %s)))supersubtract_temporals) r9r;lhsrhslhs_sql lhs_paramsrhs_sql rhs_paramsrH __class__rrrhs z%DatabaseOperations.subtract_temporalsc si}|r!dd|D}|jD]}||d}|dur |||<qtj|fi|}|r1||d<|rC|dddd|D7}|S)NcSs"i|] \}}||r dndqS)truefalse)rA)rrrrrr tsz;DatabaseOperations.explain_query_prefix..FORMATz (%s)rcsr)z%s %sNr)rirrrrrz:DatabaseOperations.explain_query_prefix..)itemsexplain_optionspoprexplain_query_prefixr)r9formatoptionsextra valid_optionrprefixrrrrps    z'DatabaseOperations.explain_query_prefixcs\|tjkrdS|tjkr%ddt|j|dddt|j|DfSt||||S)NzON CONFLICT DO NOTHINGz ON CONFLICT(%s) DO UPDATE SET %srcSsg|] }|d|qS)z = EXCLUDED.rrrrrrs z=DatabaseOperations.on_conflict_suffix_sql..)r IGNOREUPDATErmaprron_conflict_suffix_sql)r9r on_conflict update_fields unique_fieldsrrrrs"    z)DatabaseOperations.on_conflict_suffix_sqlcs@t||||\}}||j||jkrt||}||fSr{)rprepare_join_on_clauser6r7r)r9 lhs_table lhs_field rhs_table rhs_fieldlhs_exprrhs_exprrrrrs  z)DatabaseOperations.prepare_join_on_clauser{)F)NN):__name__ __module__ __qualname__"cast_char_field_without_max_lengthexplain_prefix frozensetrcast_data_typesr psycopg.typesr(Int2Int4Int8rr<rrBrIrQrWrNr\r_rbrcrdrerhrlrzr|r}rrrrrrrrrrrrrrrrrrrrrrrr __classcell__rrrrrs~       "      r)r functoolsrr django.confr"django.db.backends.base.operationsr)django.db.backends.postgresql.psycopg_anyrrrr r django.db.backends.utilsr django.db.models.constantsr django.db.models.functionsrdjango.utils.regex_helperrrrrrrrs