Edit File: render.cpython-37.opt-1.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 ZbufZprinter�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 r/ 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) r1 r0 r* Zt_opZt_linesr r r �_render_modify_tableS s$ r6 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)r7 r8 )r r r r9 u s c S s g | ]}|d k r|�qS )Nr )r7 Zrconsr r r r9 w s c s g | ]}t |� ��qS r )�_render_constraint)r7 Zcons)r r r r9 x s z*[z, �]z.%(prefix)screate_table(%(tablename)r, %(args)s)Z tablename�prefix�argsz , schema=%rz, %s=%r� �_z )) Zto_table�columnsr$ Z constraints�len�MAX_PYTHON_ARGSr# �_identr4 r r �kw�replace)r r) �tabler> r �kr )r r � _add_tablep s& "rI c C s< dt | �t|j�d� }|jr0|dt|j� 7 }|d7 }|S )Nz %(prefix)sdrop_table(%(tname)r)r= �tnamez , schema=%r�))r rD r4 r )r r) r r r r � _drop_table� s rL 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)r7 �key�val)r r r r9 � s z _add_index.<locals>.<listcomp>)r= r rG rA �uniquer �kwargs)Zto_index� _has_batchr r r rD rG r# � _get_index_rendered_expressionsrP r rB rQ �items)r r) �index� has_batch�tmplr r )r r � _add_index� s 6rX 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 r4 r )rR r r Z index_namerD r4 r )r r) rV rW r r r r � _drop_index� s rY 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 )rD )r7 r8 r r r r9 � s z&_add_fk_constraint.<locals>.<listcomp>c S s g | ] }t |��qS r )rD )r7 r8 r r r r9 � 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_namerR r3 rD Z source_tabler5 Zreferent_tableZ local_colsZ remote_cols�insertrE r r# )r r) r> rQ rH �valuer r r �_add_fk_constraint� s, re c C s t � �d S )N)�NotImplementedError)� constraintr r r r �_add_pk_constraint� s rh c C s t � �d S )N)rf )rg r r r r �_add_check_constraint s ri 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 r4 �typer )rR r r rb rD r4 Zconstraint_typer )r r) �templater r r r �_drop_constraint s rl 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)rrK )r= rJ �columnr )r r4 rm rR r r: )r r) r rJ rm rk r r r r � _add_column s rn 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)rrK )r= rJ �cnamer )r r4 � column_namerR r rD )r r) r rJ rp rk r r r r � _drop_column1 s rq 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= rJ ro z, %sexisting_type=%sFz, %sserver_default=%sz , %stype_=%sz, %snullable=%rz, %sexisting_nullable=%rz , %sexisting_server_default=%sz , %sschema=%rrK ) r4 rp Zmodify_server_defaultZ modify_typeZmodify_nullable� existing_type�existing_nullable�existing_server_defaultr rR r � _repr_type�_render_server_default)r r) rJ ro �server_default�type_�nullablerr rs rt r �indentrk 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= rD 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 rD � s rD 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� ra )rd r � wrap_in_textZ compile_kwrk r r r rM � s rM 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_schemaZColumnra rD �getattrrM )r7 �exp)r r r r9 � s z3_get_index_rendered_expressions.<locals>.<listcomp>c S s g | ]}t tt|d d����qS )r N)ra rD r� )r7 r8 r r r r9 � s )r r� Z expressionsrA )�idxr r )r r rS � s rS 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 )Nr_ r^ r r c S s g | ]}t |j��qS r )rD r )r7 r8 r r r r9 � s z"_uq_constraint.<locals>.<listcomp>c S s g | ]\}}d ||f �qS )z%s=%rr )r7 rH �vr r r r9 � s z,%(prefix)screate_unique_constraint(%(args)s)z, )r= r> c S s g | ]}t t|j���qS r )ra rD r )r7 r8 r r r r9 � s c S s g | ]\}}d ||f �qS )z%s=%rr )r7 rH r� r r r r9 � s z$%(prefix)sUniqueConstraint(%(args)s))rR r_ r3 �strr^ rG r rD r r ra rA r5 r r# r� )rg r Zalterr1 rV r> r r r rZ � s6 rZ c C s$ | j d }|d kr d|j S |S d S )NZuser_module_prefixz%s.)r1 r� )r �targetr= r r r �_user_autogenerate_prefix� s r� c C s | j d p dS )NZsqlalchemy_module_prefixr )r1 )r r r r r� � s r� c C s | j r dS | jd pdS d S )Nz batch_op.Zalembic_module_prefixr )rR r1 )r r r r r � s r c C s4 d|j kr0|j d }|r0|| ||�}|dk r0|S dS )NZ render_itemF)r1 )rx 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 ) Nrm Frw � autoincrementry z,%(prefix)sColumn(%(name)r, %(type)s, %(kw)s)z, c S s g | ]\}}d ||f �qS )z%s=%sr )r7 �kwnamerO r r r r9 s z"_render_column.<locals>.<listcomp>)r= r rj rE ) r� rw rv r3 r� ry r� rD r ru rj r# )rm r r{ r1 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 )Nrw Fz^'|'$r ) r� r r� Z DefaultClause�argr r rM ra r r )�defaultr �repr_r{ r r r rv # s rv 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 ) Nrj Fzsqlalchemy.dialectszsqlalchemy\.dialects\.(\w+)� z"from sqlalchemy.dialects import %sz%s.%rz sqlalchemy.z%s%r) r� rj r� r � startswithr �match�group�addr� r� )rx r r{ �modr Zdnamer= r r r ru 4 s ru c C s t �| �}|| |�S )N)�_constraint_renderersr- )rg r r. 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 )ra rN )r7 �cr r r r9 ` s z'_render_primary_key.<locals>.<listcomp>c S s g | ]\}}d ||f �qS )z%s=%sr )r7 r� rO r r r r9 a s )r= r> )r� rA r r3 ra r r� r# )rg r r{ r1 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�splitrB r# �parentrG �metadataZtablesrD r� r )ZfkZmetadata_schemaZcolspec�tokensrJ 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] r^ r_ r` )r\ r3 ra r] r^ r_ r` )rg r1 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)rD r� r )r7 �fr r r � <genexpr>� s z&_render_foreign_key.<locals>.<genexpr>c 3 s | ]}t t|� ��V qd S )N)ra r� )r7 r� )�apply_metadata_schemar r r� � s c S s g | ]\}}d ||f �qS )z%s=%sr )r7 r� rO r r r r9 � s z'_render_foreign_key.<locals>.<listcomp>)r= ZcolsZrefcolsr> ) r� r r3 ra r r� r� r� r r� r# r� )rg r r{ r1 r )r� r �_render_foreign_key� s" r� c C s$ t d| |�}|dk r|S t| |d�S )NrP F)r� rZ )rg 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 )r7 rH r� r r r r� � s z+_render_check_constraint.<locals>.<genexpr>r )r� )r= r1 �sqltext)r� Z _create_rule�hasattrr r� �sqltypesZ TypeEnginer r3 ra r r� r# rM r� )rg r r{ r1 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 rC Zsqlalchemy.sql.namingr r � ImportErrorr r% Z DispatcherZdefault_renderersr, r" r' r/ Z dispatch_forZModifyTableOpsr6 Z CreateTableOprI Z DropTableOprL Z CreateIndexOprX Z DropIndexOprY ZCreateUniqueConstraintOpr[ ZCreateForeignKeyOpre ZCreatePrimaryKeyOprh ZCreateCheckConstraintOpri ZDropConstraintOprl Z AddColumnOprn Z DropColumnOprq Z AlterColumnOpr| �objectr rD rM rS rZ r� r� r r� r: rv ru r� r; ZPrimaryKeyConstraintr� r� r� ZForeignKeyConstraintr� ZUniqueConstraintr� ZCheckConstraintr� �branchr r r r �<module> sj !(7 % "