3 (hr<@szddlmZddlZddlmZddlmZmZm Z ddl m Z ddl m Z edejdZed d ZGd ddeZdS)) namedtupleN) FieldType)BaseDatabaseIntrospection FieldInfo TableInfo)Index) OrderedSetrextra is_unsignedhas_json_constraintInfoLinezNcol_name data_type max_len num_prec num_scale extra column_default is_unsignedc(seZdZejdejdejdejdejdej dej dej dej dej dejdejd ejd ejd ejdejdejdejdejdejd iZfd d ZddZddZffddZddZddZddZddZ ddZ!ddZ"d d!Z#Z$S)"DatabaseIntrospectionZ TextFieldZ DecimalFieldZ DateFieldZ DateTimeFieldZ FloatField IntegerField JSONFieldBigIntegerFieldSmallIntegerFieldZ CharFieldZ TimeFieldcsttj||}d|jkr<|dkr$dS|dkr0dS|dkrReturn a list of table and view names in the current database.zSHOW FULL TABLEScSs*g|]"}t|ddddj|dqS)rtv)z BASE TABLEZVIEW)rget).0rowrrr ^sz8DatabaseIntrospection.get_table_list..)executefetchall)rcursorrrrget_table_list[s z$DatabaseIntrospection.get_table_listc Cs i}|jjr6|jjjr6|jd|gdd|jD}|jd|gdd|jD}|jd|jjj|dd }g}x|jD]}||d }|d d \} } } |j t | | | ||j p|d ||j p|d ||j p|d|d|j|j|j|d |k qW|S)zj Return a description of the table with the DB-API cursor.description interface." aZ SELECT c.constraint_name AS column_name FROM information_schema.check_constraints AS c WHERE c.table_name = %s AND LOWER(c.check_clause) = 'json_valid(`' + LOWER(c.constraint_name) + '`)' AND c.constraint_schema = DATABASE() cSsh|] }|dqS)rr)rrrrr rsz>DatabaseIntrospection.get_table_description..a SELECT column_name, data_type, character_maximum_length, numeric_precision, numeric_scale, extra, column_default, CASE WHEN column_type LIKE '%% unsigned' THEN 1 ELSE 0 END AS is_unsigned FROM information_schema.columns WHERE table_name = %s AND table_schema = DATABASE()cSsi|]}t||dqS)r)r )rlinerrr sz?DatabaseIntrospection.get_table_description..zSELECT * FROM %s LIMIT 1cSs|dk rt|S|S)N)int)irrrto_intsz;DatabaseIntrospection.get_table_description..to_intrN) connectionmysql_is_mariadbfeaturesZcan_introspect_json_fieldr!r"ops quote_namerappendrmax_lenZnum_precZ num_scaleZcolumn_defaultr r ) rr# table_nameZjson_constraints field_infor*fieldsr&infonameZ type_codeZ display_sizerrrget_table_descriptionas6    z+DatabaseIntrospection.get_table_descriptioncCs2x,|j||D]}d|jkr||jdgSqWgS)Nr)tablecolumn)r;r r:)rr#r6Z table_fieldsr7rrr get_sequencess z#DatabaseIntrospection.get_sequencescCs4|j||}i}x|D]\}}}||f||<qW|S)z Return a dictionary of {field_name: (field_name_other_table, other_table)} representing all relationships to the given table. )get_key_columns)rr#r6 constraintsZ relationsZ my_fieldnameZ other_tableZ other_fieldrrr get_relationss  z#DatabaseIntrospection.get_relationscCs$g}|jd|g|j|j|S)z Return a list of (column_name, referenced_table_name, referenced_column_name) for all key columns in the given table. a@ SELECT column_name, referenced_table_name, referenced_column_name FROM information_schema.key_column_usage WHERE table_name = %s AND table_schema = DATABASE() AND referenced_table_name IS NOT NULL AND referenced_column_name IS NOT NULL)r!extendr")rr#r6Z key_columnsrrrr?s  z%DatabaseIntrospection.get_key_columnscCs|jdj|jjj|t|j}t}x&|D]}|ddkr2|j|dq2Wi}xl|D]d}|d|krpq^|d|krddd||d<|ddkrd ||dd <|ds^d ||dd <q^W|S) NzSHOW INDEX FROM {0}r+rr,F) primary_keyuniqueZPRIMARYTrDrE) r!formatr/r2r3listr"setadd)rr#r6ZrowsZmulticol_indexesrindexesrrr get_indexess$       z!DatabaseIntrospection.get_indexescCs2x,|j||jD]}|ddr|dSqWdS)zP Returns the name of the primary key column for the given table rrDrN)rKitems)rr#r6r=rrrget_primary_key_columns  z,DatabaseIntrospection.get_primary_key_columncCs,|jd|g|j}|s$|jjjS|dS)z Retrieve the storage engine for a given table. Return the default storage engine if the table doesn't exist. zBSELECT engine FROM information_schema.tables WHERE table_name = %sr)r!Zfetchoner/r1Z_mysql_storage_engine)rr#r6resultrrrget_storage_engines z(DatabaseIntrospection.get_storage_enginecCst}tj|d}dd|jD}xV|D]N}|jtjjkr,|jjj |j |j kr,|j dd|kr,|j |j ddq,W|S)Nrcss|]}|js|VqdS)N)Z is_whitespace)rtokenrrr szBDatabaseIntrospection._parse_constraint_columns..rrR) rsqlparseparseflattenZttypetokensNamer/r2r3valuerI)r check_clausecolumnsZ check_columnsZ statementrVrPrrr_parse_constraint_columnss z/DatabaseIntrospection._parse_constraint_columnscCs>i}d}|j||gxV|jD]J\}}}}||krXtdddd|rL||fndd||<||dj|q Wd} |j| |gxR|jD]F\}} | jdkrd||d <d||d <q| jd krd||d <qW|jjjrvd } d d |j||D} |jj r d} nd} |j| |gxX|jD]L\}} |j | | }t ||hkrZ| d7} d| }|dddddd||<q&W|jd|jj j |xdd|jDD]p\}}}}}}||krtddddd||<d||d<|dkrtjn|j||d<||dj|qWx"|jD]}t|d|d<q W|S)zu Retrieve any constraints or keys (unique, pk, fk, check, index) across one or more columns. aX SELECT kc.`constraint_name`, kc.`column_name`, kc.`referenced_table_name`, kc.`referenced_column_name` FROM information_schema.key_column_usage AS kc WHERE kc.table_schema = DATABASE() AND kc.table_name = %s ORDER BY kc.`ordinal_position` FN)rZrDrEindexcheck foreign_keyrZz SELECT c.constraint_name, c.constraint_type FROM information_schema.table_constraints AS c WHERE c.table_schema = DATABASE() AND c.table_name = %s z primary keyTrDrErcSsh|] }|jqSr)r:)rr9rrrr%)sz8DatabaseIntrospection.get_constraints..a SELECT c.constraint_name, c.check_clause FROM information_schema.check_constraints AS c WHERE c.constraint_schema = DATABASE() AND c.table_name = %s a3 SELECT cc.constraint_name, cc.check_clause FROM information_schema.check_constraints AS cc, information_schema.table_constraints AS tc WHERE cc.constraint_schema = DATABASE() AND tc.table_schema = cc.constraint_schema AND cc.constraint_name = tc.constraint_name AND tc.constraint_type = 'CHECK' AND tc.table_name = %s rz__unnamed_constraint_%s__zSHOW INDEX FROM %scSs"g|]}|dd|dfqS)Nr- r)rxrrrr Rsz9DatabaseIntrospection.get_constraints..)rZrDrEr]r^r\ZBTREEtype)r!r"rrIlowerr/r1Z can_introspect_check_constraintsr;r0r[rHr2r3rsuffixvaluesrG)rr#r6r@Z name_query constraintr=Z ref_tableZ ref_columnZ type_querykindZunnamed_constraints_indexrZrYZconstraint_columnsr<Z non_uniquer\Zcolseqtype_rrrget_constraintssh        $    z%DatabaseIntrospection.get_constraints)%__name__ __module__ __qualname__rZBLOBZDECIMALZ NEWDECIMALZDATEZDATETIMEZDOUBLEFLOATZINT24ZJSONLONGZLONGLONGZSHORTSTRINGZTIMEZ TIMESTAMPZTINYZ TINY_BLOBZ MEDIUM_BLOBZ LONG_BLOBZ VAR_STRINGZdata_types_reverserr$r;r>rAr?rKrMrOr[rh __classcell__rr)rrr -s>  ;    r )r r r ) collectionsrrSZmysql.connector.constantsrZ%django.db.backends.base.introspectionrrZ BaseFieldInforZdjango.db.modelsrZdjango.utils.datastructuresr_fieldsr r rrrrs