U
|
¸ý°dZ ã
|
@ 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 dZdZd'ddZdd Zdd ZeedddZddd 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 ú[d:\z\workplace\vscode\pyvenv\venv\Lib\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 ZeedddZd 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|sq| |¡\}}|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 < qdS )zBuild 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 }| |ksz| |kr¬| |j kr| |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 kr |d d kr | j |¡} ndd | j |¡D } i }
|
t|tttfrL| rL| d ¡|
|
d< dD ] } | | d¡rPd|
|
| < qPdD ] } | | d¡rr|| |
|
| < qrt|ttfrÊt| } t|trÊd | krÊd|
|
d< || |
|
} i } d| d< | dd¡dkrúd| d< | dd¡rd| d< nt|t jr(d| d< | dd¡}|dkrBd}| d d¡}|dk r`t|}| d ¡}|dk r¦t|d }| d ¡}|dk r|d k|d!< || d"< t|| ||d#}| | ¡ |j |¡ dS )$zExtract 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 NULLZautoincrZ 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_valuesZIntegerÚ 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 ]ä fdddD \}}}}}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>T 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_createF 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 |fqS 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<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Ú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(?:.*))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ª D 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« K 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¬ R 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 a 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 8g5
|
>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¦ s 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§ y 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 K
|
÷ ÿ
|
