관리-도구
편집 파일: impl.cpython-38.pyc
U 0?�fiq � @ s� U d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d d lm Z d d lm Z d d lm Z d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl m Z e�r@d d lm Z 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'm(Z( d d#l'm)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)l-m1Z1 d d*l-m2Z2 d d+l-m3Z3 d d,l4m5Z5 d d-l6m7Z7 dd.lm8Z8 dd/l9m:Z: dd0l;m<Z< dd1l;m=Z= e�>e?�Z@G d2d3� d3eA�ZBi ZCd4eDd5< G d6d7� d7eBd8�ZEG d9d:� d:e �ZFd;d;d<d=d>�d?d@�ZGdS )A� )� annotationsN)�Any)�Callable)�Dict)�Iterable)�List)�Mapping)� NamedTuple)�Optional)�Sequence)�Set)�Tuple)�Type)� TYPE_CHECKING)�Union)�cast��schema)�text� )�_autogen)�base)�_constraint_sig)�ComparisonResult� )�util)� sqla_compat)�Literal)�TextIO)� Connection)�Dialect)� CursorResult)� Inspector)� ClauseElement)� Executable)� ColumnElement)� quoted_name)�Column)� Constraint)�ForeignKeyConstraint)�Index)�Table)�UniqueConstraint)� TableClause)� TypeEngine)�_ServerDefault)�AutogenContext)�ApplyBatchImpl)�BatchOperationsImplc @ s e Zd Zdddd�dd�ZdS )�ImplMeta�strzTuple[Type[DefaultImpl]]�Dict[str, Any])� classname�bases�dict_c C s( t �| |||�}d|kr$| t|d < |S )N� __dialect__)�type�__init__�_impls)�clsr6 r7 r8 Znewtype� r> �>/opt/hc_python/lib/python3.8/site-packages/alembic/ddl/impl.pyr; @ s zImplMeta.__init__N)�__name__� __module__� __qualname__r; r> r> r> r? r3 ? s r3 z Dict[str, Type[DefaultImpl]]r< c @ sn e Zd ZU dZdZdZdZddhfZded< d Z d ed < d Z d ed< dddddddd�dd�Z e ddd�dd ��Z d dd �d d �Zd!dd"�d#d$�Zd%d&dd'�d(d)�Zedd*�d+d,��Zd-d-e�� fd.d/d0d1d2d3�d4d5�Zd�d.d6dd7�d8d9�Zd�d d dd:d;d<d;dd=d;d<d>ddd?dd@�dAdB�Zd�d dCdDddE�dFdG�Zd�d dCd;ddE�dHdI�Zd?ddJ�dKdL�ZdMddJ�dNdO�Zd�d dPdDddQ�dRdS�Z d&ddT�dUdV�Z d&ddT�dWdX�Z dYd?ddZ�d[d\�Z d&ddT�d]d^�Z d&ddT�d_d`�Z!daddb�dcdd�Z"dYd?ddZ�dedf�Z#d�dhdidddj�dkdl�Z$dmdndb�dodp�Z%dndnddq�drds�Z&dndnddt�dudv�Z'dCdmddw�dxdy�Z(dzd{� Z)d|d}d|d}dd~�dd��Z*d�d�� Z+d�d�dd?d d��d�d��Z,d�d�d��d�d��Z-d�d�dd��d�d��Z.d�d�� Z/dd*�d�d��Z0dd*�d�d��Z1dd*�d�d��Z2d�d�d�d��d�d��Z3d�d�� Z4dYdYd;d��d�d��Z5d�d?d�d��d�d��Z6d�d?d�d��d�d��Z7dYdYd�d��d�d��Z8d�d�d�d��d�d��Z9d�d�� Z:dd dd��d�d��Z;d-S )�� DefaultImpla� Provide the entrypoint for major migration operations, including database-specific behavioral variances. While individual SQL/DDL constructs already provide for database-specific implementations, variances here allow for entirely different sequences of operations to take place for a particular migration, such as SQL Server's special 'IDENTITY INSERT' step for bulk inserts. �defaultF�;ZNUMERICZDECIMALzTuple[Set[str], ...]� type_synonymsr> z Sequence[str]�type_arg_extract)�orderZon_nullzTuple[str, ...]�identity_attrs_ignorer zOptional[Connection]�boolzOptional[bool]zOptional[TextIO]r5 �None)�dialect� connection�as_sql�transactional_ddl� output_buffer� context_opts�returnc C sZ || _ || _|| _|�dd�| _|| _i | _|| _|d k r@|| _| jrV| jsVt � d��d S )N� literal_bindsFz3Can't use literal_binds setting without as_sql mode) rL rM rN �getrS rP �memorQ rO r � CommandError)�selfrL rM rN rO rP rQ r> r> r? r; h s �zDefaultImpl.__init__zType[DefaultImpl])rL rR c C s t |j S �N)r< �name)r= rL r> r> r? �get_by_dialect� s zDefaultImpl.get_by_dialectr4 )r rR c C s, | j d k st�| j �|d � | j �� d S )Nz )rP �AssertionError�write�flush)rW r r> r> r? � static_output� s zDefaultImpl.static_outputr2 )�batch_oprR c C s dS )z�Return True if the given :class:`.BatchOperationsImpl` would need the table to be recreated and copied in order to proceed. Normally, only returns True on SQLite when operations other than add_column are present. Fr> )rW r_ r> r> r? �requires_recreate_in_batch� s z&DefaultImpl.requires_recreate_in_batchr1 r+ )� batch_impl�tablerR c C s dS )z�perform any operations needed on a table before a new one is created to replace it in batch mode. the PG dialect uses this to drop constraints on the table before the new one uses those same names. Nr> )rW ra rb r> r> r? �prep_table_for_batch� s z DefaultImpl.prep_table_for_batch)rR c C s | j S rX )rM �rW r> r> r? �bind� s zDefaultImpl.bindNzUnion[Executable, str]zOptional[Mapping[str, Any]]z%Optional[Sequence[Mapping[str, Any]]]�Mapping[str, Any]zOptional[CursorResult])� construct�execution_options� multiparams�paramsrR c C s� t |t�rt|�}| jr�|d k s$|r,td��| jrNt |tj�sNtddid�}ni }t rdt |t �sdt �|j f d| j i|��}| �t|��dd��� | j � d S | j}|d k s�t �|r�|jf |�}|r�|d k r�td��|r�|�||�S |�||�S d S ) Nz&SQL parameters not allowed with as_sqlrS T)�compile_kwargsrL � z z2Can't send params and multiparams at the same time)� isinstancer4 r rN � TypeErrorrS r Z DDLElement�dictr r# r[ �compilerL r^ �replace�strip�command_terminatorrM rh �execute)rW rg rh ri rj � compile_kw�compiled�connr> r> r? �_exec� s> ��� � zDefaultImpl._execzOptional[dict[str, Any]])�sqlrh rR c C s | � ||� d S rX )rx )rW ry rh r> r> r? rt � s zDefaultImpl.executez%Union[_ServerDefault, Literal[False]]z Optional[str]zOptional[TypeEngine]z$Optional[Union[str, Literal[False]]]zOptional[_ServerDefault]r )� table_name� column_name�nullable�server_defaultrY �type_r � autoincrement�comment�existing_comment� existing_type�existing_server_default�existing_nullable�existing_autoincrement�kwrR c K s* |d k s|d k r t jddd� |d k rF| �tj||||| | | | d�� |dk r�i }t�|| �rftj}n"t�|| �r�tj }| |d< ntj }| �||||f|| | | | d�|��� |d k r�| �tj ||||| | | | d�� | dk r�| �tj ||| || | | | d�� |d k �r&| �tj ||||| | | d�� d S )NzBautoincrement and existing_autoincrement only make sense for MySQL� )� stacklevel)r r� r� r� r� F�impl)r r� r� r� )r �warnrx r ZColumnNullabler Z_server_default_is_computedZComputedColumnDefaultZ_server_default_is_identityZIdentityColumnDefaultZ ColumnDefaultZ ColumnTypeZ ColumnCommentZ ColumnName)rW rz r{ r| r} rY r~ r r r� r� r� r� r� r� r� Zcls_r> r> r? � alter_column� s� ��� � � �� �� �� �� ��zDefaultImpl.alter_columnz Column[Any]z!Optional[Union[str, quoted_name]])rz �columnr rR c C s | � tj|||d�� d S �Nr )rx r Z AddColumn)rW rz r� r r> r> r? � add_columnK s zDefaultImpl.add_columnc K s | � tj|||d�� d S r� )rx r Z DropColumn)rW rz r� r r� r> r> r? � drop_columnS s zDefaultImpl.drop_column)�constrR c C s( |j d ks|� | �r$| �t�|�� d S rX )Z _create_rulerx r Z AddConstraint�rW r� r> r> r? �add_constraint\ s zDefaultImpl.add_constraintr( c C s | � t�|�� d S rX )rx r ZDropConstraintr� r> r> r? �drop_constraint` s zDefaultImpl.drop_constraintzUnion[str, quoted_name])�old_table_name�new_table_namer rR c C s | � tj|||d�� d S r� )rx r Z RenameTable)rW r� r� r r> r> r? � rename_tablec s �zDefaultImpl.rename_table)rb rR c C s� |j j|| jd| d� | �t�|�� |j j|| jd| d� |jD ]}| �t�|�� qB| j j oh| j j }|j }|r�|r�| � |� |jD ] }|j }|r�|r�| �|� q�d S �NF)Z checkfirstZ _ddl_runner)�dispatchZ before_createrM rx r Z CreateTableZ after_createZindexes� CreateIndexrL Zsupports_commentsZinline_commentsr� �create_table_comment�columns�create_column_comment)rW rb �indexZ with_commentr� r� r> r> r? � create_tablem s0 � � � zDefaultImpl.create_tablec C s@ |j j|| jd| d� | �t�|�� |j j|| jd| d� d S r� )r� Z before_droprM rx r Z DropTableZ after_drop�rW rb r> r> r? � drop_table� s � �zDefaultImpl.drop_tabler* )r� r� rR c K s | � tj|f|�� d S rX )rx r r� �rW r� r� r> r> r? � create_index� s zDefaultImpl.create_indexc C s | � t�|�� d S rX )rx r ZSetTableCommentr� r> r> r? r� � s z DefaultImpl.create_table_commentc C s | � t�|�� d S rX )rx r ZDropTableCommentr� r> r> r? �drop_table_comment� s z DefaultImpl.drop_table_commentzColumnElement[Any])r� rR c C s | � t�|�� d S rX )rx r ZSetColumnComment)rW r� r> r> r? r� � s z!DefaultImpl.create_column_commentc K s | � tj|f|�� d S rX )rx r Z DropIndexr� r> r> r? � drop_index� s zDefaultImpl.drop_indexTzUnion[TableClause, Table]z List[dict])rb �rows� multiinsertrR c s� t |t�std��n|r.t |d t�s.td��| jrj|D ].}| �t�� �jf � fdd�|� � D ��� q8n@|r�|r�| jt�� �|d� n"|D ] }| �t�� �jf |�� q�d S )Nz List expectedr z List of dictionaries expectedc s: i | ]2\}}|t |tj�s2tj||� j| jd �n|�qS ))r~ )rm r Z_literal_bindparam�cr: )�.0�k�v�rb r> r? � <dictcomp>� s � �� �z+DefaultImpl.bulk_insert.<locals>.<dictcomp>)ri ) rm �listrn ro rN rx r Z_insert_inline�values�items)rW rb r� r� �rowr> r� r? � bulk_insert� s. ��� ��zDefaultImpl.bulk_insertr' �Paramsc C s� | j j�|j��� }t�d|�}g }d }|D ] }t�d|�rB|}q,|�|� q,t |d |dd � g i �}|r�t�d|�D ]>}d|kr�|� d�\} } | � � |j | � � < qx|j �|� � � qx|S )Nz[\w\-_]+|\(.+?\)z^\(.*\)$r r z[^(),]+�=)rL Z type_compiler�processr: �lower�re�findall�match�appendr� �splitrr �kwargs�args) rW r� Z definition�tokensZ term_tokensZ paren_term�tokenrj �term�key�valr> r> r? �_tokenize_column_type� s z!DefaultImpl._tokenize_column_type)�inspector_params�metadata_paramsrR c C s| |j |j krdS dd� | jD �}d�|j g|j �}d�|j g|j �}|D ]*}||h�|�sp|j |j h�|�rL dS qLdS )NTc S s g | ]}d d� |D ��qS )c S s h | ] }|� � �qS r> )r� )r� �tr> r> r? � <setcomp>� s z=DefaultImpl._column_types_match.<locals>.<listcomp>.<setcomp>r> )r� �batchr> r> r? � <listcomp>� s z3DefaultImpl._column_types_match.<locals>.<listcomp>� F)�token0rF �joinr� �issubset)rW r� r� ZsynonymsZinspector_all_termsZmetadata_all_termsr� r> r> r? �_column_types_match� s$ � ���z DefaultImpl._column_types_match)�inspected_params� meta_paramsrR c C s� t |j�t |j�kr$|j|jkr$dS t |j�t |j�krH|j|jkrHdS d�|j��� }d�|j��� }| jD ]>}t�||�}t�||�}|rn|rn|�d�|�d�krn dS qndS )a We want to compare column parameters. However, we only want to compare parameters that are set. If they both have `collation`, we want to make sure they are the same. However, if only one specifies it, dont flag it for being less specific Fr� r T) �lenr� r� r� r� rG r� �search�group)rW r� r� Zinsp�meta�reg�mi�mmr> r> r? �_column_args_match s$ � �� � z DefaultImpl._column_args_match)�inspector_column�metadata_columnrR c C s8 | � |�}| � |�}| �||�s$dS | �||�s4dS dS )z�Returns True if there ARE differences between the types of the two columns. Takes impl.type_synonyms into account between retrospected and metadata types TF)r� r� r� )rW r� r� r� r� r> r> r? � compare_type# s zDefaultImpl.compare_typec C s ||kS rX r> )rW r� r� Zrendered_metadata_defaultZrendered_inspector_defaultr> r> r? �compare_server_default3 s z"DefaultImpl.compare_server_defaultzSet[UniqueConstraint]z Set[Index])� conn_uniques� conn_indexes�metadata_unique_constraints�metadata_indexesrR c C s d S rX r> )rW r� r� r� r� r> r> r? � correct_for_autogen_constraints< s z+DefaultImpl.correct_for_autogen_constraintsc C s$ |j j|jk r t|d |�|d< d S )N�expr)r: Z_type_affinityr )rW �existingZexisting_transfer�new_typer> r> r? �cast_for_batch_migrateE s �z"DefaultImpl.cast_for_batch_migrater# )r� �is_server_defaultr� rR c K s ddd�}t |j| j|d��S )zdRender a SQL expression that is typically a server default, index expression, etc. TF)rS Z include_table)rL rk )r4 rp rL )rW r� r� r� ru r> r> r? �render_ddl_sql_exprK s �z DefaultImpl.render_ddl_sql_exprr" r )� inspectorrR c C s | j S rX )�autogen_column_reflect)rW r� r> r> r? � _compat_autogen_column_reflectY s z*DefaultImpl._compat_autogen_column_reflectzSet[ForeignKeyConstraint])�conn_fks� metadata_fksrR c C s d S rX r> )rW r� r� r> r> r? � correct_for_autogen_foreignkeys\ s z+DefaultImpl.correct_for_autogen_foreignkeysc C s dS )z�A hook that is attached to the 'column_reflect' event for when a Table is reflected from the database during the autogenerate process. Dialects can elect to modify the information gathered here. Nr> )rW r� rb Z column_infor> r> r? r� c s z"DefaultImpl.autogen_column_reflectc C s dS )z�A hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. Nr> rd r> r> r? �start_migrationsl s z DefaultImpl.start_migrationsc C s | � d| j � dS )z�Emit the string ``BEGIN``, or the backend-specific equivalent, on the current connection context. This is used in offline mode and typically via :meth:`.EnvironmentContext.begin_transaction`. ZBEGINN�r^ rs rd r> r> r? � emit_begint s zDefaultImpl.emit_beginc C s | � d| j � dS )z�Emit the string ``COMMIT``, or the backend-specific equivalent, on the current connection context. This is used in offline mode and typically via :meth:`.EnvironmentContext.begin_transaction`. ZCOMMITNr� rd r> r> r? � emit_commit~ s zDefaultImpl.emit_commitr. r0 zUnion[str, Literal[False]])�type_obj�autogen_contextrR c C s dS �NFr> )rW r� r� r> r> r? � render_type� s zDefaultImpl.render_typec C sv t ||t�� dhd�\}}t|dd �}t|dd �}t|�t|�krL|�d� |�| j� ||t|�prt|�t|�kfS )N�always)�skip)�_compare_identity_optionsr ZIdentity�getattrrJ �add�difference_updaterI )rW Zmetadata_identityZinspector_identity�diffZignoredZ meta_alwaysZinspector_alwaysr> r> r? �_compare_identity_default� s � �z%DefaultImpl._compare_identity_default)�metadata_index�reflected_indexrR c C s4 t |j�}t |j�}||kr,d|� d|� �S d S d S )Nzunique=z to unique=)rJ �unique)rW r� r� Z conn_uniqueZ meta_uniquer> r> r? �_compare_index_unique� s z!DefaultImpl._compare_index_uniquez _autogen._Cz _constraint_sig[_autogen._C])� constraint�optsrR c K s t jd| |f|�S )NT�r Zfrom_constraint�rW r r r> r> r? � _create_metadata_constraint_sig� s z+DefaultImpl._create_metadata_constraint_sigc K s t jd| |f|�S r� r r r> r> r? � _create_reflected_constraint_sig� s z,DefaultImpl._create_reflected_constraint_sigr c C s� g }| � ||�}|r |�|� | �|�}| �|�}t�|�s@t�t�|�sNt�||fD ]}|jrVt� d|j � qV|j |j kr�|�d|j � d|j � �� |r�t � |�S t �� S dS )z�Compare two indexes by comparing the signature generated by ``create_index_sig``. This method returns a ``ComparisonResult``. z�Generating approximate signature for index %s. The dialect implementation should either skip expression indexes or provide a custom implementation.� expression � to N)r r� r r r Z is_index_sigr[ Zhas_expressions�log�warningr� Z column_namesr � Different�Equal)rW r� r� �msgZ unique_msgZm_sigZr_sig�sigr> r> r? �compare_indexes� s0 � � � zDefaultImpl.compare_indexesr, )�metadata_constraint�reflected_constraintrR c C sJ | � |�}| �|�}|j}|j}||kr>t�d|� d|� ��S t�� S dS )a Compare two unique constraints by comparing the two signatures. The arguments are two tuples that contain the unique constraint and the signatures generated by ``create_unique_constraint_sig``. This method returns a ``ComparisonResult``. r r N)r r �unnamedr r r )rW r r Z metadata_tupZ reflected_tupZmeta_sigZconn_sigr> r> r? �compare_unique_constraint� s ���z%DefaultImpl.compare_unique_constraintc C sh dd� |D �}t |�D ]L}|j|kr&qt�|�}|rt�d|j�d| j�dtj� d�� |�|� qd S )Nc S s i | ] }|j |�qS r> )rY )r� r� r> r> r? r� s z8DefaultImpl._skip_functional_indexes.<locals>.<dictcomp>z@autogenerate skipping metadata-specified expression-based index z ; dialect z under SQLAlchemy z6 can't reflect these indexes so they can't be compared) r� rY r Zis_expression_indexr r� r9 Zsqlalchemy_version�discard)rW r� r� Zconn_indexes_by_name�idx�iexr> r> r? �_skip_functional_indexes s �z$DefaultImpl._skip_functional_indexes)�reflected_object�kindrR c C s |� di �S )NZdialect_options)rT )rW r r r> r> r? � adjust_reflected_dialect_options s z,DefaultImpl.adjust_reflected_dialect_options)N) NFNNNNFNNNNN)N)N)N)T)F)<r@ rA rB �__doc__r9 rO rs rF �__annotations__rG rI r; � classmethodrZ r^ r` rc �propertyre r Z immutabledictrx rt r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r> r> r> r? rC O s� �0 � �.t � � � �((! � + rC )� metaclassc @ s. e Zd ZU ded< ded< ded< ded< dS ) r� r4 r� z List[str]r� r� zDict[str, str]r� N)r@ rA rB r r> r> r> r? r� s r� z-Union[schema.Identity, schema.Sequence, None]z'Union[schema.Identity, schema.Sequence]zSet[str])� metadata_io� inspector_io� default_ior� c s� t �| �}t �|�}t� � t� �ddddd�� ��fdd� }|||t �|�t|��|�� t jr�|t| di �t|di �|jt|di �� � �fS )Nrf z Iterable[str])� meta_dict� insp_dict� default_dict�attrsc s\ t |����D ]H}| �|�}|�|�}||kr|�|�}||krL��|� q� �|� qd S rX )�set� differencerT r� )r# r$ r% r& �attrZ meta_valueZ insp_value� default_value�r� Z ignored_attrr� r> r? � check_dicts1 s z._compare_identity_options.<locals>.check_dicts�dialect_kwargs)r Z_get_identity_options_dictr' �unionZidentity_has_dialect_kwargsr� r- )r r! r" r� Zmeta_dZinsp_dr, r> r+ r? r� " s& � �r� )H� __future__r �loggingr� �typingr r r r r r r r r r r r r r Z sqlalchemyr r r � r r r r r r r r Zsqlalchemy.enginer r Zsqlalchemy.engine.cursorr! Z sqlalchemy.engine.reflectionr" Zsqlalchemy.sqlr# r$ Zsqlalchemy.sql.elementsr% r& Zsqlalchemy.sql.schemar' r( r) r* r+ r, Zsqlalchemy.sql.selectabler- Zsqlalchemy.sql.type_apir. r/ Zautogenerate.apir0 Zoperations.batchr1 r2 � getLoggerr@ r r: r3 r< r rC r� r� r> r> r> r? �<module> sx Q