관리-도구
편집 파일: render.cpython-38.pyc
U 0?�f� � @ st d dl mZ d dlmZ d dlZd 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 ddlmZ ddlm Z ddlm Z e �r�d dlm Z d dl m Z d dlm!Z! d dlm"Z" d dl#m$Z$ d dl#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%l0m1Z1 d d&l2m3Z3 d d'l4m5Z5 d d(l6m7Z7 d d)l6m8Z8 d d*l9m:Z: d d+l9m;Z; d,Z<d-d.d/d0�d1d2�Z=d3d3d4�d5d6�Z>d-d7d8d9d:�d;d<�Z?e�@� ZAZBd=d-d3d>�d?d@�ZCd-dAdBdC�dDdE�ZDd-dAd3dC�dFdG�ZEeB�Fe j8�d-dHdBdC�dIdJ��ZGeB�Fe jH�d-dKd3dC�dLdM��ZIeB�Fe jJ�d-dNd3dC�dOdP��ZKeB�Fe jL�d-dQd3dC�dRdS��ZMeB�Fe jN�d-dTd3dC�dUdV��ZOd-dWdXdY�dZd[�ZPeB�Fe jQ�d-d\d3dC�d]d^��ZReB�Fe jS�d-d_d3dC�d`da��ZTeB�Fe jU�d-dbdBdC�dcdd��ZVeB�Fe jW�d-ded3dC�dfdg��ZXeB�Fe jY�dhdi� �ZZeB�Fe j[�djdk� �Z\eB�Fe j]�d-dld3dC�dmdn��Z^eB�Fe j_�d-dod3dC�dpdq��Z`eB�Fe ja�d-drd3dC�dsdt��ZbeB�Fe jc�d-dud3dC�dvdw��ZdG dxdy� dy�Zedzd{d|�d}d~�Zfdd�d�d��d�d-d�d�d�d3d��d�d��Zgd�d-dBd��d�d��Zhd�d-d�d3d��d�d��Zid�d�� Zjd-d3d��d�d��Zkd-d3d��d�d��Zld3d�d-d�d��d�d��Zmd�d-d3d��d�d��Znd�d�d��d�d��Zod�d�d-d�d{d��d�d��Zpd�d-d3d��d�d��Zqd�d-d3d��d�d��Zrd�d�d-d�d3d��d�d��Zsd�d-d3d��d�d��Ztd�d-d3d��d�d��Zud�d�d-d3d3d{d�d��d�d��Zve�@� Zwd�d-d�d{d��d�d��Zxew�Fej,�d�d-d�d{d��d�d�Zyd�d{d�d3dŜd�dDŽZzd�d�d9dʜd�d̄Z{ew�Fej)�d�d-d�d{d��d�d΄�Z|ew�Fej-�d�d-d�d3d��d�dЄ�Z}ew�Fej$�d�d-d�d{d��d�dӄ�Z~eB�Fe j�d-d�d3dC�d�dք�Z�eA��� ZBdS )�� )� annotations)�StringION)�Any)�cast)�Dict)�List)�Optional)�Tuple)� TYPE_CHECKING)�Union)� PythonPrinter)�schema)�sql)�types)�conv)� quoted_name� )�util)�ops)� sqla_compat)�Literal)� DialectKWArgs)� ColumnElement)� TextClause)�CheckConstraint)�Column)� Constraint)� FetchedValue)� ForeignKey)�ForeignKeyConstraint)�Index)�MetaData)�PrimaryKeyConstraint)�UniqueConstraint)�ARRAY)� TypeEngine)�AutogenContext)�Config)�MigrationScript)�ModifyTableOps)�Computed)�Identity� r& zsqla_compat._ConstraintNamez*Optional[Union[quoted_name, str, _f_name]])�autogen_context�name�returnc C s& t |t�rtt| �|�S t�|�S d S �N)� isinstancer �_f_name� _alembic_autogenerate_prefixr Zconstraint_name_or_none)r- r. � r4 �I/opt/hc_python/lib/python3.8/site-packages/alembic/autogenerate/render.py�_render_gen_name: s r6 �str)�textr/ c C s4 t �dt j��d| ��� } t �dt j��d| �} | S )N�^� z +$� )�re�compile�M�sub�strip)r8 r4 r4 r5 �_indentD s rA r( z Dict[str, Union[str, Config]]�None)r- �migration_script� template_argsr/ c C s\ | j }t|j|j�D ]0\}}tt|| ��||j< tt|| ��||j< qd�t |��|d< d S )N� �imports) rF �zipZupgrade_ops_listZdowngrade_ops_listrA �_render_cmd_bodyZ upgrade_tokenZdowngrade_token�join�sorted)r- rC rD rF Z upgrade_opsZ downgrade_opsr4 r4 r5 � _render_python_into_templatevarsJ s � � � rK zops.OpContainer)� op_containerr- r/ c C sr t � }t|�}|�d� d}| jD ].}t||�}|p:t|�}|D ]}|�|� q@q"|s`|�d� |�d� |�� S )Nz=# ### commands auto generated by Alembic - please adjust! ###F�passz # ### end Alembic commands ###)r r Z writeliner � render_op�bool�getvalue)rL r- �buf�printerZ has_lines�op�lines�liner4 r4 r5 rH ` s � rH zops.MigrateOperationz List[str])r- rS r/ c C s t �|�}t�|| |��}|S r0 )� renderers�dispatchr Zto_list)r- rS �rendererrT r4 r4 r5 rN { s rN c C s d� t| |��S )NrE )rI rN �r- rS r4 r4 r5 �render_op_text� s rZ r) c C s� | j }|�dd�}|jr�g }|rv| �� �F |�d|j|jf � |jD ]}t| |�}|�|� qF|�d� W 5 Q R X n |jD ]}t| |�}|�|� q||S g S d S )N�render_as_batchFz5with op.batch_alter_table(%r, schema=%r) as batch_op:r; ) �opts�getr Z _within_batch�append� table_namer rN �extend)r- rS r\ r[ rT Zt_opZt_linesr4 r4 r5 �_render_modify_table� s( �� ra zops.CreateTableCommentOpc C sh | j r d}nd}|jt| �|j|jd k r2d|j nd |jd k rHd|j nd |jd k r^d|j nd dd�S )NzV{prefix}create_table_comment( {indent}{comment}, {indent}existing_comment={existing} )z�{prefix}create_table_comment( {indent}'{tname}', {indent}{comment}, {indent}existing_comment={existing}, {indent}schema={schema} )�%r�'%s'r: )�prefix�tname�comment�existingr �indent)� _has_batch�formatr3 r_ rf �existing_commentr �r- rS Ztemplr4 r4 r5 � _render_create_table_comment� s ��� �rm zops.DropTableCommentOpc C sR | j r d}nd}|jt| �|j|jd k r2d|j nd |jd k rHd|j nd dd�S )NzA{prefix}drop_table_comment( {indent}existing_comment={existing} )zm{prefix}drop_table_comment( {indent}'{tname}', {indent}existing_comment={existing}, {indent}schema={schema} )rb rc r: )rd re rg r rh )ri rj r3 r_ rk r rl r4 r4 r5 �_render_drop_table_comment� s ��� �rn zops.CreateTableOpc s< �� � }dd� � fdd�|jD �D �tdd� � �fdd�|jD �D �� }t|�tkrjdd�|� d }n d�|�}d t�j�t � �|d � }�j r�|d t�j � 7 }|j }|r�|d t|� 7 }|j }|r�|d |��7 }t�j �D ]"}|d|�dd��j | f 7 }q�|j�r0d�dd� |jD ��} |d| 7 }|d7 }|S )Nc S s g | ] }|r|�qS r4 r4 ��.0�colr4 r4 r5 � <listcomp>� s �z _add_table.<locals>.<listcomp>c s g | ]}t |� ��qS r4 )�_render_columnro �r- r4 r5 rr � s c S s g | ]}|d k r|�qS r0 r4 )rp Zrconsr4 r4 r5 rr � s �c s g | ]}t |� �j��qS r4 )�_render_constraintZ_namespace_metadata)rp ZconsrY r4 r5 rr � s � �z*[z, �]z.%(prefix)screate_table(%(tablename)r, %(args)s)Z tablenamerd �argsz , schema=%rz , comment=%rz, info=z, %s=%r� �_�, c s s | ]}d | V qdS )rc Nr4 )rp �pr4 r4 r5 � <genexpr> s z _add_table.<locals>.<genexpr>z, prefixes=[%s]z ))Zto_table�columnsrJ � constraints�len�MAX_PYTHON_ARGSrI �_identr_ r3 r rf �info�kw�replace� _prefixes) r- rS �tablerw �args_strr8 rf r� �k�prefixesr4 rY r5 � _add_table� sJ �� ���� � r� zops.DropTableOpc C s< dt | �t|j�d� }|jr0|dt|j� 7 }|d7 }|S )Nz %(prefix)sdrop_table(%(tname)r)rd re � , schema=%r�))r3 r� r_ r )r- rS r8 r4 r4 r5 � _drop_table s �r� r z list[str])r- �itemr/ c s � fdd�|j �� D �S )Nc s$ g | ] \}}|� d t |� �� ��qS )�=��_render_potential_expr�rp �key�valrt r4 r5 rr - s �z0_render_dialect_kwargs_items.<locals>.<listcomp>)�dialect_kwargs�items)r- r� r4 rt r5 � _render_dialect_kwargs_items* s �r� zops.CreateIndexOpc C s� |� � }| j}|rd}nd}|jd k s*t�t| |�}|t| �t| |j�t|jj�d� t || ��|j pfd|jj r~dt|jj � nd|r�dd� |� ndd� }|S )NzL%(prefix)screate_index(%(name)r, [%(columns)s], unique=%(unique)r%(kwargs)s)za%(prefix)screate_index(%(name)r, %(table)r, [%(columns)s], unique=%(unique)r%(schema)s%(kwargs)s)rz Fr� r; )rd r. r� r} �uniquer �kwargs) �to_indexri r� �AssertionErrorr� r3 r6 r. r� rI � _get_index_rendered_expressionsr� r �r- rS �index� has_batch�tmplr\ r8 r4 r4 r5 � _add_index3 s. �� ���r� zops.DropIndexOpc C sv |� � }| j}|rd}nd}t| |�}|t| �t| |j�t|j�|jrTdt|j� nd|rhdd� |� ndd� }|S )Nz(%(prefix)sdrop_index(%(name)r%(kwargs)s)zM%(prefix)sdrop_index(%(name)r, table_name=%(table_name)r%(schema)s%(kwargs)s)r� r; rz )rd r. r_ r r� ) r� ri r� r3 r6 Z index_namer� r_ r rI r� r4 r4 r5 � _drop_indexY s � �r� z ops.CreateUniqueConstraintOpc C s t |�� | d�gS )NT)�_uq_constraintZ to_constraintrY r4 r4 r5 �_add_unique_constraintq s r� zops.CreateForeignKeyOpc C s� t t| |j��g}| js,|�t t|j��� |�t t|j��t dd� |j D ��t dd� |j D ��g� dddddd d g}| js�|� d d � |D ]2}||j kr�|j | }|d k r�|�d ||f � q�dt | �d�|�d� S )Nc S s g | ] }t |��qS r4 �r� ro r4 r4 r5 rr � s z&_add_fk_constraint.<locals>.<listcomp>c S s g | ] }t |��qS r4 r� ro r4 r4 r5 rr � s Zreferent_schema�onupdate�ondelete� initially� deferrable� use_alter�matchr Z source_schema�%s=%rz&%(prefix)screate_foreign_key(%(args)s)rz �rd rw )�reprr6 �constraint_nameri r^ r� Z source_tabler` Zreferent_tableZ local_colsZ remote_cols�insertr� r3 rI )r- rS rw r� r� �valuer4 r4 r5 �_add_fk_constraintx s8 ��� �r� c C s t � �d S r0 ��NotImplementedError�� constraintr- r4 r4 r5 �_add_pk_constraint� s r� c C s t � �d S r0 r� r� r4 r4 r5 �_add_check_constraint� s r� zops.DropConstraintOpc C s� t | �}t| |j�}|jr$t|j�nd }|jr8t|j�nd }g }|�t|�� | js�|�tt|j ��� |d k r�|�d|��� |d k r�|�d|��� |� dd� |�� d�S )Nzschema=ztype_=zdrop_constraint(rz r� ) r3 r6 r� r r� Zconstraint_typer^ r� ri r_ rI )r- rS rd r. r �type_Z params_strsr4 r4 r5 �_drop_constraint� s r� zops.AddColumnOpc C sZ |j |j|j }}}| jr"d}nd}|r2|d7 }|d7 }|t| �|t|| �|d� }|S )Nz %(prefix)sadd_column(%(column)s)z*%(prefix)sadd_column(%(tname)r, %(column)s�, schema=%(schema)rr� )rd re �columnr )r r_ r� ri r3 rs )r- rS r re r� �templater8 r4 r4 r5 � _add_column� s �r� zops.DropColumnOpc C s` |j |j|j }}}| jr"d}nd}|r2|d7 }|d7 }|t| �t|�t|�t|�d� }|S )Nz %(prefix)sdrop_column(%(cname)r)z*%(prefix)sdrop_column(%(tname)r, %(cname)rr� r� )rd re �cnamer )r r_ � column_nameri r3 r� )r- rS r re r� r� r8 r4 r4 r5 � _drop_column� s �r� zops.AlterColumnOpc C s� |j }|j}|j}|j}|j}|j}|j�dd �}|j} |j } |j } |j } |j } d}| j r`d}nd}|t| �||d� }| d k r�|d|t| | �f 7 }|dk r�t|| �}|d||f 7 }|d k r�|d |t|| �f 7 }|d k r�|d ||f 7 }|dk �r|d ||f 7 }| d k �r"|d || f 7 }|d k�rF| d k �rF|d || f 7 }|d k �r`|d||f 7 }|dk�r�| �r�t| | �}|d||f 7 }| �r�| j �s�|d|| f 7 }|d7 }|S )N� autoincrementz z %(prefix)salter_column(%(cname)rz+%(prefix)salter_column(%(tname)r, %(cname)r)rd re r� z, %sexisting_type=%sFz, %sserver_default=%sz , %stype_=%sz, %snullable=%rz, %scomment=%rz, %sexisting_comment=%rz, %sexisting_nullable=%rz, %sautoincrement=%rz , %sexisting_server_default=%sz , %sschema=%rr� )r_ r� Zmodify_server_defaultZ modify_typeZmodify_nullableZmodify_commentr� r] � existing_type�existing_nullablerk �existing_server_defaultr ri r3 � _repr_type�_render_server_default)r- rS re r� �server_defaultr� �nullablerf r� r� r� rk r� r rh r� r8 �renderedr4 r4 r5 � _alter_column� sf �� �r� c @ s, e Zd Zdddd�dd�Zdd�dd �Zd S ) r2 r7 r rB )rd r. r/ c C s || _ || _d S r0 )rd r. )�selfrd r. r4 r4 r5 �__init__$ s z_f_name.__init__)r/ c C s d| j t| j�f S )Nz%sf(%r))rd r� r. )r� r4 r4 r5 �__repr__( s z_f_name.__repr__N)�__name__� __module__� __qualname__r� r� r4 r4 r4 r5 r2 # s r2 z!Optional[Union[quoted_name, str]]z Optional[str])r. r/ c C s0 | dkr | S t | t�r t| �S t | t�r,| S dS )z�produce a __repr__() object for a string identifier that may use quoted_name() in SQLAlchemy 0.9 and greater. The issue worked around here is that quoted_name() doesn't have very good repr() behavior by itself when unicode is involved. N)r1 r r7 )r. r4 r4 r5 r� , s r� TF)� wrap_in_text�is_server_default�is_indexr rO )r� r- r� r� r� r/ c C sH t | tj�r<|rd}nd}|t|�|jjj| ||d�d� S t| �S d S )Nz%(prefix)stext(%(sql)r)z%(sql)r)r� r� )rd r )r1 r Z ClauseElement� _sqlalchemy_autogenerate_prefix�migration_context�implZrender_ddl_sql_exprr� )r� r- r� r� r� r� r4 r4 r5 r� < s ��r� r )�idxr- r/ c s � fdd�| j D �S )Nc s: g | ]2}t |tj�r(ttt|d d���n t|� dd��qS )r. NT)r� )r1 � sa_schemar r� r� �getattrr� )rp �exprt r4 r5 rr X s � �z3_get_index_rendered_expressions.<locals>.<listcomp>)Z expressions)r� r- r4 rt r5 r� U s �r� r# )r� r- �alterr/ c C sZ g }|j }| jr$|�dt| j�f� | jr>|�dt| j�f� |sd|rd| jjrd|�dt| jj�f� |s�| jr�|�dt || j�f� t || �}|�rt t || j��g}|s�|t t| jj��g7 }|�t dd� | j D ��� |� dd� |D �� |� |� dt|�d �|�d � S d d� | j D �}|� d d� |D �� |� |� d t|�d �|�d � S d S )Nr� r� r r. c S s g | ]}t |j��qS r4 )r� r. ro r4 r4 r5 rr { s z"_uq_constraint.<locals>.<listcomp>c S s g | ]\}}d ||f �qS �r� r4 �rp r� �vr4 r4 r5 rr | s z,%(prefix)screate_unique_constraint(%(args)s)rz r� c S s g | ]}t t|j���qS r4 )r� r� r. ro r4 r4 r5 rr � s c S s g | ]\}}d ||f �qS r� r4 r� r4 r4 r5 rr � s z$%(prefix)sUniqueConstraint(%(args)s))ri r� r^ r7 r� r� r r� r. r6 r� r� r} r` r3 rI r� )r� r- r� r\ r� Zdialect_optionsrw r4 r4 r5 r� b s> � � �r� c C s$ | j d }|d kr d|j S |S d S )NZuser_module_prefixz%s.)r\ r� )r- �targetrd r4 r4 r5 �_user_autogenerate_prefix� s r� )r- r/ c C s | j d p dS )NZsqlalchemy_module_prefixr; �r\ rt r4 r4 r5 r� � s r� c C s | j r dS | jd pdS d S )Nz batch_op.Zalembic_module_prefixr; )ri r\ rt r4 r4 r5 r3 � s r3 zUnion[str, Literal[False]])r� �object_r- r/ c C s4 d|j kr0|j d }|r0|| ||�}|dk r0|S dS )NZ render_itemFr� )r� r� r- �renderr� r4 r4 r5 �_user_defined_render� s r� z Column[Any])r� r- r/ c s< t d| � �}|dk r|S g }g }| jrZt| j� �}|rZt| j�rL|�|� n|�d|f� | jd k r�| jtjkr�|�d| jf� | jd k r�|�d| jf� | j r�|�d| j f� | j }|r�|�dd| f� d t � �t | j �t| j� �|�rd �d d � |D ��d nd d �dd � |D �� fdd �t�| ��� D � �d� S )Nr� Fr� r� r� �systemrf rb z8%(prefix)sColumn(%(name)r, %(type)s, %(args)s%(kwargs)s)rz c S s g | ] }t |��qS r4 )r7 )rp �argr4 r4 r5 rr � s z"_render_column.<locals>.<listcomp>r; c S s g | ]\}}d ||f �qS ��%s=%sr4 �rp Zkwnamer� r4 r4 r5 rr � s c s" g | ]\}}d |t |� �f �qS r� r� r� rt r4 r5 rr � s � �)rd r. �typerw r� )r� r� r� �*_should_render_server_default_positionallyr^ r� r ZAUTOINCREMENT_DEFAULTr� r� rf r� r� r. r� r� rI Z_column_kwargsr� )r� r- r� rw r\ rf r4 rt r5 rs � sR �� � � ����rs )r� r/ c C s t �| �pt �| �S r0 )r �_server_default_is_computed�_server_default_is_identity)r� r4 r4 r5 r� � s ��r� zBOptional[Union[FetchedValue, str, TextClause, ColumnElement[Any]]])�defaultr- �repr_r/ c C s� t d| |�}|dk r|S t�| �r2ttd| �|�S t�| �rLttd| �|�S t| tj �r|t| j t �rl| j } nt | j |dd�S t| t �r�|r�t t�dd| ��} tt | �S ) Nr� Fr* r+ T)r� z^'|'$r; )r� r r� �_render_computedr r� �_render_identityr1 r� Z DefaultClauser� r7 r� r� r< r? )r� r- r� r� r4 r4 r5 r� � s$ �r� r* )�computedr- r/ c C sP t | j|dd�}i }| jd k r(| j|d< dt|�|d�dd� |�� D ��d� S ) NF�r� � persistedz(%(prefix)sComputed(%(text)s, %(kwargs)s)rz c s s | ]}d | V qdS �r� Nr4 �rp �pairr4 r4 r5 r| s z#_render_computed.<locals>.<genexpr>)rd r8 r� )r� �sqltextr� r� rI r� )r� r- r8 r� r4 r4 r5 r� s � �r� r+ )�identityr- r/ c C s4 t j| dd�}dt|�d�dd� |�� D ��d� S )NT)r� z %(prefix)sIdentity(%(kwargs)s)rz c s s | ]}d | V qdS r� r4 r� r4 r4 r5 r| ! s z#_render_identity.<locals>.<genexpr>)rd r� )r Z_get_identity_options_dictr� rI r� )r� r- r� r4 r4 r5 r� s ��r� r% )r� r- �_skip_variantsr/ c C s* t d| |�}|dk r|S t|jd�r6|jj�| |�}nd }t| �j}|j}|�d�r�t � d|�}|d k slt �|� d�}|d k r�|� d| � |r�|S d|| f S n�|r�|S |s�t�| �r�t| |�S |�d ��rd | j t� k� r�t� d | j } | | |�S t|�} d | | f S nt|| �} d | | f S d S ) Nr� Fr� �sqlalchemy.dialects�sqlalchemy\.dialects\.(\w+)� z"from sqlalchemy.dialects import %sz%s.%rz sqlalchemy.z_render_%s_typez%s%r)r� �hasattrr� r� Z render_typer� r� rF � startswithr< r� r� �group�addr Z_type_has_variants�_render_Variant_typeZ__visit_name__�globalsr� r� ) r� r- r� r� Zimpl_rt�modrF r� �dname�fnrd r4 r4 r5 r� % s@ � r� r$ )r� r- r/ c C s t tt| |dd��S )NZ item_typez(.+?\())r r7 �_render_type_w_subtype)r� r- r4 r4 r5 �_render_ARRAY_typeQ s ��r c C sd t �| �\}}t||dd�}|d k r,|dk s0t�t|�D ]&}|| }|dt||dd�|f 7 }q8|S )NT)r� Fz.with_variant(%s, %r))r Z_get_variant_mappingr� r� rJ )r� r- Z base_typeZvariant_mapping�base�dialect�typr4 r4 r5 r Z s � r z$Union[Optional[str], Literal[False]])r� r- �attrname�regexprd r/ c C s� t | �}t| |d �}|d kr dS t |�}t�dd|�}tt| |�|�}t�|| d| |�} |rld|| f S t| �j} | �d�r�t�d| �} | d k s�t �| � d�} d | | f S | �d �r�t |�}d|| f S d S d S ) NFz([\(\)])z\\\1z\1%sz%s%sr� r� r� �%s.%s� sqlalchemy) r� r� r< r? r� r� r� r r� r� r r� ) r� r- r r rd Z outer_reprZ inner_typeZ inner_reprZsub_typeZ outer_typer r� r r4 r4 r5 r i s( r r zOptional[MetaData])r� r- �namespace_metadatar/ c C sJ zt �| �}W n* tk r8 t�d| � d| Y S X || ||�S d S )Nz(No renderer is established for object %rz[Unknown Python object %r])�_constraint_renderersrW � ValueErrorr �warn)r� r- r rX r4 r4 r5 ru � s ru r" c C sx t d| |�}|dk r|S | js"d S g }| jrF|�dtt|| j��f� dt|�d�dd� | jD �dd� |D � �d � S ) NZ primary_keyFr. z(%(prefix)sPrimaryKeyConstraint(%(args)s)rz c S s g | ]}t |j��qS r4 )r� r. )rp �cr4 r4 r5 rr � s z'_render_primary_key.<locals>.<listcomp>c S s g | ]\}}d ||f �qS r� r4 r� r4 r4 r5 rr � s r� )r� r} r. r^ r� r6 r� rI �r� r- r r� r\ r4 r4 r5 �_render_primary_key� s$ � ���r r r! )�fk�metadata_schemar r/ c C s� | � � }|�d�}|dd� \}}|dk rDt|�dkrDd||f }nd�|dd� �}| js�| jdk r�| jjdk r�||jkr�|j| j� |�}|dk r�t |j �}d||f }|S )zxImplement a 'safe' version of ForeignKey._get_colspec() that won't fail if the remote table can't be resolved. �.���Nr r r ���) Z _get_colspec�splitr rI Z link_to_name�parentr� Ztablesr r] r� r. ) r r r Zcolspec�tokensre ZcolnameZtable_fullnamerq r4 r4 r5 � _fk_colspec� s$ �� � r r zList[Tuple[str, str]])r� r\ r/ c C s� | j r|�dt| j �f� | jr4|�dt| j�f� | jrN|�dt| j�f� | jrh|�dt| j�f� | jr�|�dt| j�f� | jr�|�dt| j�f� d S )Nr� r� r� r� r� r� )r� r^ r� r� r� r� r� r� )r� r\ r4 r4 r5 �_populate_render_fk_opts� s r! c s� t d| |�}|dk r|S g }| jr<|�dtt|| j��f� t| |� �j� dt|�d�dd� | j D ��d�� �fdd�| j D ��d�d d � |D ��d � S ) NZ foreign_keyFr. zC%(prefix)sForeignKeyConstraint([%(cols)s], [%(refcols)s], %(args)s)rz c s s | ]}t t|jj��V qd S r0 )r� r� r r. �rp �fr4 r4 r5 r| s z&_render_foreign_key.<locals>.<genexpr>c 3 s | ]}t t|� ���V qd S r0 )r� r r" �Zapply_metadata_schemar r4 r5 r| s �c S s g | ]\}}d ||f �qS r� r4 r� r4 r4 r5 rr s z'_render_foreign_key.<locals>.<listcomp>)rd �colsZrefcolsrw ) r� r. r^ r� r6 r! r r� rI �elementsr r4 r$ r5 �_render_foreign_key� s0 � �� ����r' c C s$ t d| |�}|dk r|S t| |d�S )Nr� F)r� r� )r� r- r r� r4 r4 r5 �_render_unique_constraint s r( r c C s� t d| |�}|dk r|S | jr>t| jd�r>t| jjtj�r>d S g }| jrb|�dt t || j��f� dt |�|r�dd� dd� |D �� nd t | j|dd �d � S ) N�checkFr� r. z.%(prefix)sCheckConstraint(%(sqltext)s%(opts)s)rz c s s | ]\}}d ||f V qdS r� r4 r� r4 r4 r5 r| ? s z+_render_check_constraint.<locals>.<genexpr>r; r� )rd r\ r� )r� Z _create_ruler� r1 r� �sqltypesr% r. r^ r� r6 r� rI r� r� r r4 r4 r5 �_render_check_constraint s: � ����� ��r+ zops.ExecuteSQLOpc C s t |jt�std��d|j S )NznAutogenerate rendering of SQL Expression language constructs not supported here; please use a plain SQL stringzop.execute(%r))r1 r� r7 r� rY r4 r4 r5 � _execute_sqlI s �r, )T)F)N)�� __future__r �ior r< �typingr r r r r r r r Z mako.pygenr r r r� r r r* Zsqlalchemy.sql.elementsr r r; r � operationsr r r Zsqlalchemy.sql.baser r r Zsqlalchemy.sql.schemar r r r r r r r! r" r# Zsqlalchemy.sql.sqltypesr$ Zsqlalchemy.sql.type_apir% Zalembic.autogenerate.apir& Zalembic.configr' Zalembic.operations.opsr( r) Zalembic.util.sqla_compatr* r+ r� r6 rA rK Z DispatcherZdefault_renderersrV rH rN rZ Z dispatch_forra ZCreateTableCommentOprm ZDropTableCommentOprn Z CreateTableOpr� Z DropTableOpr� r� Z CreateIndexOpr� Z DropIndexOpr� ZCreateUniqueConstraintOpr� ZCreateForeignKeyOpr� ZCreatePrimaryKeyOpr� ZCreateCheckConstraintOpr� ZDropConstraintOpr� Z AddColumnOpr� Z DropColumnOpr� Z AlterColumnOpr� r2 r� r� r� r� r� r� r3 r� rs r� r� r� r� r� r r r r ru r r r! r' r( r+ Z ExecuteSQLOpr, �branchr4 r4 r4 r5 �<module> s� ! 6 % & < � * 9 � �, �" % % (