관리-도구
편집 파일: mssql.cpython-38.pyc
U 0?�f�7 � @ s� d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d l m Z d d l mZ d d l mZ d d lmZ d d lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddl!m"Z" d d l#m$Z$ d d l$m%Z% d d l&m'Z' e�r�d d lm(Z( d d!l)m*Z* d d"l)m+Z+ d d#l,m-Z- d d$l.m/Z/ d d%l.m0Z0 d d&l1m2Z2 d d'l3m4Z4 dd(lm5Z5 G d)d*� d*e"�Z6G d+d,� d,ee�Z7G d-d.� d.ee�Z8e'e7d/�d,d0d1d2�d3d4��Z9e'e8d/�d.d0d1d2�d5d6��Z:e'ed/�d7d8d1d2�d9d:��Z;d8d;d1d<�d=d>�Z<e'ed/�d?d8d1d2�d@dA��Z=e'ed/�dBd8d1d2�dCdD��Z>e'ed/�dEd8d1d2�dFdG��Z?e'ed/�dHd8d1d2�dIdJ��Z@e'e d/�dKd8d1d2�dLdM��ZAdS )N� )� annotationsN)�Any)�Dict)�List)�Optional)� TYPE_CHECKING)�Union)�types)�Column)� CreateIndex)� Executable)� ClauseElement� )� AddColumn)� alter_column)� alter_table)� ColumnDefault)� ColumnName)�ColumnNullable)� ColumnType)�format_column_name)�format_server_default)�format_table_name)� format_type)� RenameTable)� DefaultImpl� )�util)� sqla_compat)�compiles)�Literal)� MSDDLCompiler)� MSSQLCompiler)� CursorResult)�Index)�Table)� TableClause)� TypeEngine)�_ServerDefaultc s e Zd ZdZdZdZejddhf Zejd Zdd�� fd d � Z d d d �� fdd� Z dd�dd�Z dd�� fdd� Z d4ddddddddddd dd � � fd d � Z d d dd �d!d"�Zd#d$d dd%�� fd&d'� Zd5dd(ddd)�� fd*d+� Zd,d-� Z� fd.d/�Zd0dd0d1�d2d3�Z� ZS )6� MSSQLImpl�mssqlTZGOZVARCHARZNVARCHAR)ZminvalueZmaxvalueZ nominvalueZ nomaxvalue�cycle�cache�None)�returnc s$ t � j||� | j�d| j�| _d S )NZmssql_batch_separator)�super�__init__Z context_opts�get�batch_separator)�self�arg�kw�� __class__� �?/opt/hc_python/lib/python3.8/site-packages/alembic/ddl/mssql.pyr0 B s �zMSSQLImpl.__init__r zOptional[CursorResult])� constructr. c s0 t � j|f|�|�}| jr,| jr,| �| j� |S �N)r/ �_exec�as_sqlr2 � static_output)r3 r: �argsr5 �resultr6 r8 r9 r< H s zMSSQLImpl._execc C s | � d| j � d S )NzBEGIN TRANSACTION)r> Zcommand_terminator�r3 r8 r8 r9 � emit_beginN s zMSSQLImpl.emit_beginc s& t � �� | jr"| jr"| �| j� d S r; )r/ � emit_commitr= r2 r> rA r6 r8 r9 rC Q s zMSSQLImpl.emit_commitNF�strzOptional[bool]z/Optional[Union[_ServerDefault, Literal[False]]]� Optional[str]zOptional[TypeEngine]zOptional[_ServerDefault]) � table_name� column_name�nullable�server_default�name�type_�schema� existing_type�existing_server_default�existing_nullabler5 r. c s( |d k r.|d k r|}d }q^|d kr^t �d��n0| d k rL|d k rL| }|}d }n|d k r^t �d� d} t�|| �szt�|| �r�d} || d< | | d< t� j||f||||| d�| �� |dk �r| dk�r| dk s�|d kr�| �t ||d|�� |d k �rt� j||||d � |d k �r$t� j||||d � d S ) NzhMS-SQL ALTER COLUMN operations with NULL or NOT NULL require the existing_type or a new type_ be passed.z�MS-SQL ALTER COLUMN operations that specify type_= should also specify a nullable= or existing_nullable= argument to avoid implicit conversion of NOT NULL columns to NULL.FTrI rN )rH rK rL rM rO �sys.default_constraints)rL rI )rL rJ ) r � CommandError�warnr Z_server_default_is_identityZ_server_default_is_computedr/ r r< �_ExecDropConstraint) r3 rF rG rH rI rJ rK rL rM rN rO r5 Z used_defaultr6 r8 r9 r V s� �� � ����� �� � �zMSSQLImpl.alter_columnr$ )�indexr5 r. c K s` |j �dd �pd}|jd k s t�|D ]$}||jjkr$|j�t|tj�� q$| � t |f|�� d S )N� mssql_includer8 ) �kwargsr1 �table�AssertionError�c� append_columnr �sqltypesZNullTyper< r )r3 rT r5 rU �colr8 r8 r9 � create_index� s zMSSQLImpl.create_indexzUnion[TableClause, Table]z List[dict])rW �rowsr5 r. c s` | j rJ| �d| jj�|� � t� j||f|� | �d| jj�|� � nt� j||f|� d S )NzSET IDENTITY_INSERT %s ONzSET IDENTITY_INSERT %s OFF)r= r< �dialectZidentifier_preparerZ format_tabler/ � bulk_insert)r3 rW r^ r5 r6 r8 r9 r` � s �� ��zMSSQLImpl.bulk_insert� Column[Any])rF �columnrL r. c s� |� dd�}|r$| �t||d|�� |� dd�}|rH| �t||d|�� |� dd�}|rj| �t|||�� t� j||fd|i|�� d S )NZmssql_drop_defaultFrP Zmssql_drop_checkzsys.check_constraintsZmssql_drop_foreign_keyrL )�popr< rS �_ExecDropFKConstraintr/ � drop_column)r3 rF rb rL r5 Z drop_defaultZ drop_checkZdrop_fksr6 r8 r9 re � s0 �� �� zMSSQLImpl.drop_columnc C s4 |d k rt �dd|�}|d k r,t �dd|�}||kS )Nz [\(\) \"\']� )�re�sub)r3 Zinspector_columnZmetadata_columnZrendered_metadata_defaultZrendered_inspector_defaultr8 r8 r9 �compare_server_default� s � �z MSSQLImpl.compare_server_defaultc sH t � �||�\}}}|d kr>|d k r>|s>|jd k r>|jjr>d}|||fS )NF)r/ �_compare_identity_defaultrb Z primary_key)r3 Zmetadata_identityZinspector_identity�diffZignoredZis_alterr6 r8 r9 rj � s � ����� z#MSSQLImpl._compare_identity_defaultzDict[str, Any])�reflected_object�kindr. c C s@ |� di ��� }|� d�s&|�dd � |� d�s<|�dd � |S )NZdialect_optionsrU Zmssql_clustered)r1 �copyrc )r3 rl rm �optionsr8 r8 r9 � adjust_reflected_dialect_options s z*MSSQLImpl.adjust_reflected_dialect_options)NFNNNNNN)N)�__name__� __module__� __qualname__Z __dialect__Ztransactional_ddlr2 r Z type_synonymsZidentity_attrs_ignorer0 r< rB rC r r] r` re ri rj rp � __classcell__r8 r8 r6 r9 r) 3 s2 �*X � r) c @ s& e Zd ZdZdddddd�dd�Zd S ) rS FrD zUnion[Column[Any], str]rE r- )�tname�colnamerK rL r. c C s || _ || _|| _|| _d S r; )ru rv rK rL )r3 ru rv rK rL r8 r8 r9 r0 s z _ExecDropConstraint.__init__N�rq rr rs Z inherit_cacher0 r8 r8 r8 r9 rS s rS c @ s$ e Zd ZdZddddd�dd�Zd S ) rd FrD ra rE r- )ru rv rL r. c C s || _ || _|| _d S r; )ru rv rL )r3 ru rv rL r8 r8 r9 r0 * s z _ExecDropFKConstraint.__init__Nrw r8 r8 r8 r9 rd ' s rd r* r" rD )�element�compilerr. c K sD | j | j| j| jf\}}}}d|||t|||�|r:|d ndd� S )Na declare @const_name varchar(256) select @const_name = QUOTENAME([name]) from %(type)s where parent_object_id = object_id('%(schema_dot)s%(tname)s') and col_name(parent_object_id, parent_column_id) = '%(colname)s' exec('alter table %(tname_quoted)s drop constraint ' + @const_name)�.rf )�typeru rv � tname_quoted� schema_dot)rL ru rv rK r )rx ry r5 rL ru rv rK r8 r8 r9 �_exec_drop_col_constraint2 s � ��r~ c K s< | j | j| j }}}d||t|||�|r2|d ndd� S )Nay declare @const_name varchar(256) select @const_name = QUOTENAME([name]) from sys.foreign_keys fk join sys.foreign_key_columns fkc on fk.object_id=fkc.constraint_object_id where fkc.parent_object_id = object_id('%(schema_dot)s%(tname)s') and col_name(fkc.parent_object_id, fkc.parent_column_id) = '%(colname)s' exec('alter table %(tname_quoted)s drop constraint ' + @const_name)rz rf )ru rv r| r} )rL ru rv r )rx ry r5 rL ru rv r8 r8 r9 � _exec_drop_col_fk_constraintK s ��r r r! c K s$ dt || j| j�t|| jf|�f S )Nz%s %s)r rF rL �mssql_add_columnrb �rx ry r5 r8 r8 r9 �visit_add_column_ s �r� ra )ry rb r. c K s d| j |f|� S )NzADD %s)Zget_column_specification)ry rb r5 r8 r8 r9 r� g s r� r c K s6 dt || j| j�t|| j�t|| j�| jr.dndf S )Nz %s %s %s %sZNULLzNOT NULL)r rF rL r rG r rM rH r� r8 r8 r9 �visit_column_nullablem s �r� r c K s* dt || j| j�t|| j�t|| j�f S )Nz%s ADD DEFAULT %s FOR %s)r rF rL r �defaultr rG r� r8 r8 r9 �visit_column_defaulty s �r� r c K s* dt || j| j�t|| j�t|| j�f S )Nz$EXEC sp_rename '%s.%s', %s, 'COLUMN')r rF rL r rG Znewnamer� r8 r8 r9 �visit_rename_column� s �r� r c K s* dt || j| j�t|| j�t|| j�f S )Nz%s %s %s)r rF rL r rG r rK r� r8 r8 r9 �visit_column_type� s �r� r c K s" dt || j| j�t || jd �f S )NzEXEC sp_rename '%s', %s)r rF rL Znew_table_namer� r8 r8 r9 �visit_rename_table� s �r� )B� __future__r rg �typingr r r r r r Z sqlalchemyr r[ Zsqlalchemy.schemar r Zsqlalchemy.sql.baser Zsqlalchemy.sql.elementsr �baser r r r r r r r r r r r �implr rf r r Zutil.sqla_compatr r Z sqlalchemy.dialects.mssql.baser! r" Zsqlalchemy.engine.cursorr# Zsqlalchemy.sql.schemar$ r% Zsqlalchemy.sql.selectabler& Zsqlalchemy.sql.type_apir' r( r) rS rd r~ r r� r� r� r� r� r� r� r8 r8 r8 r9 �<module> sv e