Current File : //opt/alt/python37/lib/python3.7/site-packages/alembic/ddl/__pycache__/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)�clsZ classname�basesZdict_Znewtype� r �/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_bufferZmemo� 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 constructr8 � multiparamsZparamsZ
compile_kwZconnr r r �_exec_ s
zDefaultImpl._execc C s | � ||� d S )N)r; )r& Zsqlr8 r r r r9 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= r> Zexisting_autoincrementr r r � alter_column{ s:
zDefaultImpl.alter_columnc C s | � tj|||d�� d S )N)r )r; r Z AddColumn)r& r? �columnr r r r �
add_column� s zDefaultImpl.add_columnc K s | � tj|||d�� d S )N)r )r; r Z
DropColumn)r& r? rB 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& rF 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_createZindexes� 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 rK )r& rL r r r � create_index� s zDefaultImpl.create_indexc C s | � t�|�� d S )N)r; r Z DropIndex)r& rL 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)Zinlinec 3 s> | ]6\}}|t |tj�s0tj||� j| jd �n|fV qdS ))r@ N)r1 r Z_literal_bindparam�cr )�.0�k�v)r- r r � <genexpr>� s z*DefaultImpl.bulk_insert.<locals>.<genexpr>�_autoincrement_column)r: ) r1 �list� TypeErrorr3 r r; �insert�values�items�hasattrrV )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__�popr\ r` Z_compare_type_affinity�_type_comparatorsr r_ )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& rd re 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! ri )r& rk rl r )rk 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& rk r- rj r r r ri 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+ r7 )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+ r7 )r& r r r � emit_commit9 s zDefaultImpl.emit_commit)N)
NFNNNNNNNN)N)N)N)T)'r r r �__doc__r r% r7 r � classmethodr( r+ r, r. �propertyr/ r
Z
immutabledictr; r9 rA rC rE rG rH rI rM rN rO rP r^ rf rg rh rm rn ri ro rp rq 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
rx c C s, | j d k r| j |j kp*| jd k o*| j|jkS )N)Z precisionZscale)rv rw r r r �_numeric_compareJ s
ry N)Z
sqlalchemyr r r ZsqltypesZ util.compatr r r r
r � r r r
r r rx ry �StringZNumericrc r r r r �<module> s /
Mini Shell