관리-도구
편집 파일: reflection.cpython-37.pyc
B ��4]�E � @ s� d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd lmZ dd lm Z G d d � d e �Z e jG dd� de ��ZdZddd�Zdd� ZdS )� N� )�_EnumeratedValues)�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 �W/opt/alt/python37/lib64/python3.7/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 @ st e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dZ dd� Zdd� Zdd� ZdS ) �MySQLTableDefinitionParserz4Parses the results of a SHOW CREATE TABLE statement.c C s || _ || _| �� d S )N)�dialect�preparer� _prep_regexes)r r r r r r r $ s z#MySQLTableDefinitionParser.__init__c C s� t � }||_x�t�d|�D ]�}|�d| jj �r>| �||� q|�d�rV| �||� q|dkr`q|�d�rx| � ||� q|s~q| � |�\}}|d kr�t � d| � q|dkr�|j �|� q|dkr�|j�|� q|d kr|j�|� qqW |S ) Nz\r?\nz z) �)zCREATE zUnknown schema content: %r�key� fk_constraint� ck_constraint)r �charset�re�split� startswithr � initial_quote� _parse_column�_parse_table_options�_parse_table_name�_parse_constraintsr �warnr �appendr r )r Z show_creater# �state�line�type_�specr r r �parse) s0 z MySQLTableDefinitionParser.parsec 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>f s zAMySQLTableDefinitionParser._parse_constraints.<locals>.<listcomp>�localc S s g | ] }|d �qS )r r )r5 r6 r r r r7 h s Zforeignr! r" � partitionN) �_re_key�match� groupdict�_parse_keyexprs�_re_key_version_sqlr Zunformat_identifiers�_re_fk_constraint�_re_ck_constraint� _re_partition)r r/ �mr1 �m2r r r r+ I s8 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 r/ r. �regex�cleanuprB r r r r* { s z,MySQLTableDefinitionParser._parse_table_namec C s� i }|r||dkrnj|dd� }x\| j D ]R\}}|�|�}|s>q&|�d�|�d� }} |r`|| �} | ||�� < |�d|�}q&W xdD ]} |�| d� q�W x*|�� D ] \} } | |jd| jj | f < q�W dS )z�Build a dictionary of all reflected table-level options. :param line: The final line of SHOW CREATE TABLE output. r N� directive�val� )�auto_incrementzdata directoryzindex directoryz%s_%s) � _pr_options�searchrF �lower�sub�pop�itemsr r rD ) r r/ r. �optionsZ rest_of_linerG rH rB rI �valueZnope�optrJ r r r r) � s" z/MySQLTableDefinitionParser._parse_table_optionsc C s� d}| j �|�}|r&|�� }d|d< n | j�|�}|rF|�� }d|d< |s\t�d| � dS |d srt�d| � |d |d |d }}}y| 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< x$dD ] } |�| d��rRd| | < �qRW x(dD ] } |�| d��rx|| | | < �qxW t|t��r�t�| �} t|t��r�d | k�r�d| d< || | �} i } d| d< |�dd�dk�r d| d< |�dd��rd| d< nt|t j��r.d| d< |�dd�}|dk�rHd}|�d d�}|dk �rr|�d d ��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 %rrD Zcoltype�argz*Did not recognize type '%s' of column '%s'rK r �'���c S s g | ] }t |��qS r )�int)r5 �vr r r r7 � s z<MySQLTableDefinitionParser._parse_column.<locals>.<listcomp>Zfsp)ZunsignedZzerofill)r# �collateZretrieve_as_bitwise�nullableZnotnullzNOT NULLZautoincrZ autoincrement�default�NULL�commentz\\�\z'')rD �typer^ r` ) � _re_columnr; r< �_re_column_looser r, r Z ischema_names�KeyError�sqltypesZNullType� _re_csv_str�findall� _re_csv_int� issubclassr r r rQ �getr Z _strip_valuesr ZInteger�replace�dict�updater r- )r r/ r. r1 rB rD r0 �args�col_typeZ type_argsZtype_kw�kwZ type_instanceZcol_kwr^ r` Zcol_dr r r r( � sv z(MySQLTableDefinitionParser._parse_columnc s g }x�|D ]� � fdd�dD �\}}}}}dg} | � | j�|�� | � |� |sZ| � d� |r�d|krhnh|�d�r�|�d�r�| � d � | � |� n>|d kr�| � d � | � |� n | � d � | � d |�d d � � |r�| � |� |� d�| �� q W d�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 s g | ] }� | �qS r r )r5 �i)�rowr r r7 s zBMySQLTableDefinitionParser._describe_to_create.<locals>.<listcomp>)r r � � � � zNOT NULLrL � timestamp�C�DEFAULTr_ z'%s'rX z''rK zCREATE TABLE %s ( z, z ) )r- r Zquote_identifierr& rl �join) r r r �bufferrD rp r] r^ �extrar/ r )rs r �_describe_to_create s8 z.MySQLTableDefinitionParser._describe_to_createc C s | j �|�S )z8Unpack '"col"(2),"col" ASC'-ish strings into components.)� _re_keyexprsrh )r Z identifiersr r r r= 7 s z*MySQLTableDefinitionParser._parse_keyexprsc C s6 g | _ g | _| jj}ttddd� | jj|| j�|�fD ���}td| | jj �| _ t d| �| _ t d�| _ t d�| _t d| �| _t d | �| _t d | �| _t d �| _|�� }d |d < t d| �| _t d| �| _t d�| _xtD ]}| �|� q�W xdD ]}| �|� q�W | �dd� | �dd� | �dd� dS )z Pre-compile regular expressions.)ZiqZfqZesc_fqc S s g | ]}t �|��qS r )r$ �escape)r5 �sr r r r7 H s z<MySQLTableDefinitionParser._prep_regexes.<locals>.<listcomp>zM^CREATE (?:\w+ +)?TABLE +%(iq)s(?P<name>(?:%(esc_fq)s|[^%(fq)s])+)%(fq)s +\($zW(?:(?:%(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\(\)]+)?)))?(?: +(?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|NOACTION�ona, 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>[^\)]+?)\)(?: +(?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(?:.*))ZENGINEZTYPEZAUTO_INCREMENTZAVG_ROW_LENGTHz CHARACTER SETzDEFAULT CHARSETZCHECKSUMZCOLLATEZDELAY_KEY_WRITEZ INSERT_METHODZMAX_ROWSZMIN_ROWSZ PACK_KEYSZ ROW_FORMATZKEY_BLOCK_SIZEZUNIONz \([^\)]+\)Z TABLESPACEz.*? STORAGE DISKZ RAID_TYPEz4\w+\s+RAID_CHUNKS\s*\=\s*\w+RAID_CHUNKSIZE\s*=\s*\w+N)Z _re_columnsrM r Z final_quoterm �zipr' Z_escape_identifier� _pr_compileZ_unescape_identifierrE � _re_compiler rg ri rc rd r: r> �copyr? r@ rA �_options_of_type_string�_add_option_string�_add_option_word�_add_option_regex)r Z_final�quotesrq �optionr r r r < sb z(MySQLTableDefinitionParser._prep_regexesz(?:\s*(?:=\s*)|\s+)c C s. dt �|�| jf }| j�t|dd� �� d S )Nz0(?P<directive>%s)%s'(?P<val>(?:[^']|'')*?)'(?!')c S s | � dd�� dd�S )Nz\\ra z''rX )rl )r[ r r r �<lambda>� � z?MySQLTableDefinitionParser._add_option_string.<locals>.<lambda>)r$ r� �_optional_equalsrM r- r� )r rI rG r r r r� � 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� rM r- r� )r rI rG r r r r� � s z+MySQLTableDefinitionParser._add_option_wordc C s* dt �|�| j|f }| j�t|�� d S )Nz (?P<directive>%s)%s(?P<val>%s))r$ r� r� rM r- r� )r rI rG r r r r� � s z,MySQLTableDefinitionParser._add_option_regexN)r r r r r r2 r+ r* r) r( r~ r= r r� r� r� r� r r r r r s 2 _5 / r )�COMMENTzDATA DIRECTORYzINDEX DIRECTORYZPASSWORDZ CONNECTIONc C s t | �|fS )z1Prepare a 2-tuple of compiled regex and callable.)r� )rG rH r r r r� s r� c C s t �| t jt jB �S )z)Compile a string to regex, I and UNICODE.)r$ �compile�I�UNICODE)rG r r r r� s r� )N)r$ Z enumeratedr r r r r r rK r rf r �objectr Z class_loggerr r� r� r� r r r r �<module> s" j