a deL<@sddlZddlmZmZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZed d ZGd d d eZdS) N) lru_cachepartial)settings)BaseDatabaseOperations)InetJsonberrors is_psycopg3mogrifysplit_tzname_delta) OnConflict)_lazy_re_compilecCs|durtjSttj|dS)N)cls)jsondumpsr)encoderra/var/www/sistema_ama/venv/lib/python3.9/site-packages/django/db/backends/postgresql/operations.pyget_json_dumpssrcseZdZdZdZegdZddddZerVdd 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*erNdCdDZ+ndEdDZ+dFdGZ,dHdIZ-ertdJdKZ.dLdMZ/dNdOZ0dPdQZ1dbdRdSZ2dTdUZ3dVdWZ4fdXdYZ5dcfdZd[ Z6fd\d]Z7Z8S)dDatabaseOperationsZvarcharZEXPLAIN)ZANALYZEZBUFFERSZCOSTSZSETTINGSZSUMMARYZTIMINGVERBOSEZWALintegerZbigintZsmallint) AutoFieldZ BigAutoFieldZSmallAutoFieldr)numeric)ZSmallIntegerFieldZ IntegerFieldZBigIntegerFieldZPositiveSmallIntegerFieldZPositiveIntegerFieldZPositiveBigIntegerFieldcCs.|}|dvr*d||jddSdS)N)GenericIPAddressFieldIPAddressFieldZ TimeFieldZ UUIDFieldzCAST(%%s AS %s)(r%s)Zget_internal_typeZdb_type connectionsplit)selfZ output_field internal_typerrrunification_cast_sql:s z'DatabaseOperations.unification_cast_sqlz[A-Z_]+cCs|dkrd|d|fS|dkr0d|d|fS|dkrHd|d|fS|}|j|sjtd |d |d |d|fS) NZweek_dayzEXTRACT(DOW FROM z) + 1Z iso_week_dayzEXTRACT(ISODOW FROM )Ziso_yearzEXTRACT(ISOYEAR FROM zInvalid lookup type: zEXTRACT(z FROM )upper_extract_format_re fullmatch ValueErrorr! lookup_typesqlparamsrrrdate_extract_sqlQs z#DatabaseOperations.date_extract_sqlNcCs*||||\}}d|d|g|RfSNDATE_TRUNC(%s, r$_convert_sql_to_tzr!r*r+r,tznamerrrdate_trunc_sql`sz!DatabaseOperations.date_trunc_sqlcCs6t|\}}}|r2|dkrdnd}|||S|S)N+-r )r!r3signoffsetrrr_prepare_tzname_deltaes z(DatabaseOperations._prepare_tzname_deltacCs4|r,tjr,||}|dg||RfS||fS)Nz AT TIME ZONE %s)rZUSE_TZr9)r!r+r,r3Z tzname_paramrrrr1ls  z%DatabaseOperations._convert_sql_to_tzcCs"||||\}}d|d|fS)Nrz)::dater0r!r+r,r3rrrdatetime_cast_date_sqlrsz)DatabaseOperations.datetime_cast_date_sqlcCs"||||\}}d|d|fS)Nr)::timer0r:rrrdatetime_cast_time_sqlvsz)DatabaseOperations.datetime_cast_time_sqlcCs@||||\}}|dkr2d|ddg|RfS||||SNsecondz#EXTRACT(SECOND FROM DATE_TRUNC(%s, z)))r1r-r2rrrdatetime_extract_sqlzsz'DatabaseOperations.datetime_extract_sqlcCs*||||\}}d|d|g|RfSr.r0r2rrrdatetime_trunc_sqlsz%DatabaseOperations.datetime_trunc_sqlcCs.|dkr d|ddg|RfS||||Sr>)r-r)rrrtime_extract_sqlsz#DatabaseOperations.time_extract_sqlcCs*||||\}}d|d|g|RfS)Nr/r<r0r2rrrtime_trunc_sqlsz!DatabaseOperations.time_trunc_sqlcCsdS)Nz DEFERRABLE INITIALLY DEFERREDrr!rrrdeferrable_sqlsz!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. )Zfetchall)r!cursorrrrfetch_returned_insert_rowssz-DatabaseOperations.fetch_returned_insert_rowscCsTd}|dkr|dvrdS|dvr@|dvr.d}n|dvrz/DatabaseOperations.sql_flush..zRESTART IDENTITYZCASCADEz%s; ) SQL_KEYWORDjoinappend)r!rdZtablesr]r^Z sql_partsrrcr sql_flushszDatabaseOperations.sql_flushc CsRg}|D]D}|d}|dpd}|d|d|||||fq|S)Nrbcolumnidz7%s setval(pg_get_serial_sequence('%s','%s'), 1, false);SELECT)rjrh SQL_TABLErZr`)r!rd sequencesr+Z sequence_infoZ table_nameZ column_namerrrsequence_reset_by_name_sqls z-DatabaseOperations.sequence_reset_by_name_sqlcCs$|rd||Sd||SdS)NzUSING INDEX TABLESPACE %sz TABLESPACE %s)rZ)r!Z tablespaceinlinerrrtablespace_sqlsz!DatabaseOperations.tablespace_sqlcCsddlm}g}|j}|D]}|jjD]}t||jr&|d|d| ||jj | |j | ||j | ||j |d|d| ||jj fqq&q|S)Nr)modelszZ%s setval(pg_get_serial_sequence('%s','%s'), coalesce(max(%s), 1), max(%s) %s null) %s %s;rnzIS NOTZFROM) Z django.dbrtrZ_metaZ local_fields isinstancerrjrhrodb_tabler`rl)r!rdZ model_listrtoutputqnmodelfrrrsequence_reset_sqls*    z%DatabaseOperations.sequence_reset_sqlcCs|SrUr)r!xrrrprep_for_iexact_querysz(DatabaseOperations.prep_for_iexact_querycCsdS)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. ?rrDrrrmax_name_length s z"DatabaseOperations.max_name_lengthcCs4|r&dd|D}dd|g|fSdggfSdS)NcSsg|]}|D]}|q qSrr)raZ param_listparamrrr /rfz3DatabaseOperations.distinct_sql..zDISTINCT ON (%s)r_ZDISTINCTri)r!fieldsr,rrr distinct_sql-szDatabaseOperations.distinct_sqlcCs*z|||WStjy$YdS0dSrU)r[rZ DataErrorr!rFr+r,rrrlast_executed_query6sz&DatabaseOperations.last_executed_querycCs|jdur|jSdSrU)querydecoderrrrr>s  cs,|sdSfdd|D}dd|dfS)N)rcs,g|]$}d|jjj|jfqS)z%s.%s)rZrzrurwrlrafieldrDrrrHs  z.z RETURNING %sr_rr)r!rcolumnsrrDrreturn_insert_columnsEs  z(DatabaseOperations.return_insert_columnscCs*dd|D}ddd|D}d|S)Ncss|]}d|VqdS)r_Nr)rarowrrrreSrfz5DatabaseOperations.bulk_insert_sql..r_css|]}d|VqdS)z(%s)Nr)rar+rrrreTrfzVALUES r)r!rZplaceholder_rowsZplaceholder_rows_sqlZ values_sqlrrrbulk_insert_sqlRsz"DatabaseOperations.bulk_insert_sqlcCs$|dust|dr|S|j||S)NZresolve_expression)hasattrintegerfield_type_map)r!valuer"rrradapt_integerfield_valueYsz+DatabaseOperations.adapt_integerfield_valuecCs|SrUrr!rrrradapt_datefield_value^sz(DatabaseOperations.adapt_datefield_valuecCs|SrUrrrrradapt_datetimefield_valueasz,DatabaseOperations.adapt_datetimefield_valuecCs|SrUrrrrradapt_timefield_valuedsz(DatabaseOperations.adapt_timefield_valuecCs|SrUr)r!rZ max_digitsZdecimal_placesrrradapt_decimalfield_valuegsz+DatabaseOperations.adapt_decimalfield_valuecCs|r t|SdSrU)rrrrradapt_ipaddressfield_valuejsz-DatabaseOperations.adapt_ipaddressfield_valuecCst|t|dS)N)r)rr)r!rrrrradapt_json_valueosz#DatabaseOperations.adapt_json_valuec sF|dkr6|\}}|\}}g||R}d||f|fSt|||S)NZ DateFieldz(interval '1 day' * (%s - %s)))supersubtract_temporals) r!r"lhsrhsZlhs_sqlZ lhs_paramsZrhs_sqlZ rhs_paramsr, __class__rrrrs z%DatabaseOperations.subtract_temporalsc si}|rBdd|D}|jD] }||d}|dur |||<q tj|fi|}|rb||d<|r|dddd|D7}|S)NcSs"i|]\}}||rdndqS)truefalse)r%)rarYrrrr ~sz;DatabaseOperations.explain_query_prefix..ZFORMATz (%s)r_css|]}d|VqdS)z%s %sNr)rairrrrerfz:DatabaseOperations.explain_query_prefix..)itemsexplain_optionspoprexplain_query_prefixri)r!formatoptionsextraZ valid_optionrprefixrrrrzs    z'DatabaseOperations.explain_query_prefixcs\|tjkrdS|tjkrJddt|j|dddt|j|DfSt||||S)NzON CONFLICT DO NOTHINGz ON CONFLICT(%s) DO UPDATE SET %sr_cSsg|]}|d|qS)z = EXCLUDED.rrrrrrsz=DatabaseOperations.on_conflict_suffix_sql..)r ZIGNOREZUPDATErimaprZron_conflict_suffix_sql)r!rZ on_conflictZ update_fieldsZ unique_fieldsrrrrs"    z)DatabaseOperations.on_conflict_suffix_sql)N)N)N)F)NN)N)9__name__ __module__ __qualname__Z"cast_char_field_without_max_lengthZexplain_prefix frozensetrZcast_data_typesr Z psycopg.typesrZInt2ZInt4ZInt8rr#rr&r-r4r9r1r;r=r@rArBrCrErGrTrVrWrZr[r\rkrqrsr|r~rrrrrrrrrrrrrrr __classcell__rrrrrsz      ' "     r)r functoolsrrZ django.confrZ"django.db.backends.base.operationsrZ)django.db.backends.postgresql.psycopg_anyrrrr r Zdjango.db.backends.utilsr Zdjango.db.models.constantsr Zdjango.utils.regex_helperrrrrrrrs