관리-도구
편집 파일: toimpl.cpython-38.pyc
U 0?�f� � @ s d dl mZ d dlmZ ddlmZ ddlmZ ddl m Z ddl m Z erXd d l m Z e�ej�d d d d �dd��Ze�ej�d dd d �dd��Ze�ej�d dd d �dd��Ze�ej�d dd d �dd��Ze�ej�d dd d �dd��Ze�ej�d d d d �d d ��Ze�ej�d d d d �d!d"��Z e�ej �d d#d d �d$d%��Z e�ej �d d&d d �d'd(��Z e�ej!�d d)d d �d*d+��Z"e�ej#�d d,d d �d-d.��Z$e�ej%�d d/d d �d0d1��Z&e�ej'�d d2d d �d3d4��Z(e�ej)�d d5d d �d6d7��Z*d S )8� )� TYPE_CHECKING��schema� )�ops)� Operations� )�_copy)�sqla_14)�Tabler zops.AlterColumnOpN)� operations� operation�returnc s" | j j�| j jd �� |j}|j}|j}|j}|j}|j}|j }|j } |j } |j } |j } |j} � fdd�}|r�|r�| jj|t�||�|d�}|jD ]}||�r�| j �|� q�| j j||f| | | |||||| | d� |j�� |�r | jj|| j�||�|d�}|jD ] }||��r | j �|� �q d S )Nc s t | tj� o | j p | �� �S �N)� isinstance� sa_schema�PrimaryKeyConstraintZ _create_rule)� constraint��compiler� �G/opt/hc_python/lib/python3.8/site-packages/alembic/operations/toimpl.py�_count_constraint& s �z'alter_column.<locals>._count_constraintr ) �nullable�server_default�name�type_r � existing_type�existing_server_default�existing_nullable�comment�existing_comment)�impl�dialectZstatement_compilerr r r Z modify_type� column_name� table_namer Zmodify_server_defaultZ modify_nameZmodify_nullableZmodify_commentr! � schema_obj�tabler �Column� constraints�drop_constraint� alter_column�kw�column�add_constraint)r r r r r r r$ r% r r Znew_column_namer r r! r �tr r r r r+ sh � � �� � � r+ zops.DropTableOpc C s | j �|�| j�� d S r )r" � drop_table�to_table�migration_context�r r r r r r0 P s �r0 zops.DropColumnOpc C s0 |� | j�}| jj|j|fd|ji|j�� d S )Nr )Z to_columnr2 r" � drop_columnr% r r, )r r r- r r r r4 W s ���r4 zops.CreateIndexOpc C sD |� | j�}i }|jd k r0ts&td��|j|d< | jj|f|� d S )N�SQLAlchemy 1.4+ required� if_not_exists)�to_indexr2 r6 r �NotImplementedErrorr" � create_index)r r �idxr, r r r r9 a s r9 zops.DropIndexOpc C s@ i }|j d k r$tstd��|j |d< | jj|�| j�f|� d S )Nr5 � if_exists)r; r r8 r" � drop_indexr7 r2 )r r r, r r r r< o s ��r< zops.CreateTableOpr c C s |� | j�}| j�|� |S r )r1 r2 r" � create_table�r r r' r r r r= ~ s r= zops.RenameTableOpc C s | j j|j|j|jd� d S �Nr )r" � rename_tabler% Znew_table_namer r3 r r r r@ � s �r@ zops.CreateTableCommentOpc C s |� | j�}| j�|� d S r )r1 r2 r" �create_table_commentr> r r r rA � s rA zops.DropTableCommentOpc C s |� | j�}| j�|� d S r )r1 r2 r" �drop_table_commentr> r r r rB � s rB zops.AddColumnOpc C s� |j }|j}|j}|j}|jd k r*t|�}| jj|||d�}| jj||fd|i|�� |j D ] }t |t j �s\| j� |� q\|jD ]}| j�|� q�| jjjo�| jjj } |j} | r�| r�| j�|� d S )Nr r )r% r- r r, r' r r& r" � add_columnr) r r r r. Zindexesr9 r# Zsupports_commentsZinline_commentsr Zcreate_column_comment) r r r% r- r r, r/ r �indexZ with_commentr r r r rC � s& �rC zops.AddConstraintOpc C s | j �|�| j�� d S r )r" r. Z to_constraintr2 r3 r r r �create_constraint� s �rE zops.DropConstraintOpc C s( | j �| jj|j|j|j|jd�� d S r? )r" r* r& Zgeneric_constraintZconstraint_namer% Zconstraint_typer r3 r r r r* � s ��r* zops.BulkInsertOpc C s | j j|j|j|jd� d S )N)� multiinsert)r" � bulk_insertr' �rowsrF r3 r r r rG � s �rG zops.ExecuteSQLOpc C s | j jj|j|jd� d S )N)�execution_options)r2 r" �executeZsqltextrI r3 r r r � execute_sql� s �rK )+�typingr Z sqlalchemyr r � r �baser Zutil.sqla_compatr r Zsqlalchemy.sql.schemar Zimplementation_forZ AlterColumnOpr+ Z DropTableOpr0 Z DropColumnOpr4 Z CreateIndexOpr9 Z DropIndexOpr< Z CreateTableOpr= Z RenameTableOpr@ ZCreateTableCommentOprA ZDropTableCommentOprB Z AddColumnOprC ZAddConstraintOprE ZDropConstraintOpr* Z BulkInsertOprG Z ExecuteSQLOprK r r r r �<module> s� �> � � � � � � � � � �