관리-도구
편집 파일: introspection.cpython-38.pyc
U )�eg4 � @ s� d dl Z d dlmZ d dlZejdkr<d dlmZmZmZ n d dl mZ ejdkr�ejdk rhd dl mZ d dl m Z ejd kr�d d l m Z d d lmZ e �d �Zejdkr�ed ejd �ZG dd� de�ZdS )� N)� namedtuple�� � )�BaseDatabaseIntrospection� FieldInfo� TableInfo)r �r � )r )� force_text)r � )� OrderedSet)� FieldTypezQ\sCONSTRAINT `[^`]*` FOREIGN KEY \(`([^`]*)`\) REFERENCES `([^`]*)` \(`([^`]*)`\)r )�extrac & s� e Zd Zejdejdejdejdejdej dej dej dej dej dejejdk rVdnd ejd ejd ejdejdejdejdejdejd iZ� fd d �Zdd� Zejdkr�dd� Z ndd� Z dd� Z dd� Z dd� Z dd� Z dd � Z!d d � Z"d d � Z#� Z$S )!�DatabaseIntrospectionZ TextFieldZ DecimalFieldZ DateFieldZ DateTimeFieldZ FloatField� IntegerFieldZBigIntegerFieldr ZSmallIntegerFieldZ CharFieldZ TimeFieldc s, t t| ��||�}|dkr(d|jkr(dS |S )Nr Zauto_incrementZ AutoField)�superr �get_field_typer )�selfZ data_type� descriptionZ field_type�� __class__� �y/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysql/connector/django/introspection.pyr 8 s ��z$DatabaseIntrospection.get_field_typec C s< |� d� tjdkr&dd� |�� D �S dd� |�� D �S dS )z6Returns a list of table names in the current database.zSHOW FULL TABLESr c S s* g | ]"}t |d ddd��|d ���qS )r �t�v)z BASE TABLEZVIEWr )r �get��.0�rowr r r � <listcomp>D s �z8DatabaseIntrospection.get_table_list.<locals>.<listcomp>c S s g | ] }|d �qS )r r r r r r r I s N)�execute�django�VERSION�fetchall)r �cursorr r r �get_table_list@ s �z$DatabaseIntrospection.get_table_listc s� t dd�� |�d|g� t� fdd�|�� D ��}|�d| jj�|� � dd� }g }|jD ]�}t|d �}|� t |f|d d � ||| j �p�|d ||| j �p�|d ||| j �p�|d f |d f || jf � � q\|S )zx Returns a description of the table, with the DB-API cursor.description interface." �InfoLinez3col_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()c 3 s | ]}|d � |� fV qdS �r Nr �r �line�r' r r � <genexpr>a s �>DatabaseIntrospection.get_table_description.<locals>.<genexpr>zSELECT * FROM %s LIMIT 1c S s | d k rt | �S | S )N��int)�ir r r �<lambda>g � z=DatabaseIntrospection.get_table_description.<locals>.<lambda>r r � � � r )r r! �dictr$ � connection�ops� quote_namer r �appendr �max_lenZnum_precZ num_scaler )r r% � table_nameZ field_infoZto_int�fieldsr* Zcol_namer r+ r �get_table_descriptionL sF �� � � ���� �� z+DatabaseIntrospection.get_table_descriptionc s� |� d|g� t|�� �� |� d|g� tdd� |�� D ���|� d�| jj�|��� tjdkrx� �fdd�|j D �S � fd d�|j D �S d S ) zw Returns a description of the table, with the DB-API cursor.description interface. z�SELECT column_name, character_maximum_length FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = %s AND table_schema = DATABASE() AND character_maximum_length IS NOT NULLz�SELECT column_name, numeric_precision, numeric_scale FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = %s AND table_schema = DATABASE() AND data_type='decimal'c s s0 | ](}|d t dd� |dd� D ��fV qdS )r c S s g | ] }t |��qS r r. )r �nr r r r � s zIDatabaseIntrospection.get_table_description.<locals>.<genexpr>.<listcomp>r N)�tupler) r r r r, � s �r- zSELECT * FROM {0} LIMIT 1r c sb g | ]Z}t t|d �f|dd� � �|d |d �f ��|d |dd� � |d f � �qS )r r r3 r4 r )r r r r) �� length_mapZ numeric_mapr r r � s � ����z?DatabaseIntrospection.get_table_description.<locals>.<listcomp>c s: g | ]2}|d d� � � |d |d �f |dd � �qS )Nr3 r r4 )r r) )rB r r r � s �N) r! r6 r$ �formatr7 r8 r9 r" r# r �r r% r< r rA r r> y s, � �� � � �c C s t dd� t| �||��D ��S )zu Returns a dictionary of {field_name: field_index} for the given table. Indexes are 0-based. c s s | ]\}}|d |fV qdS r( r )r r0 �dr r r r, � s z7DatabaseIntrospection._name_to_index.<locals>.<genexpr>)r6 � enumerater> rD r r r �_name_to_index� s �z$DatabaseIntrospection._name_to_indexc C s� | � ||�}i }tjdkr:|D ]\}}}||f||< q |S | �||�}|D ].\}}}| �||�| } || } | |f|| < qJ|S dS )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 N)�get_key_columnsr" r# rG ) r r% r< � constraintsZ relationsZ my_fieldnameZ other_tableZ other_fieldZ my_field_dictZother_field_indexZmy_field_indexr r r � get_relations� s" ��z#DatabaseIntrospection.get_relationsc C s$ g }|� d|g� |�|�� � |S )z� Returns a list of (column_name, referenced_table_name, referenced_column_name) for all key columns in given table. z�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$ )r r% r< Z key_columnsr r r rH � s �z%DatabaseIntrospection.get_key_columnsc C s� |� d�| jj�|��� t|�� �}t� }|D ] }|d dkr0|�|d � q0i }|D ]d}|d |krjqX|d |kr�ddd�||d < |d dkr�d ||d d < |d sXd ||d d < qX|S ) NzSHOW INDEX FROM {0}r3 r � r4 F)� primary_key�uniqueZPRIMARYTrM rN ) r! rC r7 r8 r9 �listr$ �set�add)r r% r< �rowsZmulticol_indexesr Zindexesr r r � get_indexes� s& � z!DatabaseIntrospection.get_indexesc C s2 | � ||��� D ] }|d d r|d S qdS )zP Returns the name of the primary key column for the given table r rM r N)rS �items)r r% r< �columnr r r �get_primary_key_column� s z,DatabaseIntrospection.get_primary_key_columnc C s, |� d|g� |�� }|s$| jjjS |d S )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 )r! Zfetchoner7 �featuresZmysql_storage_engine)r r% r< �resultr r r �get_storage_engine� s � z(DatabaseIntrospection.get_storage_enginec C s� i }d}|� || jjd |g� |�� D ]L\}}}}||krbt� dddd|rT||fndfd�||< || d �|� q(d} |� | | jjd |g� |�� D ]F\}} | �� dkr�d || d < d || d < q�| �� d kr�d || d < q�|� d | jj�|� � d d� |�� D �D ]P\} } } }}| |k�r<t� ddd ddd�|| < d || d< || d �|� �q |� � D ]}t |d �|d< �qf|S )zv Retrieves any constraints or keys (unique, pk, fk, check, index) across one or more columns. z�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 = %s AND kc.table_name = %s�NAMEFN)�columnsrM rN �index�checkZ foreign_keyr[ z� 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 keyTrM rN zSHOW INDEX FROM %sc S s g | ]}|d d� �qS )Nr5 r )r �xr r r r 2 s z9DatabaseIntrospection.get_constraints.<locals>.<listcomp>r\ ) r! r7 Z settings_dictr$ r rQ �lowerr8 r9 �valuesrO )r r% r< rI Z name_query� constraintrU Z ref_tableZ ref_columnZ type_query�kind�tableZ non_uniquer\ Zcolseqr r r �get_constraints s\ ���� � �� � z%DatabaseIntrospection.get_constraints)%�__name__� __module__� __qualname__r ZBLOBZDECIMALZ NEWDECIMALZDATEZDATETIME�DOUBLE�FLOATZINT24�LONGZLONGLONGZSHORTr" r# �STRINGZTIME� TIMESTAMPZTINYZ TINY_BLOBZ MEDIUM_BLOBZ LONG_BLOBZ VAR_STRINGZdata_types_reverser r& r> rG rJ rH rS rV rY rd � __classcell__r r r r r sf � -) r )�re� collectionsr r" r# Z%django.db.backends.base.introspectionr r r Zdjango.db.backendsZdjango.utils.encodingr Zdjango.utils.datastructuresr Zmysql.connector.constantsr �compileZforeign_key_re�_fieldsr r r r r �<module> s