File: //opt/alt/python37/lib/python3.7/site-packages/alembic/ddl/__pycache__/mysql.cpython-37.pyc
B
�M!V�- � @ sR d dl mZ d dlmZ d dlmZ ddlmZ ddlm Z ddl
m Z dd l m
Z
mZmZmZmZmZmZ dd
l mZ dd lmZ G d d
� d
e �ZG dd� de�ZG dd� de�ZG dd� de�Zee
d�eed�eed�eed�dd� ����Zeed�dd� �Z eed�dd� �Z eed�dd � �Z d d � Z d d � Z eej!d�d!d"� �Z"d#S )$� )�compiles)�types)�schema� )� string_types)�util� )� DefaultImpl)�ColumnNullable�
ColumnName�
ColumnDefault�
ColumnType� AlterColumn�format_column_name�format_server_default)� alter_table)�comparec
@ s6 e Zd ZdZdZd dd�Zdd� Zdd � Zd
d � ZdS )
� MySQLImpl�mysqlFNc
K s |d k rd| � t|||||d k r"|n|
d k r.|
nd|d k r<|n||dk rJ|n| | d k rX| n| d�� n�|d k s||d k s|| d k r�| � t||||d k r�|n||d k r�|n|
d k r�|
nd|d k r�|n||dk r�|n| | d k r�| n| d�� n |dk �r| � t||||d�� d S )NTF)r �newname�nullable�type_�default�
autoincrement)r )�_exec�MySQLChangeColumn�MySQLModifyColumn�MySQLAlterDefault)�self�
table_name� column_namer �server_default�namer r Z
existing_typeZexisting_server_defaultZexisting_nullabler Zexisting_autoincrement�kw� r$ �B/opt/alt/python37/lib/python3.7/site-packages/alembic/ddl/mysql.py� alter_column sF
zMySQLImpl.alter_columnc C s6 |j jtjkr*|jr*|js*|s*|dkr*dS ||kS d S )Nz'0'F)�typeZ_type_affinity�sqltypesZIntegerZ primary_keyr )r Zinspector_columnZmetadata_columnZrendered_metadata_defaultZrendered_inspector_defaultr$ r$ r% �compare_server_defaultM s z MySQLImpl.compare_server_defaultc C sz t � }x�t|�D ]�}|jr qxr|jD ]h}|j|jkrL|�|� |�|j� P x2|jD ](}|j|jkrT|�|� |�|j� P qTW |j|kr$P q$W qW x$t|�D ]}|j|kr�|�|� q�W t dd� |D ��} t dd� |D ��}
t dd� |D ��} tdd� |D ��} tdd� |D ��}
x`t | �� |
�D ]N}|| k�rXt
� | | �j |
k�rp|�
| | � n|| k�r$|�
|
| � �q$W d S )Nc S s g | ]}|j d k r|j �qS )N)r" )�.0�consr$ r$ r% �
<listcomp>� s z=MySQLImpl.correct_for_autogen_constraints.<locals>.<listcomp>c S s" g | ]}|j d krt�|�j�qS )N)r" r �_uq_constraint_sig�sig)r* r+ r$ r$ r% r, � s c S s g | ]}|j r|j�qS r$ )�uniquer" )r* r+ r$ r$ r% r, � s c s s | ]}|j |fV qd S )N)r" )r* r+ r$ r$ r% � <genexpr>� s z<MySQLImpl.correct_for_autogen_constraints.<locals>.<genexpr>c s s | ]}|j r|j|fV qd S )N)r/ r" )r* r+ r$ r$ r% r0 � s )�set�listr/ �columnsr" �remove�addZ foreign_keys�dict� intersectionr r- r. �discard)r Zconn_unique_constraintsZ conn_indexesZmetadata_unique_constraintsZmetadata_indexes�removed�idx�col�fkZmetadata_uq_namesZunnamed_metadata_uqsZmetadata_ix_namesZ
conn_uq_namesZ
conn_ix_names�overlapr$ r$ r% � correct_for_autogen_constraints] sJ
z)MySQLImpl.correct_for_autogen_constraintsc C s� t dd� |D ��}t dd� |D ��}xvt|��|�D ]d}|| }|| }|jd k rp|j�� dkrp|jd krpd|_|jd k r4|j�� dkr4|jd kr4d|_q4W d S )Nc s s | ]}t �|�j|fV qd S )N)r �_fk_constraint_sigr. )r* r< r$ r$ r% r0 � s z<MySQLImpl.correct_for_autogen_foreignkeys.<locals>.<genexpr>c s s | ]}t �|�j|fV qd S )N)r r? r. )r* r< r$ r$ r% r0 � s ZrestrictZRESTRICT)r6 r1 r7 Zondelete�lowerZonupdate)r Zconn_fksZ metadata_fksZconn_fk_by_sigZmetadata_fk_by_sigr. ZmdfkZcnfkr$ r$ r% � correct_for_autogen_foreignkeys� s
z)MySQLImpl.correct_for_autogen_foreignkeys)
NFNNNNNNNN) �__name__�
__module__� __qualname__Z __dialect__Ztransactional_ddlr& r) r> rA r$ r$ r$ r% r s
/Dr c s e Zd Zd� fdd� Z� ZS )r Nc s$ t t| �j||d� || _|| _d S )N)r )�superr �__init__r r )r r" r r r )� __class__r$ r% rF � s zMySQLAlterDefault.__init__)N)rB rC rD rF �
__classcell__r$ r$ )rG r% r � s r c s e Zd Zd� fdd� Z� ZS )r NFc sT t t| �j||d� || _|| _|| _|| _|| _|d krDt� d��t
� |�| _ d S )N)r zDAll MySQL CHANGE/MODIFY COLUMN operations require the existing type.)
rE r rF r r r r r r Z CommandErrorr( Z to_instancer ) r r" r r r r r r r )rG r$ r% rF � s zMySQLChangeColumn.__init__)NNNNFN)rB rC rD rF rH r$ r$ )rG r% r � s r c @ s e Zd ZdS )r N)rB rC rD r$ r$ r$ r% r � s r r c K s t d��d S )Nz9Individual alter column constructs not supported by MySQL)�NotImplementedError)�element�compilerr# r$ r$ r% � _mysql_doesnt_support_individual� s rL 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 r r )rJ rK r# r$ r$ r% �_mysql_alter_default� s
rM c
K s8 dt || j| j�t|| j�t|| j| j| j| j d�f S )Nz%s MODIFY %s %s)r r! r r )
r r r r r �_mysql_colspecr r r r )rJ rK r# r$ r$ r% �_mysql_modify_column� s
rO c K sB dt || j| j�t|| j�t|| j�t|| j| j| j | j
d�f S )Nz%s CHANGE %s %s %s)r r! r r ) r r r r r r rN r r r r )rJ rK r# r$ r$ r% �_mysql_change_column s
rP c C s" t |t�rd| S | j�|�S d S )Nz'%s')�
isinstancer Z sql_compiler�process)rK �exprr$ r$ r% �
_render_value s
rT c C sP d| j j�|�|rdndf }|r*|d7 }|dk rL|d k rL|dt| |� 7 }|S )Nz%s %s�NULLzNOT NULLz AUTO_INCREMENTFz DEFAULT %s)ZdialectZ
type_compilerrR rT )rK r r! r r �specr$ r$ r% rN s rN c K sL | j }t|tjtjtjf�r*|j| f|�S t|tj�r@td��ntd��dS )zVRedefine SQLAlchemy's drop constraint to
raise errors for invalid constraint type.z)MySQL does not support CHECK constraints.zFNo generic 'DROP CONSTRAINT' in MySQL - please specify constraint typeN) rJ rQ r ZForeignKeyConstraintZPrimaryKeyConstraintZUniqueConstraintZvisit_drop_constraintZCheckConstraintrI )rJ rK r# Z
constraintr$ r$ r% �_mysql_drop_constraint% s
rW N)#Zsqlalchemy.ext.compilerr Z
sqlalchemyr r( r Z util.compatr � r �implr �baser
r r r
r r r r Z autogenerater r r r r rL rM rO rP rT rN ZDropConstraintrW r$ r$ r$ r% �<module> s. $ +