o (S-gq<@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)encoderrg/var/www/html/map_library/venv/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%d_d9d:Z&d;d<Z'd=d>Z(d?d@Z)dAdBZ*erdCdDZ+ndEdDZ+dFdGZ,erdHdIZ-dJdKZ.dLdMZ/dNdOZ0d`dPdQZ1dRdSZ2dTdUZ3fdVdWZ4d^fdXdY Z5fdZd[Z6fd\d]Z7Z8S)aDatabaseOperationsvarcharEXPLAIN) ANALYZEBUFFERSCOSTS GENERIC_PLANSETTINGSSUMMARYTIMINGVERBOSEWALintegerbigintsmallint) 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 ValueErrorr: lookup_typesqlparamsrrrdate_extract_sqlSs z#DatabaseOperations.date_extract_sqlNcC*||||\}}d|d|g|RfSNDATE_TRUNC(%s, r@_convert_sql_to_tzr:rGrHrItznamerrrdate_trunc_sqlbz!DatabaseOperations.date_trunc_sqlcCs6t|\}}}|r|dkrdnd}|||S|S)N+-r )r:rQsignoffsetrrr_prepare_tzname_deltags z(DatabaseOperations._prepare_tzname_deltacCs4|rtjr||}|dg||RfS||fS)Nz AT TIME ZONE %s)rUSE_TZrX)r:rHrIrQ tzname_paramrrrrOns  z%DatabaseOperations._convert_sql_to_tzcC"||||\}}d|d|fS)Nr4z)::daterNr:rHrIrQrrrdatetime_cast_date_sqltz)DatabaseOperations.datetime_cast_date_sqlcCr[)Nr4)::timerNr\rrrdatetime_cast_time_sqlxr^z)DatabaseOperations.datetime_cast_time_sqlcCs@||||\}}|dkrd|ddg|RfS||||SNsecondz#EXTRACT(SECOND FROM DATE_TRUNC(%s, z)))rOrJrPrrrdatetime_extract_sql|sz'DatabaseOperations.datetime_extract_sqlcCrKrLrNrPrrrdatetime_trunc_sqlrSz%DatabaseOperations.datetime_trunc_sqlcCs.|dkrd|ddg|RfS||||Sra)rJrFrrrtime_extract_sqlsz#DatabaseOperations.time_extract_sqlcCrK)NrMr_rNrPrrrtime_trunc_sqlsz!DatabaseOperations.time_trunc_sqlcCdS)Nz DEFERRABLE INITIALLY DEFERREDrr:rrrdeferrable_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)r:cursorrrrfetch_returned_insert_rowssz-DatabaseOperations.fetch_returned_insert_rowscCs2d}|dvr|dvr d}nd}|dvrd|}|S)Nr5) iexactcontains icontains startswith istartswithendswith iendswithregexiregex)r1r0zHOST(%s)z%s::text)rnrprrrtz UPPER(%s)r)r:rGr<lookuprrr lookup_casts zDatabaseOperations.lookup_castcCsdSNrrhrrrno_limit_valuerjz!DatabaseOperations.no_limit_valuecCs|gSryr)r:rHrrrprepare_sql_scriptsz%DatabaseOperations.prepare_sql_scriptcCs |dr |dr |Sd|S)N"z"%s")rqrs)r:namerrr quote_nameszDatabaseOperations.quote_namecCst|||jSry)r r8)r:rHrIrrr compose_sqlszDatabaseOperations.compose_sqlcCrg)Nz(SELECT set_config('TimeZone', %s, false)rrhrrrset_time_zone_sqlrjz$DatabaseOperations.set_time_zone_sqlF)reset_sequences allow_cascadecsd|sgSddfdd|Dg}|r |d|r*|ddd|gS) NTRUNCATE, c3s |] }|VqdSry) SQL_FIELDr~).0tabler:stylerr sz/DatabaseOperations.sql_flush..zRESTART IDENTITYCASCADEz%s; ) SQL_KEYWORDjoinappend)r:rtablesrr 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_TABLEr~r)r:r sequencesrH 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~)r: 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)r:r model_listroutputqnmodelfrrrsequence_reset_sqls.    z%DatabaseOperations.sequence_reset_sqlcC|Sryr)r:xrrrprep_for_iexact_queryrjz(DatabaseOperations.prep_for_iexact_querycCrg)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. ?rrhrrrmax_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)r:fieldsrIrrr distinct_sql!s zDatabaseOperations.distinct_sqlcCsR|jjjrz|||WStjyYdSw|jr'|jjdur'|jjSdSry) r8featuresuses_server_side_bindingrr DataError_queryquerydecoder:rlrHrIrrrlast_executed_query*s  z&DatabaseOperations.last_executed_querycCs|jdur |jSdSry)rrrrrrr7s  cs,|sdSfdd|D}dd|dfS)N)rcs,g|]}d|jjj|jfqS)z%s.%s)r~rrrrrfieldrhrrrAs z.z RETURNING %srrr)r:rcolumnsrrhrreturn_insert_columns>s  z(DatabaseOperations.return_insert_columnscCs$|dus t|dr |S|j||S)Nresolve_expression)hasattrintegerfield_type_map)r:valuer<rrradapt_integerfield_valueMsz+DatabaseOperations.adapt_integerfield_valuecCrryrr:rrrradapt_datefield_valueRrjz(DatabaseOperations.adapt_datefield_valuecCrryrrrrradapt_datetimefield_valueUrjz,DatabaseOperations.adapt_datetimefield_valuecCrryrrrrradapt_timefield_valueXrjz(DatabaseOperations.adapt_timefield_valuecCrryr)r:r max_digitsdecimal_placesrrradapt_decimalfield_value[rjz+DatabaseOperations.adapt_decimalfield_valuecCs|rt|SdSry)rrrrradapt_ipaddressfield_value^sz-DatabaseOperations.adapt_ipaddressfield_valuecCst|t|dS)N)r)rr)r:rrrrradapt_json_valuecsz#DatabaseOperations.adapt_json_valuec sF|dkr|\}}|\}}g||R}d||f|fSt|||S)N DateFieldz(interval '1 day' * (%s - %s)))supersubtract_temporals) r:r<lhsrhslhs_sql lhs_paramsrhs_sql rhs_paramsrI __class__rrrfs z%DatabaseOperations.subtract_temporalsc si}|r!dd|D}|jD]}||d}|dur |||<qtj|fi|}|r1||d<|rC|dddd|D7}|S)NcSs"i|] \}}||r dndqS)truefalse)rB)rr}rrrr rsz;DatabaseOperations.explain_query_prefix..FORMATz (%s)rcss|]}d|VqdS)z%s %sNr)rirrrr~sz:DatabaseOperations.explain_query_prefix..)itemsexplain_optionspoprexplain_query_prefixr)r:formatoptionsextra valid_optionrprefixrrrrns    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 IGNOREUPDATErmapr~ron_conflict_suffix_sql)r:r on_conflict update_fields unique_fieldsrrrrs"    z)DatabaseOperations.on_conflict_suffix_sqlcs@t||||\}}||j||jkrt||}||fSry)rprepare_join_on_clauser7r8r)r: lhs_table lhs_field rhs_table rhs_fieldlhs_exprrhs_exprrrrrs  z)DatabaseOperations.prepare_join_on_clausery)F)NN)9__name__ __module__ __qualname__"cast_char_field_without_max_lengthexplain_prefix frozensetrcast_data_typesr psycopg.typesr)Int2Int4Int8rr=rrCrJrRrXrOr]r`rcrdrerfrirmrxrzr{r~rrrrrrrrrrrrrrrrrrrrrr __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