U ‰d,ã@s`ddlmZddlmZddlmZddlmZddlmZedej dƒZGdd „d eƒZ d S) é)Ú namedtuple)ÚBaseDatabaseIntrospection)Ú FieldInfo)Ú TableInfo)ÚIndexr)Ú is_autofieldcs|eZdZdddddddddd d d d d d d d ddddœZdZgZ‡fdd„Zdd„Zdd„Zd dd„Z dd„Z dd„Z ‡Z S)!ÚDatabaseIntrospectionZ BooleanFieldZ BinaryFieldÚBigIntegerFieldÚSmallIntegerFieldÚ IntegerFieldZ TextFieldZ FloatFieldZGenericIPAddressFieldZ CharFieldZ DateFieldZ TimeFieldZ DateTimeFieldZ DurationFieldZ DecimalFieldZ UUIDFieldZ JSONField)ééééééi¼i½ieiii:i;iZi i¢iòi¤i† iÚZbtreecsLtƒ ||¡}|js$|jrHd|jkrH|dkr0dS|dkrReturn a list of table and view names in the current database.a SELECT c.relname, CASE WHEN c.relispartition THEN 'p' WHEN c.relkind IN ('m', 'v') THEN 'v' ELSE 't' END FROM pg_catalog.pg_class c LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace WHERE c.relkind IN ('f', 'm', 'p', 'r', 'v') AND n.nspname NOT IN ('pg_catalog', 'pg_toast') AND pg_catalog.pg_table_is_visible(c.oid) cs"g|]}|dˆjkrt|Ž‘qS)r)Úignored_tablesr©Ú.0Úrow©rrrÚ Isþz8DatabaseIntrospection.get_table_list..©ÚexecuteÚfetchall)rÚcursorrrrÚget_table_list7s ÿ þz$DatabaseIntrospection.get_table_listcsL| d|g¡dd„| ¡Dƒ‰| d|jj |¡¡‡fdd„|jDƒS)zi Return a description of the table with the DB-API cursor.description interface. aœ SELECT a.attname AS column_name, NOT (a.attnotnull OR (t.typtype = 'd' AND t.typnotnull)) AS is_nullable, pg_get_expr(ad.adbin, ad.adrelid) AS column_default, CASE WHEN collname = 'default' THEN NULL ELSE collname END AS collation, a.attidentity != '' AS is_autofield FROM pg_attribute a LEFT JOIN pg_attrdef ad ON a.attrelid = ad.adrelid AND a.attnum = ad.adnum LEFT JOIN pg_collation co ON a.attcollation = co.oid JOIN pg_type t ON a.atttypid = t.oid JOIN pg_class c ON a.attrelid = c.oid JOIN pg_namespace n ON c.relnamespace = n.oid WHERE c.relkind IN ('f', 'm', 'p', 'r', 'v') AND c.relname = %s AND n.nspname NOT IN ('pg_catalog', 'pg_toast') AND pg_catalog.pg_table_is_visible(c.oid) cSsi|]}|d|dd…“qS)réNr©rÚlinerrrÚ lsz?DatabaseIntrospection.get_table_description..zSELECT * FROM %s LIMIT 1c s6g|].}t|j|j|j|j|j|jfˆ|jžŽ‘qSr)rÚnameZ type_codeZ display_sizeZ internal_sizeZ precisionZscaler'©Z field_maprrr ps ÷úùz?DatabaseIntrospection.get_table_description..)r"r#Ú connectionÚopsZ quote_namer©rr$Ú table_namerr+rÚget_table_descriptionOsíÿ öz+DatabaseIntrospection.get_table_descriptionrcs$| dˆg¡‡fdd„| ¡DƒS)Naˆ SELECT s.relname AS sequence_name, a.attname AS colname FROM pg_class s JOIN pg_depend d ON d.objid = s.oid AND d.classid = 'pg_class'::regclass AND d.refclassid = 'pg_class'::regclass JOIN pg_attribute a ON d.refobjid = a.attrelid AND d.refobjsubid = a.attnum JOIN pg_class tbl ON tbl.oid = d.refobjid AND tbl.relname = %s AND pg_catalog.pg_table_is_visible(tbl.oid) WHERE s.relkind = 'S'; cs g|]}|dˆ|ddœ‘qS)rr&)r*ÚtableÚcolumnrr©r/rrr ’sÿz7DatabaseIntrospection.get_sequences..r!)rr$r/Z table_fieldsrr3rÚ get_sequences}sî þz#DatabaseIntrospection.get_sequencescCs | d|g¡dd„| ¡DƒS)z– Return a dictionary of {field_name: (field_name_other_table, other_table)} representing all foreign keys in the given table. a{ SELECT a1.attname, c2.relname, a2.attname FROM pg_constraint con LEFT JOIN pg_class c1 ON con.conrelid = c1.oid LEFT JOIN pg_class c2 ON con.confrelid = c2.oid LEFT JOIN pg_attribute a1 ON c1.oid = a1.attrelid AND a1.attnum = con.conkey[1] LEFT JOIN pg_attribute a2 ON c2.oid = a2.attrelid AND a2.attnum = con.confkey[1] WHERE c1.relname = %s AND con.contype = 'f' AND c1.relnamespace = c2.relnamespace AND pg_catalog.pg_table_is_visible(c1.oid) cSs"i|]}|d|d|df“qS)rér&rrrrrr)®sz7DatabaseIntrospection.get_relations..r!r.rrrÚ get_relations—s ðz#DatabaseIntrospection.get_relationsc Csi}| d|g¡| ¡D]L\}}}}}||dk|dk|dkrNt| dd¡ƒnd|dkd d|d œ||<q| d |j|g¡| ¡D]~\} }} } } } }}| |kr‚| |jkoº|  d ¡ oº|dk}|dgkrÊ|ng| dgkrÚ| ng| | dd d |rðtjn| ||dœ || <q‚|S)z· Retrieve any constraints or keys (unique, pk, fk, check, index) across one or more columns. Also retrieve the definition of expression-based indexes. aH SELECT c.conname, array( SELECT attname FROM unnest(c.conkey) WITH ORDINALITY cols(colid, arridx) JOIN pg_attribute AS ca ON cols.colid = ca.attnum WHERE ca.attrelid = c.conrelid ORDER BY cols.arridx ), c.contype, (SELECT fkc.relname || '.' || fka.attname FROM pg_attribute AS fka JOIN pg_class AS fkc ON fka.attrelid = fkc.oid WHERE fka.attrelid = c.confrelid AND fka.attnum = c.confkey[1]), cl.reloptions FROM pg_constraint AS c JOIN pg_class AS cl ON c.conrelid = cl.oid WHERE cl.relname = %s AND pg_catalog.pg_table_is_visible(cl.oid) Úp)r7ÚuÚfÚ.r&NÚcF)ÚcolumnsÚ primary_keyÚuniqueÚ foreign_keyÚcheckÚindexÚ definitionÚoptionsaa SELECT indexname, array_agg(attname ORDER BY arridx), indisunique, indisprimary, array_agg(ordering ORDER BY arridx), amname, exprdef, s2.attoptions FROM ( SELECT c2.relname as indexname, idx.*, attr.attname, am.amname, CASE WHEN idx.indexprs IS NOT NULL THEN pg_get_indexdef(idx.indexrelid) END AS exprdef, CASE am.amname WHEN %s THEN CASE (option & 1) WHEN 1 THEN 'DESC' ELSE 'ASC' END END as ordering, c2.reloptions as attoptions FROM ( SELECT * FROM pg_index i, unnest(i.indkey, i.indoption) WITH ORDINALITY koi(key, option, arridx) ) idx LEFT JOIN pg_class c ON idx.indrelid = c.oid LEFT JOIN pg_class c2 ON idx.indexrelid = c2.oid LEFT JOIN pg_am am ON c2.relam = am.oid LEFT JOIN pg_attribute attr ON attr.attrelid = c.oid AND attr.attnum = idx.key WHERE c.relname = %s AND pg_catalog.pg_table_is_visible(c.oid) ) s2 GROUP BY indexname, indisunique, indisprimary, amname, exprdef, attoptions; Z_btreeT) r<Úordersr=r>r?r@rAÚtyperBrC)r"r#ÚtupleÚsplitÚindex_default_access_methodÚendswithrÚsuffix)rr$r/Ú constraintsÚ constraintr<ÚkindZ used_colsrCrAr>ZprimaryrDÚtype_rBZ basic_indexrrrÚget_constraints°s^ëø (×4÷  üú  ö z%DatabaseIntrospection.get_constraints)r) Ú__name__Ú __module__Ú __qualname__Zdata_types_reverserHrrr%r0r4r6rOÚ __classcell__rrrrr s:ì . rN) Ú collectionsrZ%django.db.backends.base.introspectionrrZ BaseFieldInforZdjango.db.modelsrÚ_fieldsrrrrrÚs