관리-도구
편집 파일: reflection.cpython-38.pyc
U -?�f2Y � @ s� d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd lmZ dd lm Z G d d � d �Z e j G dd� d��ZdZd'dd�Zdd� Zdd� Zeed�dd�Zddd d d d d d!d"d#� Ze �d$�d%d&� eD ���ZdS )(� N� )�ENUM)�SET)�DATETIME)�TIME)� TIMESTAMP� )�log)�types)�utilc @ s e Zd ZdZdd� ZdS )�ReflectedStatez;Stores raw information about a SHOW CREATE TABLE statement.c C s( g | _ i | _d | _g | _g | _g | _d S �N)�columns� table_options� table_name�keys�fk_constraints�ck_constraints)�self� r �T/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/mysql/reflection.py�__init__ s zReflectedState.__init__N)�__name__� __module__� __qualname__�__doc__r r r r r r s r c @ s� e Zd ZdZdd� Zdd� Zeed�dd�Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� ZdZdd� Zd d � Zd d � Zd d!� Zd"S )#�MySQLTableDefinitionParserz4Parses the results of a SHOW CREATE TABLE statement.c C s || _ || _| �� d S r )�dialect�preparer� _prep_regexes)r r r r r r r & s z#MySQLTableDefinitionParser.__init__c C s t � }||_t�d|�D ]�}|�d| jj �r<| �||� q|�d�rT| �||� q|dkr^q|�d�rv| � ||� qd|kr�| � ||� q|s�q| � |�\}}|d kr�t � d| � q|dkr�|j�|� q|d kr�|j�|� q|d kr|j�|� qq|S ) Nz\r?\nz z) �)zCREATE � PARTITIONzUnknown schema content: %r�key� fk_constraint� ck_constraint)r �charset�re�split� startswithr � initial_quote� _parse_column�_parse_table_options�_parse_table_name�_parse_partition_options�_parse_constraintsr �warnr �appendr r )r Z show_creater% �state�line�type_�specr r r �parse+ s4 z MySQLTableDefinitionParser.parse)�sql�returnc C s t | j�|��S r )�bool� _re_is_view�match)r r6 r r r � _check_viewM s z&MySQLTableDefinitionParser._check_viewc C s: | j �|�}|r�|�� }| �|d �|d< |d rb| j�|d �}|rb|�� d rb|�� d |d< |d r�| j�|d �d |d< d|fS | j�|�}|r�|�� }| j�|d �|d< dd� | �|d �D �|d < d d� | �|d �D �|d < d |fS | j�|�}|�r|�� }d |fS | j �|�}|�r2d|fS d|fS )zaParse a KEY or CONSTRAINT line. :param line: A line of SHOW CREATE TABLE output r Z version_sql�parserr r" �tablec S s g | ] }|d �qS �r r ��.0�cr r r � <listcomp>m s zAMySQLTableDefinitionParser._parse_constraints.<locals>.<listcomp>�localc S s g | ] }|d �qS r> r r? r r r rB n s Zforeignr# r$ � partitionN) �_re_keyr: � groupdict�_parse_keyexprs�_re_key_version_sqlr Zunformat_identifiers�_re_fk_constraint�_re_ck_constraint� _re_partition)r r2 �mr4 �m2r r r r. P s@ �� � z-MySQLTableDefinitionParser._parse_constraintsc C s, | j \}}|�|�}|r(||�d��|_dS )zZExtract the table name. :param line: The first line of SHOW CREATE TABLE �nameN)�_pr_namer: �groupr )r r2 r1 �regex�cleanuprL r r r r, � s z,MySQLTableDefinitionParser._parse_table_namec C s� i }|rn|dkrn|}| j D ]R\}}|�|�}|s2q|�d�|�d� }} |rT|| �} | ||�� < |�d|�}qdD ]} |�| d� qr|�� D ] \} } | |jd| jj | f < q�dS )z�Build a dictionary of all reflected table-level options. :param line: The final line of SHOW CREATE TABLE output. r � directive�val� )�auto_incrementzdata directoryzindex directoryN�%s_%s) � _pr_options�searchrP �lower�sub�pop�itemsr r rN ) r r2 r1 �optionsZ rest_of_linerQ rR rL rS �valueZnope�optrT r r r r+ � s z/MySQLTableDefinitionParser._parse_table_optionsc C s� i }|d d � }|� d�s$|� d�r2|dd � }q| jD �]\}}|�|�}|r8d|jkr\q8|�d�}|�� }|dk} |dks�| �r |�dd �}|�d d �}| r�|�d �r�|d d � }| jj d kr�|�d �r�d|ks�d|ks�d|kr�|d d � }d| jj |f } ||| < n,|�dd�}|�d�} |�r.|| �} | ||< �q>q8|� � D ]~\} } d| jj }d| jj }| |k�sz| |k�r�| |j k�r�| |j | < nd|j | | f |j | < n| |j d| jj | f < �qFd S )N�(� r r! rS Z subpartitionrD z) */rU �,r ���ZmariadbZMAXVALUEZMINVALUE�ENGINEz%s_%s_definitions�_rT z%s_partition_definitionsz%s_subpartition_definitionsz%s, %srW ) r( rX rY �patternrP rZ �replace�endswithr rN r] r )r r2 r1 r^ �new_linerQ rR rL rS Zis_subpartitionZdefsr_ r` rT Zpart_defZ subpart_defr r r r- � sV ��� �z3MySQLTableDefinitionParser._parse_partition_optionsc C s� d}| j �|�}|r&|�� }d|d< n | j�|�}|rF|�� }d|d< |s\t�d| � dS |d srt�d| � |d |d |d }}}z| jj| }W n, tk r� t�d ||f � t j }Y nX |dks�|d kr�g } n@|d d k�r |d d k�r | j � |�} ndd� | j � |�D �} i } t|tttf��rL| �rL| �d �| d< dD ] } |�| d��rPd| | < �qPdD ] } |�| d��rr|| | | < �qrt|ttf��r�t| �} t|t��r�d | k�r�d| d< || | �} i } d| d< |�dd�dk�r�d| d< |�dd�dk�rd| d< |�dd��r,d| d< nt|t j��rBd| d< |�dd�}|d k�r\d}|�d d�}|dk �rzt|�}|�d �}|dk �r�t|d �}|�d �}|dk �r�|d!k|d"< || d#< t|| ||d$�}|�| � |j� |� dS )%z�Extract column details. Falls back to a 'minimal support' variant if full parse fails. :param line: Any column-bearing line from SHOW CREATE TABLE NT�fullFz Unknown column definition %rz-Incomplete reflection of column definition %rrN Zcoltype�argz*Did not recognize type '%s' of column '%s'rU r �'rd c S s g | ] }t |��qS r ��int)r@ �vr r r rB s z<MySQLTableDefinitionParser._parse_column.<locals>.<listcomp>Zfsp)ZunsignedZzerofill)r% ZcollateZretrieve_as_bitwise�nullableZnotnull�NOT NULLZnotnull_generatedZautoincrZ autoincrement�default�NULL�comment� generated)�sqltextZ persistenceZSTORED� persisted�computed)rN �typers ru ) � _re_columnr: rF �_re_column_looser r/ r Z ischema_names�KeyError�sqltypesZNullType� _re_csv_str�findall� _re_csv_int� issubclassr r r r\ �getr r � _strip_values�Integer� cleanup_text�dict�updater r0 )r r2 r1 r4 rL rN r3 �args�col_typeZ type_argsZtype_kw�kwZ type_instanceZcol_kwrs ru rw ry rx Zcol_dr r r r* � s� � � z(MySQLTableDefinitionParser._parse_columnc s g }|D ]� � fdd�dD �\}}}}}dg} | � | j�|�� | � |� |sX| � d� |r�d|krfnh|�d�r�|�d�r�| � d � | � |� n>|d kr�| � d � | � |� n | � d � | � d |�d d � � |r�| � |� |� d�| �� qd�d| j�|� d�|�dg�S )a� Re-format DESCRIBE output as a SHOW CREATE TABLE string. DESCRIBE is a much simpler reflection and is sufficient for reflecting views for runtime use. This method formats DDL for columns only- keys are omitted. :param columns: A sequence of DESCRIBE or SHOW COLUMNS 6-tuples. SHOW FULL COLUMNS FROM rows must be rearranged for use with this function. c 3 s | ]}� | V qd S r r )r@ �i��rowr r � <genexpr>W s zAMySQLTableDefinitionParser._describe_to_create.<locals>.<genexpr>)r r � � � rb rr rV � timestamp�C�DEFAULTrt z'%s'rm �''rU zCREATE TABLE %s ( z, z ) )r0 r Zquote_identifierr( rh �join) r r r �bufferrN r� rq rs �extrar2 r r� r �_describe_to_createI sD � � ���z.MySQLTableDefinitionParser._describe_to_createc C s dd� | j �|�D �S )z8Unpack '"col"(2),"col" ASC'-ish strings into components.c S s( g | ] \}}}||r t |�nd |f�qS r rn )r@ Zcolname�length� modifiersr r r rB � s �z>MySQLTableDefinitionParser._parse_keyexprs.<locals>.<listcomp>)� _re_keyexprsr� )r Z identifiersr r r rG ~ s ��z*MySQLTableDefinitionParser._parse_keyexprsc C sP g | _ g | _| jj}ttddd� | jj|| j�|�fD ���}td| | jj �| _ t d�| _ t d| �| _ t d�| _t d�| _t d | �| _t d | �| _t d | �| _t d �| _|�� }d |d< t d| �| _t d| �| _t d�| _tD ]}| �|� q�dD ]}| �|� �q dD ]}| �|� �q| � dd� | � dd� | � dd� dS )z Pre-compile regular expressions.)ZiqZfqZesc_fqc S s g | ]}t �|��qS r �r& �escape)r@ �sr r r rB � s �z<MySQLTableDefinitionParser._prep_regexes.<locals>.<listcomp>zM^CREATE (?:\w+ +)?TABLE +%(iq)s(?P<name>(?:%(esc_fq)s|[^%(fq)s])+)%(fq)s +\($z ^CREATE(?! TABLE)(\s.*)?\sVIEWzW(?:(?:%(iq)s((?:%(esc_fq)s|[^%(fq)s])+)%(fq)s)(?:\((\d+)\))?(?: +(ASC|DESC))?(?=\,|$))+z \x27(?:\x27\x27|[^\x27])*\x27z\d+a� %(iq)s(?P<name>(?:%(esc_fq)s|[^%(fq)s])+)%(fq)s +(?P<coltype>\w+)(?:\((?P<arg>(?:\d+|\d+,\d+|(?:'(?:''|[^'])*',?)+))\))?(?: +(?P<unsigned>UNSIGNED))?(?: +(?P<zerofill>ZEROFILL))?(?: +CHARACTER SET +(?P<charset>[\w_]+))?(?: +COLLATE +(?P<collate>[\w_]+))?(?: +(?P<notnull>(?:NOT )?NULL))?(?: +DEFAULT +(?P<default>(?:NULL|'(?:''|[^'])*'|[\-\w\.\(\)]+(?: +ON UPDATE [\-\w\.\(\)]+)?)))?(?: +(?:GENERATED ALWAYS)? ?AS +(?P<generated>\(.*\))? ?(?P<persistence>VIRTUAL|STORED)?(?: +(?P<notnull_generated>(?:NOT )?NULL))?)?(?: +(?P<autoincr>AUTO_INCREMENT))?(?: +COMMENT +'(?P<comment>(?:''|[^'])*)')?(?: +COLUMN_FORMAT +(?P<colfmt>\w+))?(?: +STORAGE +(?P<storage>\w+))?(?: +(?P<extra>.*))?,?$z� %(iq)s(?P<name>(?:%(esc_fq)s|[^%(fq)s])+)%(fq)s +(?P<coltype>\w+)(?:\((?P<arg>(?:\d+|\d+,\d+|\x27(?:\x27\x27|[^\x27])+\x27))\))?.*?(?P<notnull>(?:NOT )NULL)?aX (?:(?P<type>\S+) )?KEY(?: +%(iq)s(?P<name>(?:%(esc_fq)s|[^%(fq)s])+)%(fq)s)?(?: +USING +(?P<using_pre>\S+))? +\((?P<columns>.+?)\)(?: +USING +(?P<using_post>\S+))?(?: +KEY_BLOCK_SIZE *[ =]? *(?P<keyblock>\S+))?(?: +WITH PARSER +(?P<parser>\S+))?(?: +COMMENT +(?P<comment>(\x27\x27|\x27([^\x27])*?\x27)+))?(?: +/\*(?P<version_sql>.+)\*/ *)?,?$z+\!\d+ (?: *WITH PARSER +(?P<parser>\S+) *)?z/RESTRICT|CASCADE|SET NULL|NO ACTION|SET DEFAULT�onaJ CONSTRAINT +%(iq)s(?P<name>(?:%(esc_fq)s|[^%(fq)s])+)%(fq)s +FOREIGN KEY +\((?P<local>[^\)]+?)\) REFERENCES +(?P<table>%(iq)s[^%(fq)s]+%(fq)s(?:\.%(iq)s[^%(fq)s]+%(fq)s)?) +\((?P<foreign>(?:%(iq)s[^%(fq)s]+%(fq)s(?: *, *)?)+)\)(?: +(?P<match>MATCH \w+))?(?: +ON DELETE (?P<ondelete>%(on)s))?(?: +ON UPDATE (?P<onupdate>%(on)s))?z[ CONSTRAINT +%(iq)s(?P<name>(?:%(esc_fq)s|[^%(fq)s])+)%(fq)s +CHECK +\((?P<sqltext>.+)\),?z (?:.*)(?:SUB)?PARTITION(?:.*))re ZTYPEZAUTO_INCREMENTZAVG_ROW_LENGTHz CHARACTER SETzDEFAULT CHARSETZCHECKSUMZCOLLATEZDELAY_KEY_WRITEZ INSERT_METHODZMAX_ROWSZMIN_ROWSZ PACK_KEYSZ ROW_FORMATZKEY_BLOCK_SIZEZSTATS_SAMPLE_PAGES)� PARTITION BY�SUBPARTITION BY� PARTITIONS� SUBPARTITIONSr! Z SUBPARTITIONZUNIONz \([^\)]+\)Z TABLESPACEz.*? STORAGE DISKZ RAID_TYPEz4\w+\s+RAID_CHUNKS\s*\=\s*\w+RAID_CHUNKSIZE\s*=\s*\w+N) Z _re_columnsrX r Z final_quoter� �zipr) Z_escape_identifier� _pr_compileZ_unescape_identifierrO � _re_compiler9 r� r r� r{ r| rE rH �copyrI rJ rK �_options_of_type_string�_add_option_string�_add_option_word�_add_partition_option_word�_add_option_regex)r Z_final�quotesr� �optionr r r r � s� ������ �� �� �� ��� ���� �z(MySQLTableDefinitionParser._prep_regexesz(?:\s*(?:=\s*)|\s+)c C s* dt �|�| jf }| j�t|t�� d S )Nz0(?P<directive>%s)%s'(?P<val>(?:[^']|'')*?)'(?!'))r& r� �_optional_equalsrX r0 r� r� �r rS rQ r r r r� I s �z-MySQLTableDefinitionParser._add_option_stringc C s( dt �|�| jf }| j�t|�� d S )Nz (?P<directive>%s)%s(?P<val>\w+)�r& r� r� rX r0 r� r� r r r r� P s �z+MySQLTableDefinitionParser._add_option_wordc C sp |dks|dkr&dt �|�| jf }n6|dks6|dkrLdt �|�| jf }ndt �|�f }| j�t|�� d S )Nr� r� z((?<!\S)(?P<directive>%s)%s(?P<val>\w+.*)r� r� z&(?<!\S)(?P<directive>%s)%s(?P<val>\d+)z (?<!\S)(?P<directive>%s)(?!\S)r� r� r r r r� W s ��z5MySQLTableDefinitionParser._add_partition_option_wordc C s* dt �|�| j|f }| j�t|�� d S )Nz (?P<directive>%s)%s(?P<val>%s)r� r� r r r r� f s �z,MySQLTableDefinitionParser._add_option_regexN)r r r r r r5 �strr8 r; r. r, r+ r- r* r� rG r r� r� r� r� r� r r r r r " s$ "2 8j5 @r )�COMMENTzDATA DIRECTORYzINDEX DIRECTORYZPASSWORDZ CONNECTIONc C s t | �|fS )z1Prepare a 2-tuple of compiled regex and callable.)r� )rQ rR r r r r� x s r� c C s t �| t jt jB �S )z)Compile a string to regex, I and UNICODE.)r& �compile�I�UNICODE)rQ r r r r� ~ s r� c C s\ g }| D ]N}|dd� dks,|dd� dkrL|dd� � |d d |d �}|�|� q|S )z Strip reflected values quotesr r �"rm rd r� )rh r0 )�valuesZ strip_values�ar r r r� � s r� )�raw_textr7 c C s& d| krt �tdd� | �} | �dd�S )N�\c S s t | d S )Nr )�_control_char_map)r� r r r �<lambda>� � z cleanup_text.<locals>.<lambda>r� rm )r& r[ �_control_char_regexprh )r� r r r r� � s �r� r� � ��� � � � � ) z\\z\0z\az\bz\tz\nz\vz\fz\r�|c c s | ]}t �|�V qd S r r� )r@ �kr r r r� � s r� )N)r& Z enumeratedr r r r r r rU r r~ r r Z class_loggerr r� r� r� r� r� r� r� r� r� r� r r r r �<module> sB P � �