o h<@sddlZddlmZddlZejdkrddlmZmZmZnddl mZejdkrEejdkr4ddl mZddl m Z ejd krEdd l m Z dd lmZed Zejdkr]ed ejdZGdddeZdS)N) namedtuple)BaseDatabaseIntrospection FieldInfo TableInfo)rr)r) force_text)r) OrderedSet) FieldTypezQ\sCONSTRAINT `[^`]*` FOREIGN KEY \(`([^`]*)`\) REFERENCES `([^`]*)` \(`([^`]*)`\)r)extracs6eZdZiejdejdejdejdejdej dej dej dej dej dejejdkr6dnd ejd ejd ejdejdejdejdejdejd iZfd d ZddZejdkrmddZnejdkrwddZnddZddZddZddZddZ ddZ!dd Z"d!d"Z#Z$S)#DatabaseIntrospection TextField DecimalField DateField DateTimeField FloatField IntegerFieldBigIntegerFieldrSmallIntegerField CharField TimeFieldcs,tt|||}|dkrd|jvrdS|S)Nrauto_increment AutoField)superrget_field_typer)self data_type description field_type __class__V/var/www/html/env/lib/python3.10/site-packages/mysql/connector/django/introspection.pyr8s  z$DatabaseIntrospection.get_field_typecCs8|dtjdkrdd|DSdd|DS)z6Returns a list of table names in the current database.zSHOW FULL TABLESrcSs*g|]}t|dddd|dqS)rtv)z BASE TABLEVIEWr)rget.0rowr%r%r& Dsz8DatabaseIntrospection.get_table_list..cSsg|]}|dqS)rr%r+r%r%r&r.I)executedjangoVERSIONfetchall)rcursorr%r%r&get_table_list@s  z$DatabaseIntrospection.get_table_list)r c stdd|d|gtfdd|D}|d|jj|dd}g}|jD]A}t|d }| t |f|d d |||j pK|d |||j pU|d |||j p_|d |d||j||jfq.|S)x Returns a description of the table, with the DB-API cursor.description interface." InfoLinezBcol_name data_type max_len num_prec num_scale extra column_defaulta SELECT column_name, data_type, character_maximum_length, numeric_precision, numeric_scale, extra, column_default FROM information_schema.columns WHERE table_name = %s AND table_schema = DATABASE()c3 |] }|d|fVqdSrNr%r,liner8r%r& ` >DatabaseIntrospection.get_table_description..SELECT * FROM %s LIMIT 1cS|durt|S|SNintir%r%r&fr/=DatabaseIntrospection.get_table_description..rrr )rr0dictr3 connectionops quote_namer!r appendrmax_lennum_prec num_scalecolumn_defaultrrr4 table_name field_infoto_intfieldsr<col_namer%r=r&get_table_descriptionLs<      z+DatabaseIntrospection.get_table_descriptionc stdd|d|gtfdd|D}|d|jj|dd}g}|jD]A}t|d }| t |f|d d |||j pK|d |||j pU|d |||j p_|d f|df||jfq.|S)r7r8z3col_name data_type max_len num_prec num_scale extraz SELECT column_name, data_type, character_maximum_length, numeric_precision, numeric_scale, extra FROM information_schema.columns WHERE table_name = %s AND table_schema = DATABASE()c3r9r:r%r;r=r%r&r>r?r@rAcSrBrCrDrFr%r%r&rHr/rIrrrJrKrLr )rr0rMr3rNrOrPr!r rQrrRrSrTrrVr%r=r&r\zsF        cs|d|gt||d|gtdd|D|d|jj|tjdkr<fdd|j DSfd d|j DS) zw Returns a description of the table, with the DB-API cursor.description interface. zSELECT column_name, character_maximum_length FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = %s AND table_schema = DATABASE() AND character_maximum_length IS NOT NULLzSELECT column_name, numeric_precision, numeric_scale FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = %s AND table_schema = DATABASE() AND data_type='decimal'css2|]}|dtdd|ddDfVqdS)rcSsg|]}t|qSr%rD)r,nr%r%r&r.r/zIDatabaseIntrospection.get_table_description...rN)tupler;r%r%r&r>s*r@zSELECT * FROM {0} LIMIT 1r c sbg|]-}tt|df|dd|d|df|d|dd|dfqS)rrrJrKr )rr r*r; length_map numeric_mapr%r&r.s  z?DatabaseIntrospection.get_table_description..cs:g|]}|dd|d|df|ddqS)NrJrrK)r*r;)r`r%r&r.s,) r0rMr3formatrNrOrPr1r2r!rr4rWr%r_r&r\s,     cCstddt|||DS)zu Returns a dictionary of {field_name: field_index} for the given table. Indexes are 0-based. css |] \}}|d|fVqdSr:r%)r,rGdr%r%r&r>sz7DatabaseIntrospection._name_to_index..)rM enumerater\rcr%r%r&_name_to_indexs   z$DatabaseIntrospection._name_to_indexc Cs~|||}i}tjdkr|D] \}}}||f||<q|S|||}|D]\}}}||||} ||} | |f|| <q%|S)z Returns a dictionary of {field_index: (field_index_other_table, other_table)} representing all relationships to the given table. Indexes are 0-based. r)get_key_columnsr1r2rf) rr4rW constraints relations my_fieldname other_table other_field my_field_dictother_field_indexmy_field_indexr%r%r& get_relationss   z#DatabaseIntrospection.get_relationscCs$g}|d|g|||S)z Returns a list of (column_name, referenced_table_name, referenced_column_name) for all key columns in given table. zSELECT 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)r0extendr3)rr4rW key_columnsr%r%r&rgsz%DatabaseIntrospection.get_key_columnscCs|d|jj|t|}t}|D]}|ddkr'||dqi}|D]2}|d|vr5q,|d|vrDddd||d<|ddkrRd ||dd <|ds^d ||dd <q,|S) NzSHOW INDEX FROM {0}rJrrKF) primary_keyuniquePRIMARYTrtru) r0rbrNrOrPlistr3setadd)rr4rWrowsmulticol_indexesr-indexesr%r%r& get_indexess*     z!DatabaseIntrospection.get_indexescCs2|||D]}|ddr|dSqdS)zP Returns the name of the primary key column for the given table rrtrN)r}items)rr4rWcolumnr%r%r&get_primary_key_columns   z,DatabaseIntrospection.get_primary_key_columncCs,|d|g|}|s|jjjS|dS)z Retrieves the storage engine for a given table. Returns the default storage engine if the table doesn't exist. zBSELECT engine FROM information_schema.tables WHERE table_name = %sr)r0fetchonerNfeaturesmysql_storage_engine)rr4rWresultr%r%r&get_storage_engine s z(DatabaseIntrospection.get_storage_enginecCszi}d}|||jjd|g|D]%\}}}}||vr0tdddd|r*||fndd||<||d|qd} || |jjd|g|D]#\}} | dkrcd ||d <d ||d <qL| d krod ||d <qL|d |jj|d d|DD]&\} } } }}| |vrtddd ddd|| <d || d<|| d|q| D] }t |d|d<q|S)zv Retrieves any constraints or keys (unique, pk, fk, check, index) across one or more columns. zSELECT 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 = %s AND kc.table_name = %sNAMEFN)columnsrtruindexcheck foreign_keyrz SELECT c.constraint_name, c.constraint_type FROM information_schema.table_constraints AS c WHERE c.table_schema = %s AND c.table_name = %s z primary keyTrtruzSHOW INDEX FROM %scSsg|]}|ddqS)NrLr%)r,xr%r%r&r._sz9DatabaseIntrospection.get_constraints..r) r0rN settings_dictr3r rylowerrOrPvaluesrw)rr4rWrh name_query constraintr ref_table ref_column type_querykindtable non_uniquercolseqr%r%r&get_constraints.s\        z%DatabaseIntrospection.get_constraints)%__name__ __module__ __qualname__rBLOBDECIMAL NEWDECIMALDATEDATETIMEDOUBLEFLOATINT24LONGLONGLONGSHORTr1r2STRINGTIME TIMESTAMPTINY TINY_BLOB MEDIUM_BLOB LONG_BLOB VAR_STRINGdata_types_reverserr5r\rfrprgr}rrr __classcell__r%r%r#r&rsl      - -) r)re collectionsrr1r2%django.db.backends.base.introspectionrrrdjango.db.backendsdjango.utils.encodingr django.utils.datastructuresr mysql.connector.constantsrcompileforeign_key_re_fieldsrr%r%r%r&s"