Edit File: render.cpython-37.pyc
B �M!VYZ � @ s� d dl mZmZmZ ddlmZ ddlm Z d dl Z ddl m Z ddl mZ d dlmZ dd l mZ d Zyd d lmZ d d � ZW n ek r� dd � ZY nX dd� Zdd� Ze�� ZZdd� Zdd� Z dd� Z e� ej �dd� �Z e� ej!�dd � �Z"e� ej#�d d � �Z$e� ej%�d d � �Z&e� ej'�d!d"� �Z(e� ej)�d#d$� �Z*e� ej+�d%d&� �Z,e� ej-�d'd(� �Z.e� ej/�d)d*� �Z0e� ej1�d+d,� �Z2e� ej3�d-d.� �Z4e� ej5�d/d0� �Z6e� ej7�d1d2� �Z8G d3d4� d4e9�Z:d5d6� Z;dZd8d9�Z<d:d;� Z=d<d=� Z>d>d?� Z?d@dA� Z@dBdC� ZAdDdE� ZBdFdG� ZCd[dHdI�ZDdJdK� ZEe�� ZFdLdM� ZGeF� ejH�dNdO� �ZIdPdQ� ZJdRdS� ZKeF� ejL�dTdU� �ZMeF� ejN�dVdW� �ZOeF� ejP�dXdY� �ZQe�R� ZdS )\� )�schema�types�sql� )�ops)�compatN)� string_types)�util)� PythonPrinter)�StringIO� )�convc C s t |t�rtt| �|�S |S d S )N)� isinstancer �_f_name� _alembic_autogenerate_prefix)�autogen_context�name� r �L/opt/alt/python37/lib/python3.7/site-packages/alembic/autogenerate/render.py�_render_gen_name s r c C s |S )Nr )r r r r r r s c C s4 t �dt j��d| ��� } t �dt j��d| �} | S )N�^z z +$� )�re�compile�M�sub�strip)�textr r r �_indent s r c C s` | j }xBt|j|j�D ]0\}}tt|| ��||j< tt|| ��||j< qW d�t |��|d< d S )N� �imports) r �zipZupgrade_ops_listZdowngrade_ops_listr �_render_cmd_bodyZ upgrade_tokenZdowngrade_token�join�sorted)r Zmigration_scriptZ template_argsr Z upgrade_opsZ downgrade_opsr r r � _render_python_into_templatevars s r% c C sn t � }t|�}|�d� | js*|�d� n2x0| jD ]&}t||�}x|D ]}|�|� qFW q2W |�d� |�� S )Nz;### commands auto generated by Alembic - please adjust! ###�passz ### end Alembic commands ###)r r Z writeliner � render_op�getvalue)Z op_containerr �buf�printer�op�lines�liner r r r" 1 s r" c C s t �|�}t�|| |��}|S )N)� renderers�dispatchr Zto_list)r r+ �rendererr, r r r r' I s r' c C s d� t| |��S )Nr )r# r' )r r+ r r r �render_op_textO s r1 c C s� | j }|�dd�}|jr�g }|rz| �� �J |�d|j|jf � x"|jD ]}t| |�}|�|� qHW |�d� W d Q R X n$x"|jD ]}t| |�}|�|� q�W |S dgS d S )N�render_as_batchFz5with op.batch_alter_table(%r, schema=%r) as batch_op:r r& ) �opts�getr Z _within_batch�append� table_namer r' �extend)r r+ r3 r2 r, Zt_opZt_linesr r r �_render_modify_tableS s$ r8 c s� |� � }dd� � fdd�|jD �D �tdd� � fdd�|jD �D �� }t|�tkrhdd�|� d }n d�|�}d t|j�t � �|d � }|j r�|d t|j � 7 }x0t|j �D ]"}|d |� d d�|j | f 7 }q�W |d7 }|S )Nc S s g | ] }|r|�qS r r )�.0�colr r r � <listcomp>t s z _add_table.<locals>.<listcomp>c s g | ]}t |� ��qS r )�_render_column)r9 r: )r r r r; u s c S s g | ]}|d k r|�qS )Nr )r9 Zrconsr r r r; w s c s g | ]}t |� ��qS r )�_render_constraint)r9 Zcons)r r r r; x s z*[z, �]z.%(prefix)screate_table(%(tablename)r, %(args)s)Z tablename�prefix�argsz , schema=%rz, %s=%r� �_z )) Zto_table�columnsr$ � constraints�len�MAX_PYTHON_ARGSr# �_identr6 r r �kw�replace)r r+ �tabler@ r �kr )r r � _add_tablep s& "rL c C s< dt | �t|j�d� }|jr0|dt|j� 7 }|d7 }|S )Nz %(prefix)sdrop_table(%(tname)r)r? �tnamez , schema=%r�))r rG r6 r )r r+ r r r r � _drop_table� s rO c s� |� � }� j}|rd}nd}|t� �t� |j�t|jj�d�t|� ��|j pNd|jj rfdt|jj � ndt |j �r�dd�� fdd�|j � � D �� 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)z, Fz , schema=%rr c s" g | ]\}}d |t |� �f �qS )z%s=%s)�_render_potential_expr)r9 �key�val)r r r r; � s z _add_index.<locals>.<listcomp>)r? r rJ rC �uniquer �kwargs)Zto_index� _has_batchr r r rG rJ r# � _get_index_rendered_expressionsrS r rE rT �items)r r+ �index� has_batch�tmplr r )r r � _add_index� s 6r[ c C sP | j }|rd}nd}|t| �t| |j�t|j�|jrBdt|j� ndd� }|S )Nz %(prefix)sdrop_index(%(name)r)zC%(prefix)sdrop_index(%(name)r, table_name=%(table_name)r%(schema)s)z , schema=%rr )r? r r6 r )rU r r Z index_namerG r6 r )r r+ rY rZ r r r r � _drop_index� s r\ c C s t |�� | d�gS )NT)�_uq_constraintZ to_constraint)r r+ r r r �_add_unique_constraint� s r^ c C s� t t| |j��g}| js,|�t t|j��� |�t t|j��t dd� |j D ��t dd� |j D ��g� dddddd g}| js�|� d d � x:|D ]2}||j kr�|j | }|d k r�|�d ||f � q�W d t | �d�|�d� S )Nc S s g | ] }t |��qS r )rG )r9 r: r r r r; � s z&_add_fk_constraint.<locals>.<listcomp>c S s g | ] }t |��qS r )rG )r9 r: r r r r; � s Zreferent_schema�onupdate�ondelete� initially� deferrable� use_alterr Z source_schemaz%s=%rz&%(prefix)screate_foreign_key(%(args)s)z, )r? r@ )�reprr �constraint_namerU r5 rG Z source_tabler7 Zreferent_tableZ local_colsZ remote_cols�insertrH r r# )r r+ r@ rT rK �valuer r r �_add_fk_constraint� s, rh c C s t � �d S )N)�NotImplementedError)� constraintr r r r �_add_pk_constraint� s rk c C s t � �d S )N)ri )rj r r r r �_add_check_constraint s rl c C sP | j r d}nd}|t| �t| |j�t|j�|j|jrBdt|j� ndd� }|S )Nz3%(prefix)sdrop_constraint(%(name)r, type_=%(type)r)zO%(prefix)sdrop_constraint(%(name)r, '%(table_name)s'%(schema)s, type_=%(type)r)z , schema='%s'r )r? r r6 �typer )rU r r re rG r6 Zconstraint_typer )r r+ �templater r r r �_drop_constraint s ro c 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)sz, schema=%(schema)rrN )r? rM �columnr )r r6 rp rU r r< )r r+ r rM rp rn r r r r � _add_column s rq c 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)rz, schema=%(schema)rrN )r? rM �cnamer )r r6 � column_namerU r rG )r r+ r rM rs rn r r r r � _drop_column1 s rt c C sH |j }|j}|j}|j}|j}|j}|j}|j} |j} d} | j rFd} 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 } |d kr�|d k r�| d | |f 7 } |dk�r | �r t | | �}| d | |f 7 } | �r<| j �s<| d | | f 7 } | d 7 } | S )Nz z %(prefix)salter_column(%(cname)rz+%(prefix)salter_column(%(tname)r, %(cname)r)r? rM rr z, %sexisting_type=%sFz, %sserver_default=%sz , %stype_=%sz, %snullable=%rz, %sexisting_nullable=%rz , %sexisting_server_default=%sz , %sschema=%rrN ) r6 rs Zmodify_server_defaultZ modify_typeZmodify_nullable� existing_type�existing_nullable�existing_server_defaultr rU r � _repr_type�_render_server_default)r r+ rM rr �server_default�type_�nullableru rv rw r �indentrn r �renderedr r r � _alter_columnG sZ r c @ s e Zd Zdd� Zdd� ZdS )r c C s || _ || _d S )N)r? r )�selfr? r r r r �__init__� s z_f_name.__init__c C s d| j t| j�f S )Nz%sf(%r))r? rG r )r� r r r �__repr__� s z_f_name.__repr__N)�__name__� __module__� __qualname__r� r� r r r r r ~ s r c C sn | dkr | S t jrZt| tjj�rZt jrNy | �d�S tk rJ t � | �S X qjt � | �S nt| t j �rj| 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�ascii) r Zsqla_09r r �elementsZ quoted_nameZpy2k�encode� UnicodeError� text_typer )r r r r rG � s rG Tc C sj t | tj�r^tjr"tddid�}ni }|r0d}nd}|t|�t�| jf d|j i|���d� S t | �S d S )NZ literal_bindsT)Zcompile_kwargsz%(prefix)stext(%(sql)r)z%(sql)r�dialect)r? r ) r r Z ClauseElementr �sqla_08�dict� _sqlalchemy_autogenerate_prefixr� r r� rd )rg r � wrap_in_textZ compile_kwrn r r r rP � s rP c s. t jr� fdd�| jD �S dd� | jD �S d S )Nc s6 g | ].}t |tj�r(ttt|d d���nt|� ��qS )r N)r � sa_schemaZColumnrd rG �getattrrP )r9 �exp)r r r r; � s z3_get_index_rendered_expressions.<locals>.<listcomp>c S s g | ]}t tt|d d����qS )r N)rd rG r� )r9 r: r r r r; � s )r r� Z expressionsrC )�idxr r )r r rV � s rV c C s: g }|j }| jr$|�dt| j�f� | jr>|�dt| j�f� |sd|rd| jjrd|�dt| jj�f� |s�| jr�|�dt || j�f� |r�t 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 )Nrb ra r r c S s g | ]}t |j��qS r )rG r )r9 r: r r r r; � s z"_uq_constraint.<locals>.<listcomp>c S s g | ]\}}d ||f �qS )z%s=%rr )r9 rK �vr r r r; � s z,%(prefix)screate_unique_constraint(%(args)s)z, )r? r@ c S s g | ]}t t|j���qS r )rd rG r )r9 r: r r r r; � s c S s g | ]\}}d ||f �qS )z%s=%rr )r9 rK r� r r r r; � s z$%(prefix)sUniqueConstraint(%(args)s))rU rb r5 �strra rJ r rG r r rd rC r7 r r# r� )rj r Zalterr3 rY r@ r r r r] � s6 r] c C s$ | j d }|d kr d|j S |S d S )NZuser_module_prefixz%s.)r3 r� )r �targetr? r r r �_user_autogenerate_prefix� s r� c C s | j d p dS )NZsqlalchemy_module_prefixr )r3 )r r r r r� � s r� c C s | j r dS | jd pdS d S )Nz batch_op.Zalembic_module_prefixr )rU r3 )r r r r r � s r c C s4 d|j kr0|j d }|r0|| ||�}|dk r0|S dS )NZ render_itemF)r3 )r{ Zobject_r Zrenderr~ r r r �_user_defined_render� s r� c C s� t d| |�}|dk r|S g }| jr@t| j|�}|r@|�d|f� | jsV|�d| jf� | jd k rp|�d| jf� dt|�t| j�t | j |�d� dd � |D ��d � S ) Nrp Frz � autoincrementr| z,%(prefix)sColumn(%(name)r, %(type)s, %(kw)s)z, c S s g | ]\}}d ||f �qS )z%s=%sr )r9 �kwnamerR r r r r; s z"_render_column.<locals>.<listcomp>)r? r rm rH ) r� rz ry r5 r� r| r� rG r rx rm r# )rp r r~ r3 r r r r< s$ r< c C sj t d| |�}|dk r|S t| tj�rFt| jtj�r:| j} n t| j|�S t| t�rf|rftt � dd| ��} | S )Nrz Fz^'|'$r ) r� r r� Z DefaultClause�argr r rP rd r r )�defaultr �repr_r~ r r r ry # s ry c C s� t d| |�}|dk r|S t| �j}|j}|�d�rft�d|��d�}|d k rZ|�d| � d|| f S |�d�r�t |�}d || f S t || �}d || f S d S ) Nrm Fzsqlalchemy.dialectszsqlalchemy\.dialects\.(\w+)� z"from sqlalchemy.dialects import %sz%s.%rz sqlalchemy.z%s%r) r� rm r� r � startswithr �match�group�addr� r� )r{ r r~ �modr Zdnamer? r r r rx 4 s rx c C s t �| �}|| |�S )N)�_constraint_renderersr/ )rj r r0 r r r r= K s 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)z, c S s g | ]}t |j��qS r )rd rQ )r9 �cr r r r; ` s z'_render_primary_key.<locals>.<listcomp>c S s g | ]\}}d ||f �qS )z%s=%sr )r9 r� rR r r r r; a s )r? r@ )r� rC r r5 rd r r� r# )rj r r~ r3 r r r �_render_primary_keyP s r� c C s� | � � }|�d�}|dd� \}}|dk rDt|�dkrDd||f }nd�|dd� �}| jdk r�| jjdk r�| jjj}||jkr�t|j| j | j �}d||f }|S )zoImplement a 'safe' version of ForeignKey._get_colspec() that never tries to resolve the remote table. �.���Nr z%s.%sr ���) Z _get_colspec�splitrE r# �parentrJ �metadataZtablesrG r� r )ZfkZmetadata_schemaZcolspec�tokensrM ZcolnameZtable_fullnameZparent_metadatar r r � _fk_colspecf s 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� d S )Nr_ r` ra rb rc )r_ r5 rd r` ra rb rc )rj r3 r r r �_populate_render_fk_opts� s r� c s� t d| |�}|dk r|S g }| jr<|�dtt|| j��f� t| |� | jjj� 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)z, c s s | ]}d t |jj� V qdS )z%rN)rG r� r )r9 �fr r r � <genexpr>� s z&_render_foreign_key.<locals>.<genexpr>c 3 s | ]}t t|� ��V qd S )N)rd r� )r9 r� )�apply_metadata_schemar r r� � s c S s g | ]\}}d ||f �qS )z%s=%sr )r9 r� rR r r r r; � s z'_render_foreign_key.<locals>.<listcomp>)r? �colsZrefcolsr@ ) r� r r5 rd r r� r� r� r r� r# r� )rj r r~ r3 r )r� r �_render_foreign_key� s" r� c C s$ t d| |�}|dk r|S t| |d�S )NrS F)r� r] )rj r r~ r r r �_render_unique_constraint� s 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)z, c s s | ]\}}d ||f V qdS )z%s=%sNr )r9 rK r� r r r r� � s z+_render_check_constraint.<locals>.<genexpr>r )r� )r? r3 �sqltext)r� Z _create_rule�hasattrr r� �sqltypesZ TypeEnginer r5 rd r r� r# rP r� )rj r r~ r3 r r r �_render_check_constraint� s( r� )T)T)SZ sqlalchemyr r� r r� r Z operationsr r r r Z util.compatr r Z mako.pygenr r rF Zsqlalchemy.sql.namingr r � ImportErrorr r% Z DispatcherZdefault_renderersr. r" r' r1 Z dispatch_forZModifyTableOpsr8 Z CreateTableOprL Z DropTableOprO Z CreateIndexOpr[ Z DropIndexOpr\ ZCreateUniqueConstraintOpr^ ZCreateForeignKeyOprh ZCreatePrimaryKeyOprk ZCreateCheckConstraintOprl ZDropConstraintOpro Z AddColumnOprq Z DropColumnOprt Z AlterColumnOpr �objectr rG rP rV r] r� r� r r� r< ry rx r� r= ZPrimaryKeyConstraintr� r� r� ZForeignKeyConstraintr� ZUniqueConstraintr� ZCheckConstraintr� �branchr r r r �<module> sj !(7 % "