Edit File: impl.cpython-37.pyc
B �M!V5/ � @ s� d dl mZmZ d dl mZ ddlmZmZmZ ddl m Z ddl m Z ddl m Z G d d � d e �Zi ZG d d � d ee��Zd d� Zdd� ZejeejeiZdS )� )�schema�text)�types� )� string_types� text_type�with_metaclass)� sqla_compat)�util� )�basec @ s e Zd Zdd� ZdS )�ImplMetac C s( t �| |||�}d|kr$| t|d < |S )N� __dialect__)�type�__init__�_impls)�cls� classname�basesZdict_Znewtype� r �A/opt/alt/python37/lib/python3.7/site-packages/alembic/ddl/impl.pyr s zImplMeta.__init__N)�__name__� __module__� __qualname__r r r r r r s r c @ s e Zd ZdZdZdZdZdd� Zedd� �Z d d � Z d d � Z d d� Z e dd� �Zdde�� fdd�Zd@dd�ZdAdd�ZdBdd�ZdCdd �Zd d � Zd d � ZdDd!d"�Zd#d$� Zd%d&� Zd'd(� Zd)d*� Z dEd,d-�Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z!d6d7� Z"d8d9� Z#d:d;� Z$d<d=� Z%d>d?� Z&dS )F� 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�;c C sv || _ || _|| _|�dd�| _| jr<tjs<t�d� d| _|| _i | _ || _ |d k r\|| _ | jrr| jsrt� d��d S )N� literal_bindsFz4'literal_binds' flag not supported in SQLAlchemy 0.7z3Can't use literal_binds setting without as_sql mode) �dialect� connection�as_sql�getr r �sqla_08�warn� output_buffer�memo� context_opts�transactional_ddlZ CommandError)�selfr r r r' r$ r& r r r r ) s zDefaultImpl.__init__c C s t |j S )N)r �name)r r r r r �get_by_dialect? s zDefaultImpl.get_by_dialectc C s" | j �t|d �� | j �� d S )Nz )r$ �writer �flush)r( r r r r � static_outputC s zDefaultImpl.static_outputc 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 )r( Zbatch_opr r r �requires_recreate_in_batchG s z&DefaultImpl.requires_recreate_in_batchc 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 )r( �tabler r r �prep_table_for_batchR s z DefaultImpl.prep_table_for_batchc C s | j S )N)r )r( r r r �bind[ s zDefaultImpl.bindNr c C s� t |t�rt|�}| jr�|s |r(td��| jrJt |tj�sJtddid�}ni }| � t |j f d| j i|���� dd��� | j � n(| j}|r�|jf |�}|j|f|�|�S d S )Nz+Execution arguments not allowed with as_sqlr T)Zcompile_kwargsr � z )� isinstancer r r � Exceptionr r Z DDLElement�dictr- r �compiler �replace�strip�command_terminatorr �execution_options�execute)r( Z constructr: � multiparams�paramsZ compile_kwZconnr r r �_exec_ s zDefaultImpl._execc C s | � ||� d S )N)r> )r( Zsqlr: r r r r; x s zDefaultImpl.executec C s� |d k s| d k rt �d� |d k r@| �tj||||| | | d�� |dk rf| �tj||||| | | d�� |d k r�| �tj||||| | | d�� |d k r�| �tj||||| | | d�� d S )NzBautoincrement and existing_autoincrement only make sense for MySQL)r � existing_type�existing_server_default�existing_nullableF)r r# r> r ZColumnNullableZ ColumnDefaultZ ColumnTypeZ ColumnName) r( � table_nameZ column_nameZnullableZserver_defaultr) �type_r Z autoincrementr? r@ rA Zexisting_autoincrementr r r � alter_column{ s: zDefaultImpl.alter_columnc C s | � tj|||d�� d S )N)r )r> r Z AddColumn)r( rB �columnr r r r � add_column� s zDefaultImpl.add_columnc K s | � tj|||d�� d S )N)r )r> r Z DropColumn)r( rB rE r �kwr r r � drop_column� s zDefaultImpl.drop_columnc C s( |j d ks|� | �r$| �t�|�� d S )N)Z _create_ruler> r Z AddConstraint)r( �constr r r �add_constraint� s zDefaultImpl.add_constraintc C s | � t�|�� d S )N)r> r ZDropConstraint)r( rI r r r �drop_constraint� s zDefaultImpl.drop_constraintc C s | � tj|||d�� d S )N)r )r> r Z RenameTable)r( Zold_table_nameZnew_table_namer r r r � rename_table� s zDefaultImpl.rename_tablec C sl t jr |jj|| jd| d� | �t�|�� t jrH|jj|| jd| d� x |j D ]}| �t� |�� qPW d S )NF)Z checkfirstZ _ddl_runner) r Zsqla_07�dispatchZ before_creater r> r Z CreateTableZ after_create�indexes� CreateIndex)r( r/ �indexr r r � create_table� s zDefaultImpl.create_tablec C s | � t�|�� d S )N)r> r Z DropTable)r( r/ r r r � drop_table� s zDefaultImpl.drop_tablec C s | � t�|�� d S )N)r> r rO )r( rP r r r � create_index� s zDefaultImpl.create_indexc C s | � t�|�� d S )N)r> r Z DropIndex)r( rP r r r � drop_index� s zDefaultImpl.drop_indexTc s� t |t�std��n|r.t |d t�s.td��| jrtx�|D ]4}| �� jdd�jf t� fdd�|�� D ���� q:W nXt � d�s�d � _ |r�|r�| j� jdd�|d � n(x&|D ] }| �� jdd�jf |�� q�W d S ) Nz List expectedr z List of dictionaries expectedT)�inlinec 3 s> | ]6\}}|t |tj�s0tj||� j| jd �n|fV qdS ))rC N)r3 r Z_literal_bindparam�cr )�.0�k�v)r/ r r � <genexpr>� s z*DefaultImpl.bulk_insert.<locals>.<genexpr>�_autoincrement_column)r< ) r3 �list� TypeErrorr5 r r> �insert�values�items�hasattrr[ )r( r/ ZrowsZ multiinsert�rowr )r/ r � bulk_insert� s zDefaultImpl.bulk_insertc C sz |j }|j }|�| j�}|j�dd � t|d�rL|�| j|�}|d k rL| S |�|�rrt� |j d �}|op|||�S dS d S )N�_type_affinity�compare_against_backendT) r Z dialect_implr �__dict__�popra re Z_compare_type_affinity�_type_comparatorsr! rd )r( �inspector_column�metadata_columnZ conn_typeZ metadata_typeZ metadata_implZ comparisonZ comparatorr r r � compare_type� s zDefaultImpl.compare_typec C s ||kS )Nr )r( ri rj Zrendered_metadata_defaultZrendered_inspector_defaultr r r �compare_server_default s z"DefaultImpl.compare_server_defaultc C s d S )Nr )r( Z conn_uniquesZ conn_indexesZmetadata_unique_constraintsZmetadata_indexesr r r � correct_for_autogen_constraints s z+DefaultImpl.correct_for_autogen_constraintsc s" t jr �jS � �fdd�}|S d S )Nc s �� � | |�S )N)�autogen_column_reflect)r/ � column_info)� inspectorr( r r �adapt s z9DefaultImpl._compat_autogen_column_reflect.<locals>.adapt)r r" rn )r( rp rq r )rp r( r � _compat_autogen_column_reflect s z*DefaultImpl._compat_autogen_column_reflectc C s d S )Nr )r( Zconn_fksZ metadata_fksr 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 )r( rp r/ ro r r r rn 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 )r( r r r �start_migrations' 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- r9 )r( r r r � emit_begin/ 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`. ZCOMMITN)r- r9 )r( r r r � emit_commit9 s zDefaultImpl.emit_commit)N) NFNNNNNNNN)N)N)N)T)'r r r �__doc__r r' r9 r � classmethodr* r- r. r0 �propertyr1 r Z immutabledictr> r; rD rF rH rJ rK rL rQ rR rS rT rc rk rl rm rr rs rn rt ru rv r r r r r sV & r c C s | j d k o| j |j kS )N)�length)�t1�t2r r r �_string_compareD s r} c C s, | j d k r| j |j kp*| jd k o*| j|jkS )N)Z precisionZscale)r{ r| r r r �_numeric_compareJ s r~ N)Z sqlalchemyr r r ZsqltypesZ util.compatr r r r r � r r r r r r} r~ �StringZNumericrh r r r r �<module> s /