Edit File: schemaobj.cpython-37.pyc
B �M!V� � @ sH d dl mZ d dlmZmZ ddlmZ ddlm Z G dd� de �Z dS ) � )�schema)�NULLTYPE�Integer� )� string_types)�utilc @ sz e Zd Zddd�Zddd�Zd dd�Zd dd �Zd d d �Zd d d �Zdd� Z dd� Z dd� Z d dd�Z dd� Z dd� ZdS )!� SchemaObjectsNc C s || _ d S )N)�migration_context)�selfr � r �M/opt/alt/python37/lib/python3.7/site-packages/alembic/operations/schemaobj.py�__init__ s zSchemaObjects.__init__c sZ | � � }dd� |D �}tj||f|�d|i�� tj� fdd�|D �d|i�}� �|� |S )Nc S s g | ]}t �|t��qS r )� sa_schema�Columnr )�.0�nr r r � <listcomp> s z8SchemaObjects.primary_key_constraint.<locals>.<listcomp>r c s g | ]}� j | �qS r )�c)r r )�tr r r s �name)�metadatar �Table�PrimaryKeyConstraint�append_constraint)r r � table_name�colsr �m�columns�pr )r r �primary_key_constraint s z$SchemaObjects.primary_key_constraintc s� | � � }||kr|| }n&|}tj||fdd� |D ��d| i� tj||fdd� |D ��d| i�}| rrd| |f n|� tjr�| | d< tj|� fdd�|D �f||||| d�| ��}|�|� |S ) Nc S s g | ]}t �|t��qS r )r r r )r r r r r r & s z8SchemaObjects.foreign_key_constraint.<locals>.<listcomp>r c S s g | ]}t �|t��qS r )r r r )r r r r r r + s z%s.%s�matchc s g | ]}d � |f �qS )z%s.%sr )r r )�tnamer r r 6 s )r �onupdate�ondelete� deferrable� initially)r r r r Zsqla_08�ForeignKeyConstraintr )r r �sourceZreferent� local_colsZ remote_colsr" r# r$ Z source_schemaZreferent_schemar% r Z dialect_kwr Zt1_cols�t1�fr )r! r �foreign_key_constraint s6 z$SchemaObjects.foreign_key_constraintc sV t j|| �� fdd� |D ��d|i�� ||d< t j� fdd�|D �|�}� �|� |S )Nc S s g | ]}t �|t��qS r )r r r )r r r r r r F s z3SchemaObjects.unique_constraint.<locals>.<listcomp>r r c s g | ]}� j | �qS r )r )r r )r r r r I s )r r r �UniqueConstraintr )r r r' r( r �kwZuqr )r r �unique_constraintC s z SchemaObjects.unique_constraintc K sB t j|| �� t �dt�|d�}t j|fd|i|��}|�|� |S )N�x)r r )r r r r r �CheckConstraintr )r r r' � conditionr r- r Zckr r r �check_constraintO s z SchemaObjects.check_constraintc K s� | j ||d�}dd� tjtjdd� tjd�}y || }W n4 tk rl tdd�tdd � |D ��� ��Y nX ||d �}|� |� |S d S ) N)r c S s t jg g | d�S )N)r )r r& )r r r r �<lambda>Y s z2SchemaObjects.generic_constraint.<locals>.<lambda>c S s t jd| d�S )N� )r )r r0 )r r r r r3 ] � )Z foreignkeyZprimary�unique�checkNz'type' can be one of %sz, c s s | ]}t |�V qd S )N)�repr)r r/ r r r � <genexpr>d s z3SchemaObjects.generic_constraint.<locals>.<genexpr>)r ) �tabler r r, Z Constraint�KeyError� TypeError�join�sortedr ) r r r �type_r r- r �types�constr r r �generic_constraintV s " z SchemaObjects.generic_constraintc C sF i }| j d k r:d| j jkr:| j jd }t|d�r:|j|d< tjf |�S )NZtarget_metadata�naming_convention)r �opts�hasattrrC r ZMetaData)r r- �mtr r r r j s zSchemaObjects.metadatac O s< | � � }tj||f|�|�}x|jD ]}| �||� q$W |S )N)r r r Z foreign_keys�_ensure_table_for_fk)r r r r- r r r* r r r r: s s zSchemaObjects.tablec K s t j||f|�S )N)r r )r r r? r- r r r �columnz s zSchemaObjects.columnc s< t j|p d| �� |d�� t j|f� fdd�|D ��|�}|S )NZno_table)r c s g | ]}t j�� |��qS r )r Z sqla_compatZ_textual_index_column)r r )r r r r � s z'SchemaObjects.index.<locals>.<listcomp>)r r r �Index)r r Z tablenamer r r- �idxr )r r �index} s zSchemaObjects.indexc C s> d|kr.|� d�}d�|dd� �}|d }n|}d }||fS )N�.r ���)�splitr= )r � table_key�tokens�snamer! r r r �_parse_table_key� s z SchemaObjects._parse_table_keyc C sr t |jt�rn|j�dd�\}}| �|�\}}||jkrHtj|||d�}n |j| }||jkrn|� t� |t �� dS )zUcreate a placeholder Table object for the referent of a ForeignKey. rL � )r N) � isinstanceZ_colspecr �rsplitrR Ztablesr r r Z append_columnr r )r r ZfkrO ZcnamerQ r! Zrel_tr r r rG � s z"SchemaObjects._ensure_table_for_fk)N)N)NNNNNNN)N)N)N)N)�__name__� __module__� __qualname__r r r+ r. r2 rB r r: rH rK rR rG r r r r r s % r N) Z sqlalchemyr r Zsqlalchemy.typesr r Z util.compatr r4 r �objectr r r r r �<module> s