관리-도구
편집 파일: postgresql.cpython-38.pyc
U 0?�f�t � @ s* d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d d lm Z d d lm Z d d lm Z d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dl m Z d dl m!Z! ddl"m#Z# ddl"m$Z$ dd l"m%Z% dd l"m&Z& dd l"m'Z' dd l"m(Z( dd l"m)Z) dd!l"m*Z* dd"l"m+Z+ dd#l,m-Z- dd$l,m.Z. d%d&l/m0Z0 d%d'l1m2Z2 d%d(l3m4Z4 d%d)l3m5Z5 d%d*l6m7Z7 d%d+l6m8Z8 d%d,l0m9Z9 d%d-l:m;Z; e �r�d d.lm<Z< d d/lm=Z= d d0lm>Z> d d1l?m@Z@ d d2lAmBZB d d3lCmDZD d d4lEmFZF d d5lEmGZG d d6lmHZH d d7lmIZI d d8lmJZJ d d9lKmLZL d d:lKmMZM d d;lNmOZO dd<l"mPZP d%d=lQmRZR d%d>lSmTZT d%d?lUmVZV e�WeX�ZYG d@dA� dAe.�ZZG dBdC� dCe%�Z[e;e+dD�dEdFdGdH�dIdJ��Z\e;e[dD�dCdFdGdH�dKdL��Z]e;e&dD�dMdFdGdH�dNdO��Z^e;e*dD�dPdFdQ�dRdS��Z_e8�`dT�e7�`dTdU�e4ja�bdV�G dWdX� dXe4ja����Zce2jd�eec�dYdXdGdZ�d[d\��Zfe2jg�ee�d]dYd^dGd_�d`da��ZhdYdGdb�dcdd�Zid]dYdedGdf�dgdh�ZjdidYdGdj�dkdl�ZkdS )m� )� annotationsN)�Any)�cast)�Dict)�List)�Optional)�Sequence)�Tuple)� TYPE_CHECKING)�Union)�Column)�literal_column)�Numeric)�text)�types)�BIGINT)�ExcludeConstraint)�INTEGER)� CreateIndex)� ColumnClause)� TextClause)�FunctionElement)�NULLTYPE� )� alter_column)� alter_table)� AlterColumn)� ColumnComment)�format_column_name)�format_table_name)� format_type)�IdentityColumnDefault)� RenameTable)�ComparisonResult)� DefaultImpl� )�util)�render)�ops)� schemaobj)�BatchOperations)� Operations)� sqla_compat)�compiles)�Literal)�Index)�UniqueConstraint)�ARRAY)� PGDDLCompiler)�HSTORE)�JSON)�JSONB)� ClauseElement)� ColumnElement)� quoted_name)�MetaData)�Table)� TypeEngine)�_ServerDefault)�AutogenContext)�_f_name)�MigrationContextc s� e Zd ZdZdZejddhf Zdddd�d d �Zd d � Zd d� Z dRddddddddddddddd�� fdd� Z dd� Z d d � Z e �d �e �d �e �d �e �d!�e �d"�fZdddd#�d$d%�Zd&d'd(�d)d*�Zddd+d,�d-d.�Zd/d/d+d0�d1d2�Zd3dd3d4�d5d6�Zd7dd8�d9d:�ZdSd;d<d<ddd=�� fd>d?� Zd@dAdBdC�dDdE�ZdFdAddC�dGdH�ZdIdAddC�dJdK�ZdLdAddC�dMdN�ZdOdAddC�dPdQ�Z� Z S )T�PostgresqlImpl� postgresqlT�FLOATzDOUBLE PRECISIONr/ r �None)�index�kw�returnc K sR |j �dd �pd}|D ]$}||jjkr|j�t|tj�� q| �t |f|�� d S )N�postgresql_include� ) �kwargs�get�table�c� append_columnr �sqltypesZNullType�_execr )�selfrD rE rG �colrH rH �D/opt/hc_python/lib/python3.8/site-packages/alembic/ddl/postgresql.py� create_indexU s �zPostgresqlImpl.create_indexc C s0 |j D ]$}|jd k r|j|jkr| �|� qd S �N)� constraints�nameZnamed_constraintsZdrop_constraint)rP Z batch_implrK � constraintrH rH rR �prep_table_for_batcha s � �z#PostgresqlImpl.prep_table_for_batchc C s� |j r||jjkrdS |}||k}|r*dS d |||jfkr@| S |jj}t|t�r�t|jt�sxt � dd|�}d|� d�}t |�}| j }|d k s�t �|�t�t |�|k�� S )NFz^'|'$� �')Z primary_keyrK Z_autoincrement_column�server_default�arg� isinstance�str�typer �re�subr � connection�AssertionErrorZscalarr, �_select) rP Zinspector_columnZmetadata_columnZrendered_metadata_defaultZrendered_inspector_defaultZconn_col_defaultZdefaults_equalZmetadata_default�connrH rH rR �compare_server_defaulti s8 � �� ��z%PostgresqlImpl.compare_server_defaultNFr^ zOptional[bool]z%Union[_ServerDefault, Literal[False]]� Optional[str]zOptional[TypeEngine]zOptional[_ServerDefault])� table_name� column_name�nullabler[ rV �type_�schema� autoincrement� existing_type�existing_server_default�existing_nullable�existing_autoincrementrE rF c sz | � dd �}|d k r&|d kr&t�d��|d k rL| �t|||||| | | d�� t� j||f|||||| | | | d� | �� d S )NZpostgresql_usingz6postgresql_using must be used with the type_ parameter)rl �usingrn ro rp ) rj r[ rV rl rm rn ro rp rq )�popr& � CommandErrorrO �PostgresqlColumnType�superr )rP rh ri rj r[ rV rk rl rm rn ro rp rq rE rr �� __class__rH rR r � sD ��� �� �zPostgresqlImpl.alter_columnc C s� |� d�r~t|d ttf�r~t�d|d �}|r~tj|td�|� d�d�� � }|r~|\}}||d kr~t � d||j |� |d= d S ) N�defaultr_ z nextval\('(.+?)'::regclass\)a9 select c.relname, a.attname from pg_class as c join pg_depend d on d.objid=c.oid and d.classid='pg_class'::regclass and d.refclassid='pg_class'::regclass join pg_class t on t.oid=d.refobjid join pg_attribute a on a.attrelid=t.oid and a.attnum=d.refobjsubid where c.relkind='S' and c.oid=cast(:seqname as regclass)r )�seqnamerV z^Detected sequence named '%s' as owned by integer column '%s(%s)', assuming SERIAL and omitting)rJ r] r r r` �matchr, Z_exec_on_inspectorr �group�first�log�inforV )rP Z inspectorrK Z column_infoZ seq_matchr rz ZcolnamerH rH rR �autogen_column_reflect� s4 � �� � � z%PostgresqlImpl.autogen_column_reflectc C s8 dd� |D �}|D ]}|� |� qtjs4| �||� d S )Nc S s h | ]}|j �d �r|�qS )Zduplicates_constraint)r rJ )�.0rD rH rH rR � <setcomp>� s �zAPostgresqlImpl.correct_for_autogen_constraints.<locals>.<setcomp>)�remover, Zsqla_2Z_skip_functional_indexes)rP Zconn_unique_constraintsZ conn_indexesZmetadata_unique_constraintsZmetadata_indexesZdoubled_constraintsZixrH rH rR � correct_for_autogen_constraints� s � z.PostgresqlImpl.correct_for_autogen_constraintsz( asc nulls last)$z(?<! desc)( nulls last)$z( asc)$z( asc) nulls first$z desc( nulls first)$)rD �exprrF c C s* |� � �dd��dd�}|jd k r<|�|jj� � � d�d�}d|krRt�dd|�}|r||d dkr||d d kr||d d � }qR| jD ]<}|�|� }r�|�d �\}}|d |� ||d � } q�q�|r�|d dkr�|d d kr�|d d � }q�t� d �}|� |��r|�d|�}t�d d|�}|�dd�}|S )N�"rY rZ �.z::z (::[\w ]+\w)r �(����)r z cast\s*\(z as\s+[^)]+\)� ) �lower�replacerK rV r` ra �_default_modifiers_re�search�span�compiler{ )rP rD r� �rsr{ �start�endZcast_rerH rH rR �_cleanup_index_expr s( z"PostgresqlImpl._cleanup_index_exprz Union[Index, UniqueConstraint]zTuple[Any, ...])�itemrF c C s |j �d�rdS dS )NZ postgresql_nulls_not_distinct)Znulls_not_distinctrH )Zdialect_kwargsrJ )rP r� rH rH rR �_dialect_options9 s z PostgresqlImpl._dialect_optionsr# )�metadata_index�reflected_indexrF c C s� g }| � ||�}|r |�|� |j}|j}t|�t|�krX|�dt|�� dt|�� �� |rft�|�S g }tt||�d�D ]�\}\} } | �| �} | � || �} | �| �} | � || �}| |kr�qzqz| � � � d��rd| ks�d| k�r|�d|� d| �d�� t � d |� d| �d |j�d �� qz|�d|� d| �d| ��� qz| �|�}| �|�}||k�rl|�d |� d|� �� |�r|t�|�S |�r�t�|�S t�� S d S ) Nzexpression number � to r Z_opsr� r� z expression #z' detected as including operator clause.z Expression #z in index a detected to include an operator clause. Expression compare cannot proceed. Please move the operator clause to the ``postgresql_ops`` dict to enable proper compare of the index expressions: https://docs.sqlalchemy.org/en/latest/dialects/postgresql.html#operator-classeszoptions )Z_compare_index_unique�appendZ expressions�lenr# � Different� enumerate�zip�_compile_elementr� �strip�endswithr& �warnrV r� �extendZSkip�Equal)rP r� r� �msgZ unique_msgZm_exprsZr_exprs�skip�posZm_eZr_eZ m_compileZm_textZ r_compileZr_textZ m_optionsZ r_optionsrH rH rR �compare_indexesB sT � ���� z PostgresqlImpl.compare_indexesr0 )�metadata_constraint�reflected_constraintrF c C s| | � |�}| �|�}|j}|j}||kr>t�d|� d|� ��S | �|j�}| �|j�}||krtt�d|� d|� ��S t�� S )Nz expression r� )Z _create_metadata_constraint_sigZ _create_reflected_constraint_sig�unnamedr# r� r� �constr� ) rP r� r� Z metadata_tupZ reflected_tupZmeta_sigZconn_sigZ metadata_doZconn_dorH rH rR �compare_unique_constraint~ s& ��� �z(PostgresqlImpl.compare_unique_constraintzDict[str, Any])�reflected_options�kindrF c C s* |� di ��� }|� d�s&|�dd � |S )NZdialect_optionsrG )rJ �copyrs )rP r� r� �optionsrH rH rR � adjust_reflected_dialect_options� s z/PostgresqlImpl.adjust_reflected_dialect_optionszUnion[ClauseElement, str])�elementrF c C s& t |t�r|S |j| jddd�d�jS )NTF)Z literal_bindsZ include_table)�dialect�compile_kwargs)r] r^ r� r� �string)rP r� rH rH rR r� � s �z PostgresqlImpl._compile_elementr6 �bool)r� �is_server_default�is_indexrE rF c s0 |rt |t�s|�� }t� j|f||d�|��S )zdRender a SQL expression that is typically a server default, index expression, etc. )r� r� )r] r Z self_grouprv �render_ddl_sql_expr)rP r� r� r� rE rw rH rR r� � s � ��z"PostgresqlImpl.render_ddl_sql_exprr; r= zUnion[str, Literal[False]])rk �autogen_contextrF c C sF t |�j}|�d�sdS t| d|j �rBt| d|j �}|||�S dS )Nz sqlalchemy.dialects.postgresqlFz_render_%s_type)r_ � __module__� startswith�hasattrZ__visit_name__�getattr)rP rk r� �mod�methrH rH rR � render_type� s zPostgresqlImpl.render_typer3 c C s t tt�||dd��S )N� text_typez(.+?\(.*text_type=)�r r^ r' Z_render_type_w_subtype�rP rk r� rH rH rR �_render_HSTORE_type� s ��z"PostgresqlImpl._render_HSTORE_typer1 c C s t tt�||dd��S )NZ item_typez(.+?\()r� r� rH rH rR �_render_ARRAY_type� s ��z!PostgresqlImpl._render_ARRAY_typer4 c C s t tt�||dd��S �NZ astext_typez(.+?\(.*astext_type=)r� r� rH rH rR �_render_JSON_type� s ��z PostgresqlImpl._render_JSON_typer5 c C s t tt�||dd��S r� r� r� rH rH rR �_render_JSONB_type� s ��z!PostgresqlImpl._render_JSONB_type) NFNNNNNNNN)FF) �__name__r� � __qualname__Z __dialect__Ztransactional_ddlr$ Z type_synonymsrS rX rf r r� r� r` r� r� r� r� r� r� r� r� r� r� r� r� r� r� � __classcell__rH rH rw rR r@ N sR � 1 �.4' �" < � r@ c s( e Zd Zddddd�� fdd� Z� ZS )ru r^ r; rC )rV ri rk rF c s4 |� dd �}t� j||f|� t�|�| _|| _d S )Nrr )rs rv �__init__rN Z to_instancerk rr )rP rV ri rk rE rr rw rH rR r� � s z PostgresqlColumnType.__init__)r� r� r� r� r� rH rH rw rR ru � s ru rA r" r2 r^ )r� �compilerrF c K s" dt || j| j�t|| jd �f S )Nz%s RENAME TO %s)r rh rl r Znew_table_name�r� r� rE rH rH rR �visit_rename_table s �r� c K s@ dt || j| j�t|| j�dt|| j� | jr8d| j ndf S )Nz %s %s %s %szTYPE %szUSING %srY )r rh rl r ri r rk rr r� rH rH rR �visit_column_type s �r� r c K sJ d}| j d k r"|j�| j t�� �nd}|jt|| j| j�t || j �|d�S )Nz9COMMENT ON COLUMN {table_name}.{column_name} IS {comment}ZNULL)rh ri �comment) r� Z sql_compilerZrender_literal_valuerN �String�formatr rh rl r ri )r� r� rE Zddlr� rH rH rR �visit_column_comment s" � �� � �r� r! )r� r� c K s� dt || j| j�t|| j�f }| jd kr6|d7 }|S | jd kr\|d7 }||�| j�7 }|S | j� | j| j�\}}}| j}t |�D ]H}|dkr�|d|j r�dnd 7 }q�|d|� t jf |t||�i�� 7 }q�|S d S ) Nz%s %s z DROP IDENTITYzADD �alwayszSET GENERATED %s ZALWAYSz BY DEFAULTzSET %s )r rh rl r ri ry ro �visit_identity_column�implZ_compare_identity_default�sortedr� Zget_identity_optionsr, ZIdentityr� )r� r� rE r �diff�_�identity�attrrH rH rR r� . s2 � � �� r� �create_exclude_constraint� batch_create_exclude_constraintZexclude_constraintc @ s� e Zd ZdZdZd!dddddd d d �d d �Zedd d�dd��Zd"ddd�dd�Zeddddddd�dd ��Z ed ddddd �d d ��Z dS )#�CreateExcludeConstraintOpz0Represent a create exclude constraint operation.�excludeNzsqla_compat._ConstraintNamezUnion[str, quoted_name]zIUnion[Sequence[Tuple[str, str]], Sequence[Tuple[ColumnClause[Any], str]]]z)Optional[Union[ColumnElement[bool], str]]rg zOptional[ExcludeConstraint]rC )�constraint_namerh �elements�whererl �_orig_constraintrF c K s. || _ || _|| _|| _|| _|| _|| _d S rT )r� rh r� r� rl r� rE )rP r� rh r� r� rl r� rE rH rH rR r� [ s z"CreateExcludeConstraintOp.__init__r )rW rF c C sD t �|�}| |j|jdd� |jD �td|j�|j||j|j|j d� S )Nc S s g | ]\}}}||f�qS rH rH )r� r� rV �oprH rH rR � <listcomp>x s z=CreateExcludeConstraintOp.from_constraint.<locals>.<listcomp>zColumnElement[bool] | None)r� rl r� � deferrable� initiallyrr ) r, Z_table_for_constraintrV � _render_exprsr r� rl r� r� rr )�clsrW Zconstraint_tablerH rH rR �from_constraintp s � �z)CreateExcludeConstraintOp.from_constraintzOptional[MigrationContext])�migration_contextrF c C sx | j d k r| j S t�|�}|j| j| jd�}t| j| j| j d�| j ��}|j D ]\}}}|� t |t�� qN|�|� |S )N)rl )rV r� )r� r) Z SchemaObjectsrK rh rl r r� r� r� rE r� rM r r Zappend_constraint)rP r� Z schema_obj�tZexclr� rV �operrH rH rR � to_constraint� s& �� � z'CreateExcludeConstraintOp.to_constraintr+ r^ r zOptional[Table])� operationsr� rh r� rE rF c O s | |||f|�}|� |�S )a� Issue an alter to create an EXCLUDE constraint using the current migration context. .. note:: This method is Postgresql specific, and additionally requires at least SQLAlchemy 1.0. e.g.:: from alembic import op op.create_exclude_constraint( "user_excl", "user", ("period", "&&"), ("group", "="), where=("group != 'some group'"), ) Note that the expressions work the same way as that of the ``ExcludeConstraint`` object itself; if plain strings are passed, quoting rules must be applied manually. :param name: Name of the constraint. :param table_name: String name of the source table. :param elements: exclude conditions. :param where: SQL expression or SQL string with optional WHERE clause. :param deferrable: optional bool. If set, emit DEFERRABLE or NOT DEFERRABLE when issuing DDL for this constraint. :param initially: optional string. If set, emit INITIALLY <value> when issuing DDL for this constraint. :param schema: Optional schema name to operate within. )�invoke)r� r� r� rh r� rE r� rH rH rR r� � s +z3CreateExcludeConstraintOp.create_exclude_constraintr* )r� r� r� rE rF c O s* |j j|d< | ||j j|f|�}|�|�S )a3 Issue a "create exclude constraint" instruction using the current batch migration context. .. note:: This method is Postgresql specific, and additionally requires at least SQLAlchemy 1.0. .. seealso:: :meth:`.Operations.create_exclude_constraint` rl )r� rl rh r� )r� r� r� r� rE r� rH rH rR r� � s z9CreateExcludeConstraintOp.batch_create_exclude_constraint)NNN)N) r� r� r� �__doc__Zconstraint_typer� � classmethodr� r� r� r� rH rH rH rR r� Q s � �-r� r= )r� r� rF c C s t |�� | dd�S )NT)�alter)�_exclude_constraintr� )r� r� rH rH rR �_add_exclude_constraint� s r� r r9 )rW r� �namespace_metadatarF c C s& t �d| |�}|dk r|S t| |d�S )Nr� F)r' Z_user_defined_renderr� )rW r� r� �renderedrH rH rR �!_render_inline_exclude_constraint� s �r� )r� rF c C s | j }|d k r|�d� dS )Nz*from sqlalchemy.dialects import postgresqlz postgresql.)�imports�add)r� r� rH rH rR � _postgresql_autogenerate_prefix� s r� r� )rW r� r� rF c s0 g �� j }�jr$��dt�j�f� �jr>��dt�j�f� �jrX��dt�j�f� |s�|r��jjr���dt� �jj�f� |s��j r���dt� � �j �f� � ��fdd�}|�rt t� � �j ��g}|s�|t t� �jj ��g7 }|� |� � dt�� �d �|�d � S |� }d t� �d �|�d � S d S ) Nr� r� rr rl rV c sN � fdd��j D �} �jd k r6| �dt��j� � � | �dd� �D �� | S )Nc s$ g | ] \}}}d t |� �|f �qS )z(%s, %r))�_render_potential_column)r� ZsqltextrV Zopstring)r� rH rR r� s ����zC_exclude_constraint.<locals>.do_expr_where_opts.<locals>.<listcomp>zwhere=%sc S s g | ]\}}d ||f �qS )z%s=%rrH )r� �k�vrH rH rR r� % s )r� r� r� r' �_render_potential_exprr� )�args�r� rW �optsrH rR �do_expr_where_opts s � ���z/_exclude_constraint.<locals>.do_expr_where_optsz-%(prefix)screate_exclude_constraint(%(args)s)z, )�prefixr z%%(prefix)sExcludeConstraint(%(args)s))Z _has_batchr� r� r^ r� rr rK rl r' �_identrV Z_render_gen_name�reprr� Z _alembic_autogenerate_prefix�joinr� )rW r� r� Z has_batchr r rH r rR r� � s< �� ��r� zGUnion[ColumnClause[Any], Column[Any], TextClause, FunctionElement[Any]])�valuer� rF c C sN t | t�r0| jrd}nd}|t�|�| jd� S tj| |t | ttf�d�S d S )Nz"%(prefix)sliteral_column(%(name)r)z%(prefix)scolumn(%(name)r))r rV )Z wrap_in_text) r] r Z is_literalr' Z _sqlalchemy_autogenerate_prefixrV r r r )r r� �templaterH rH rR r ; s � �r )l� __future__r �loggingr` �typingr r r r r r r r r Z sqlalchemyr r r r r rN Z sqlalchemy.dialects.postgresqlr r r Zsqlalchemy.schemar Zsqlalchemy.sql.elementsr r Zsqlalchemy.sql.functionsr Zsqlalchemy.typesr �baser r r r r r r r! r" r� r# r$ rY r& Z autogenerater' r� r( r) Zoperations.baser* r+ r, Zutil.sqla_compatr- r. r/ r0 Z$sqlalchemy.dialects.postgresql.arrayr1 Z#sqlalchemy.dialects.postgresql.baser2 Z%sqlalchemy.dialects.postgresql.hstorer3 Z#sqlalchemy.dialects.postgresql.jsonr4 r5 r6 r7 r8 Zsqlalchemy.sql.schemar9 r: Zsqlalchemy.sql.type_apir; r< Zautogenerate.apir= Zautogenerate.renderr> Zruntime.migrationr? � getLoggerr� r~ r@ ru r� r� r� r� Zregister_operationZAddConstraintOpZregister_add_constraintr� Z renderersZ dispatch_forr� Z_constraint_renderersr� r� r� r rH rH rH rR �<module> s� - " � ?