관리-도구
편집 파일: mysql.cpython-38.pyc
U 0?�f�C � @ sr d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d d l m Z d d l mZ d d l mZ d d l mZ d dl mZ d dl mZ d dl mZ d dl mZ d dlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z e�rRd dlm Z d dl m Z d dl!m"Z" d d l#m$Z$ d d l%m&Z& d d l m'Z' G d d � d e�Z(G d!d"� d"e(�Z)G d#d$� d$e�Z*G d%d&� d&e�Z+G d'd(� d(e+�Z,e ed)d*�e ed)d*�e ed)d*�e ed)d*�d+d,� ����Z-e e*d)d*�d$d-d.d/�d0d1��Z.e e,d)d*�d(d-d.d/�d2d3��Z/e e+d)d*�d&d-d.d/�d4d5��Z0d-d6d7d8d6d9d.d:�d;d<�Z1e e j"d)d*�d=d-d.d/�d>d?��Z2dS )@� )� annotationsN)�Any)�Optional)� TYPE_CHECKING)�Union��schema)�types� )� alter_table)� AlterColumn)� ColumnDefault)� ColumnName)�ColumnNullable)� ColumnType)�format_column_name)�format_server_default)� DefaultImpl� )�util)� sqla_compat)� _is_mariadb)�_is_type_bound)�compiles)�Literal)�MySQLDDLCompiler)�DropConstraint)� Constraint)� TypeEngine)�_ServerDefaultc s� e Zd ZdZdZejddhddhf ZddgZd$d d d d d dd ddd d d dd ddd�� fdd� Zddd�� fdd� Z dd dd�d d �Z d d � Z d d!� Z d"d#� Z � ZS )%� MySQLImpl�mysqlFZBOOLZTINYINT�JSONZLONGTEXTzcharacter set ([\w\-_]+)zcollate ([\w\-_]+)N�str�Optional[bool]z%Union[_ServerDefault, Literal[False]]� Optional[str]�Optional[TypeEngine]zOptional[_ServerDefault]�$Optional[Union[str, Literal[False]]]r �None)� table_name� column_name�nullable�server_default�name�type_r � existing_type�existing_server_default�existing_nullable� autoincrement�existing_autoincrement�comment�existing_comment�kw�returnc s� t �|| �st �|| �r>t� j||f||||| || d�|�� |d k s^| �|d k rV|n||�r�| �t||||d k rv|n||d k r�|n| d k r�| nd|d k r�|n||dk r�|n| | d k r�| n| | dk r�| n|d� � n�|d k s�|d k s�| d k s�| dk �rz| �t||||d k �r|n||d k �r |n| d k �r.| nd|d k �r>|n||dk �rN|n| | d k �r^| n| | dk �rn| n|d� � n |dk �r�| �t ||||d�� d S )N)r+ r. r r/ r1 r, r0 TF)r �newnamer+ r. �defaultr2 r4 r ) r Z_server_default_is_identityZ_server_default_is_computed�super� alter_column�$_is_mysql_allowed_functional_default�_exec�MySQLChangeColumn�MySQLModifyColumn�MySQLAlterDefault)�selfr) r* r+ r, r- r. r r/ r0 r1 r2 r3 r4 r5 r6 �� __class__� �?/opt/hc_python/lib/python3.8/site-packages/alembic/ddl/mysql.pyr; 3 s� � ���� � ������� ���������� ��zMySQLImpl.alter_columnr )�constr7 c s( t |tj�rt|�rd S t� �|� d S �N)� isinstancer �CheckConstraintr r: �drop_constraint)rA rF rB rD rE rJ � s zMySQLImpl.drop_constraint�bool)r. r, r7 c C s |d k o|j tjko|d k S rG )�_type_affinity�sqltypes�DateTime)rA r. r, rD rD rE r< � s ��z.MySQLImpl._is_mysql_allowed_functional_defaultc C s" |j jtjkr*|jr*|js*|s*|dkr*dS |r^|j jtjkr^|d k rRt�dd|�nd }||kS |r�|j jtjkr�t�dd|�}|d|� d�kS |�r|�rt� d|� � �}t� d|� � �}|r�|s�dS |� d�|� d�kr�dS |� d �}|� d �}t�d d |� � �t�d d |� � �kS ||kS d S ) Nz'0'Fz^'|'$� �'z (.*) (on update.*?)(?:\(\))?$Tr r z(.*?)(?:\(\))?$z\1) �typerL rM �IntegerZ primary_keyr2 �re�sub�String�match�lower�group)rA Zinspector_columnZmetadata_columnZrendered_metadata_defaultZrendered_inspector_defaultZmetadata_defaultZ onupdate_insZ onupdate_metrD rD rE �compare_server_default� sl ������ ���� � �� � ��z MySQLImpl.compare_server_defaultc C s� t � }t|�D ]~}|jrq|jD ]j}|j|jkrJ|�|� |�|j� q|jD ]*}|j|jkrP|�|� |�|j� q|qP|j|kr qq qt|�D ]}|j|kr�|�|� q�d S rG )�set�list�unique�columnsr- �remove�addZ foreign_keys) rA Zconn_unique_constraintsZ conn_indexesZmetadata_unique_constraintsZmetadata_indexesZremoved�idx�col�fkrD rD rE � correct_for_autogen_constraints� s& z)MySQLImpl.correct_for_autogen_constraintsc s� � fdd�|D �}� fdd�|D �}t |��|�D ]d}|| }|| }|jd k rn|j�� dkrn|jd krnd|_|jd k r2|j�� dkr2|jd kr2d|_q2d S )Nc s i | ]}� � |�j|�qS rD )Z _create_reflected_constraint_sig�unnamed_no_options��.0rb �rA rD rE � <dictcomp># s � z=MySQLImpl.correct_for_autogen_foreignkeys.<locals>.<dictcomp>c s i | ]}� � |�j|�qS rD )Z _create_metadata_constraint_sigrd re rg rD rE rh ' s � ZrestrictZRESTRICT)rZ � intersectionZondeleterW Zonupdate)rA Zconn_fksZ metadata_fksZconn_fk_by_sigZmetadata_fk_by_sig�sigZmdfkZcnfkrD rg rE � correct_for_autogen_foreignkeys" s. � �� ��� ��z)MySQLImpl.correct_for_autogen_foreignkeys) NFNNNNNNNNFN)�__name__� __module__� __qualname__� __dialect__Ztransactional_ddlr Z type_synonymsZtype_arg_extractr; rJ r< rY rc rk � __classcell__rD rD rB rE r ) s2 � �2s C%r c @ s e Zd ZdZdS )� MariaDBImpl�mariadbN)rl rm rn ro rD rD rD rE rq @ s rq c s, e Zd Zd dddddd�� fdd� Z� ZS ) r@ Nr# r r% r( )r- r* r9 r r7 c s$ t t| �j||d� || _|| _d S )Nr )r: r �__init__r* r9 )rA r- r* r9 r rB rD rE rs E s zMySQLAlterDefault.__init__)N�rl rm rn rs rp rD rD rB rE r@ D s �r@ c s6 e Zd Zd dddddddddd d � � fd d � Z� ZS )r> NFr# r% r& r$ �/Optional[Union[_ServerDefault, Literal[False]]]r' r( ) r- r* r r8 r. r+ r9 r2 r4 r7 c sZ t t| �j||d� || _|| _|| _|| _|| _| | _|d krJt � d��t � |�| _ d S )Nr zDAll MySQL CHANGE/MODIFY COLUMN operations require the existing type.)r: r rs r* r+ r8 r9 r2 r4 r � CommandErrorrM Z to_instancer. ) rA r- r* r r8 r. r+ r9 r2 r4 rB rD rE rs R s �zMySQLChangeColumn.__init__)NNNNFNFrt rD rD rB rE r> Q s �r> c @ s e Zd ZdS )r? N)rl rm rn rD rD rD rE r? n s r? r! rr c K s t d��d S )Nz9Individual alter column constructs not supported by MySQL)�NotImplementedError��element�compilerr6 rD rD rE � _mysql_doesnt_support_individualr s �r{ r r# )ry rz r7 c K s< dt || j| j�t|| j�| jd k r4dt|| j� ndf S )Nz%s ALTER COLUMN %s %szSET DEFAULT %sz DROP DEFAULT)r r) r r r* r9 r rx rD rD rE �_mysql_alter_default| s ��r| c K s< dt || j| j�t|| j�t|| j| j| j| j | j d�f S )Nz%s MODIFY %s %s�r+ r, r. r2 r4 ) r r) r r r* �_mysql_colspecr+ r9 r. r2 r4 rx rD rD rE �_mysql_modify_column� s ��r c K sF dt || j| j�t|| j�t|| j�t|| j| j| j | j | j d�f S )Nz%s CHANGE %s %s %sr} ) r r) r r r* r8 r~ r+ r9 r. r2 r4 rx rD rD rE �_mysql_change_column� s ��r� r$ ru r r' )rz r+ r, r. r2 r4 r7 c C sn d| j j�|�|rdndf }|r*|d7 }|dk rL|d k rL|dt| |� 7 }|rj|d| j�|t�� � 7 }|S )Nz%s %sZNULLzNOT NULLz AUTO_INCREMENTFz DEFAULT %sz COMMENT %s)�dialectZ type_compiler�processr Z sql_compilerZrender_literal_valuerM rU )rz r+ r, r. r2 r4 �specrD rD rE r~ � s � �r~ r c K s� | j }t|tjtjtjf�r.|r$t�|�| �S t|tj�r�t |j �rbd|j � |j �|j �|�f S d|j � |j �|j �|�f S ntd��dS )zVRedefine SQLAlchemy's drop constraint to raise errors for invalid constraint type.z!ALTER TABLE %s DROP CONSTRAINT %sz ALTER TABLE %s DROP CHECK %szFNo generic 'DROP CONSTRAINT' in MySQL - please specify constraint typeN)ry rH r ZForeignKeyConstraintZPrimaryKeyConstraintZUniqueConstraint�AssertionErrorZvisit_drop_constraintrI r r� �preparerZ format_table�tableZformat_constraintrw )ry rz r6 � constraintrD rD rE �_mysql_drop_constraint� s. �� � ��r� )3� __future__r rS �typingr r r r Z sqlalchemyr r rM �baser r r r r r r r �implr rO r r Zutil.sqla_compatr r r r Z sqlalchemy.dialects.mysql.baser Zsqlalchemy.sql.ddlr Zsqlalchemy.sql.schemar Zsqlalchemy.sql.type_apir r r rq r@ r> r? r{ r| r r� r~ r� rD rD rD rE �<module> sb