관리-도구
편집 파일: sqlite.cpython-38.pyc
U 0?�f� � @ sj d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d d l m Z d d l m Z d d l m Z d d lmZ d dlmZ d dlmZ d dlmZ ddlmZ ddlmZ e�r:d dlmZ d dlmZ d dl m Z d dl m Z d dl m Z d dl m!Z! d dl m"Z" d dl#m$Z$ dd l%m&Z& G d d � d e�Z'eed �d d!d"d#�d$d%��Z(dS )&� )� annotationsN)�Any)�Dict)�Optional)� TYPE_CHECKING)�Union)�cast)�JSON)�schema)�sql� )� alter_table)�format_table_name)� RenameTable)� DefaultImpl� )�util)�compiles)� Inspector)� DDLCompiler)�Cast)� ClauseElement)�Column)� Constraint)�Table)� TypeEngine)�BatchOperationsImplc s� e Zd ZdZdZddd�dd�Zdd �d d �Zdd �d d �Zdddddd�dd�Zddd�dd�Z ddddd�dd �Z d)d dd d �� fd d!� Z dd"d#dd$�d%d&�Z d'd(� Z � ZS )*� SQLiteImpl�sqliteFr �bool)�batch_op�returnc C s~ |j D ]r}|d dkrf|d d }t|jtj�rFt|jjtj�rF dS t|jtj j �rx|jj rx dS q|d dkr dS qdS )z�Return True if the given :class:`.BatchOperationsImpl` would need the table to be recreated and copied in order to proceed. Normally, only returns True on SQLite when operations other than add_column are present. r � add_columnr T)Z create_indexZ drop_indexFN) �batch� isinstanceZserver_defaultr Z DefaultClause�argr r r Z sqla_compatZComputedZ persisted)�selfr �op�col� r) �@/opt/hc_python/lib/python3.8/site-packages/alembic/ddl/sqlite.py�requires_recreate_in_batch. s$ ���� z%SQLiteImpl.requires_recreate_in_batchr )�constc C s, |j d krtd��n|� | �r(t�d� d S )N��No support for ALTER of constraints in SQLite dialect. Please refer to the batch mode feature which allows for SQLite migrations using a copy-and-move strategy.z�Skipping unsupported ALTER for creation of implicit constraint. Please refer to the batch mode feature which allows for SQLite migrations using a copy-and-move strategy.)� _create_rule�NotImplementedErrorr �warn�r& r, r) r) r* �add_constraintJ s � �zSQLiteImpl.add_constraintc C s |j d krtd��d S )Nr- )r. r/ r1 r) r) r* �drop_constraint[ s �zSQLiteImpl.drop_constraintz Column[Any]z Optional[str])�inspector_column�metadata_column�rendered_metadata_default�rendered_inspector_defaultr! c C sP |d k r$t �dd|�}t �dd|�}|d k rHt �dd|�}t �dd|�}||kS )Nz ^\((.+)\)$z\1z^\"?'(.+)'\"?$)�re�sub)r& r4 r5 r6 r7 r) r) r* �compare_server_defaultc s. � � � �z!SQLiteImpl.compare_server_default)�exprr! c C s@ |sdS t �d|�rdS t �d|�r(dS t �d|�r8dS dS dS )a Determine if a server default is a SQL expression or a constant. There are too many assertions that expect server defaults to round-trip identically without parenthesis added so we will add parens only in very specific cases. Fz ^[0-9\.]$z^'.+'$z^\(.+\)$TN)r8 �match)r& r; r) r) r* �-_guess_if_default_is_unparenthesized_sql_expr~ s z8SQLiteImpl._guess_if_default_is_unparenthesized_sql_exprr r zDict[str, Any]�None)� inspector�table� column_infor! c C s( | � |�dd ��r$d|d f |d< d S )N�default�(%s))r= �get)r& r? r@ rA r) r) r* �autogen_column_reflect� s �z!SQLiteImpl.autogen_column_reflectr �str)r; �is_server_defaultr! c s4 t � j|fd|i|��}|r0| �|�r0d|f }|S )NrG rC )�super�render_ddl_sql_exprr= )r& r; rG �kwZstr_expr�� __class__r) r* rI � s ����� z SQLiteImpl.render_ddl_sql_exprz"Dict[str, Union[TypeEngine, Cast]]r )�existing�existing_transfer�new_typer! c C s. |j j|jk r*t|t�s*t|d |�|d< d S )Nr; )�typeZ_type_affinityr$ r r )r& rM rN rO r) r) r* �cast_for_batch_migrate� s �� �z!SQLiteImpl.cast_for_batch_migratec C s | � ||� d S )N)Z_skip_functional_indexes)r& Zconn_unique_constraintsZ conn_indexesZmetadata_unique_constraintsZmetadata_indexesr) r) r* � correct_for_autogen_constraints� s z*SQLiteImpl.correct_for_autogen_constraints)F)�__name__� __module__� __qualname__Z __dialect__Ztransactional_ddlr+ r2 r3 r: r= rE rI rQ rR � __classcell__r) r) rK r* r & s �r r r r rF )�element�compilerr! c K s" dt || j| j�t|| jd �f S )Nz%s RENAME TO %s)r Z table_namer r Znew_table_name)rW rX rJ r) r) r* �visit_rename_table� s �rY ))� __future__r r8 �typingr r r r r Z sqlalchemyr r r r �baser r r �implr � r Zutil.sqla_compatr Z sqlalchemy.engine.reflectionr Zsqlalchemy.sql.compilerr Zsqlalchemy.sql.elementsr r Zsqlalchemy.sql.schemar r r Zsqlalchemy.sql.type_apir Zoperations.batchr r rY r) r) r) r* �<module> s< #