File: //lib/python3/dist-packages/django/db/backends/postgresql/__pycache__/schema.cpython-310.pyc
o
�aS+ � @ s@ d dl Z d dlmZ d dlmZ d dlmZ G dd� de�ZdS )� N)�BaseDatabaseSchemaEditor)� IndexColumns)� strip_quotesc
s� e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d d
� Z� fdd�Zdd� Zdd� Zdd� Z� fdd�Z d)� fdd� Z� fdd �Zd)d d �Zd)d d �Zd*� fd"d#� Zd!d!d$d$d!d%d!d%d!dd!d!d&� � fd'd(�
Z� ZS )+�DatabaseSchemaEditorz CREATE SEQUENCE %(sequence)sz,DROP SEQUENCE IF EXISTS %(sequence)s CASCADEz=SELECT setval('%(sequence)s', MAX(%(column)s)) FROM %(table)sz9ALTER SEQUENCE %(sequence)s OWNED BY %(table)s.%(column)szZCREATE INDEX %(name)s ON %(table)s%(using)s (%(columns)s)%(include)s%(extra)s%(condition)szgCREATE INDEX CONCURRENTLY %(name)s ON %(table)s%(using)s (%(columns)s)%(include)s%(extra)s%(condition)sz DROP INDEX IF EXISTS %(name)sz*DROP INDEX CONCURRENTLY IF EXISTS %(name)szyCONSTRAINT %(name)s REFERENCES %(to_table)s(%(to_column)s)%(deferrable)s; SET CONSTRAINTS %(namespace)s%(name)s IMMEDIATEzRSET CONSTRAINTS %(name)s IMMEDIATE; ALTER TABLE %(table)s DROP CONSTRAINT %(name)sz,DROP FUNCTION %(procedure)s(%(param_types)s)c C s> t |t�r |�dd�}tj�|�}t|d�rd|_|�� � � S )N�%z%%�encoding�utf8)
�
isinstance�str�replace�psycopg2�
extensions�adapt�hasattrr � getquoted�decode)�self�value�adapted� r �F/usr/lib/python3/dist-packages/django/db/backends/postgresql/schema.py� quote_value& s
z DatabaseSchemaEditor.quote_valuec s0 t � �||�}| �||�}|d ur|�|� |S �N)�super�_field_indexes_sql�_create_like_index_sql�append)r �model�field�output�like_index_statement�� __class__r r r / s
z'DatabaseSchemaEditor._field_indexes_sqlc C s. |j r |�| j�S | jj�|�� |�| j��S r )� is_relation� rel_db_type�
connection�
data_types�get�get_internal_type�db_type�r r r r r �_field_data_type6 s
�z%DatabaseSchemaEditor._field_data_typec c s8 � |j �� dkr| �|j �E d H d S | �|j �V d S )N�
ArrayField)�
base_fieldr( �_field_base_data_typesr+ r* r r r r. > s �z+DatabaseSchemaEditor._field_base_data_typesc C sr |j | jd�}|dur7|js|jr7d|v rdS |�d�r'| j||gddgd�S |�d�r7| j||gdd gd�S dS )
z�
Return the statement to create an index with varchar operator pattern
when the column type is 'varchar' or 'text', otherwise return None.
�r% N�[�varchar�_like�varchar_pattern_ops)�fields�suffix� opclasses�text�text_pattern_ops)r) r% �db_index�unique�
startswith�_create_index_sql)r r r r) r r r r E s&
�
�z+DatabaseSchemaEditor._create_like_index_sqlc s� d| _ d}|�� }|�� }|dkr+||kr+t| �|��t| �|��kr*| j |7 _ n| �|�| �|�kr<| j |7 _ t|jj�}dddd�} |�� | v r�t|j �}
d||
f } | j | �
|
�| |�� d � g f| j d
| �
| �i g f| j d
| �
| �i g f| j
| �
|�| j| �
|
�d | �
| � d � d
� g f| j| �
|�| �
|
�| �
| �d� g f| j| �
|�| �
|
�| �
| �d� g fgfS |j| jd�d | v r�t|j �}
d||
f } t� �||||�\} }
| | j d
| �
| �i g fgfS t� �||||�S )Nz%ALTER COLUMN %(column)s TYPE %(type)sz USING %(column)s::%(type)sr, �bigint�integer�smallint)� bigserial�serial� smallserialz %s_%s_seq)�column�type�sequencez
nextval('%s'))rC �default)�table�changes)rG rC rE r/ rD )�sql_alter_column_typer( �listr. r+ r �_meta�db_table�lowerrC �
quote_name�sql_delete_sequence�sql_create_sequence�sql_alter_column�sql_alter_column_default�sql_set_sequence_max�sql_set_sequence_owner�
db_parametersr% r �_alter_column_type_sql)r r � old_field� new_field�new_type� using_sql�new_internal_type�old_internal_typerG �serial_fields_maprC �
sequence_name�fragment�_r! r r rV e s� �
��
��
�� ��� �� ����1
��� z+DatabaseSchemaEditor._alter_column_type_sqlFc
s |j s|jr9|�d�r|�d�r$|�d�r|�d�r$|�d�r9|�d�s9| j|jj|jgdd�} | �| �|| �� t � �
||||||||� |j sO|jsO|j sU|jsd|jrd| � ||�}
|
d urd| �|
� |jr�|j s�|js�| j|jj|jgdd�} | �| �|| �� d S d S d S d S )Nr1 r7 �citextr2 )r5 ) r9 r: r; �_create_index_namerK rL rC �execute�_delete_index_sqlr � _alter_fieldr ) r r rW rX �old_typerY �
old_db_params�
new_db_params�strict�
index_namer �index_to_remover! r r re � s> ������ ���
�z!DatabaseSchemaEditor._alter_fieldc s* |r t ||| j||d�S t� �||||�S )N)� col_suffixesr6 )r rN r �_index_columns)r rG �columnsrl r6 r! r r rm � s z#DatabaseSchemaEditor._index_columnsc C s | j |j|| |d�d d� d S )N�� concurrently)�params)rc �
create_sql�r r �indexrp r r r � add_index� s z DatabaseSchemaEditor.add_indexc C s | � |j|| |d�� d S )Nro )rc �
remove_sqlrs r r r � remove_index� s z!DatabaseSchemaEditor.remove_indexNc s |r| j n| j}t� �|||�S r )� sql_delete_index_concurrently�sql_delete_indexr rd )r r �name�sqlrp r! r r rd � s z&DatabaseSchemaEditor._delete_index_sql� r ) r4 rz r5 �using�
db_tablespacerl r{ r6 � conditionrp �include� expressionsc s4 | s| j n| j}t� j||||||||| |
| |
d� S )N) r4 rz r5 r} r~ rl r{ r6 r r� r� )�sql_create_index� sql_create_index_concurrentlyr r< )r r r4 rz r5 r} r~ rl r{ r6 r rp r� r� r! r r r< � s
�z&DatabaseSchemaEditor._create_index_sql)F)NF)�__name__�
__module__� __qualname__rP rO rS rT r� r� ry rx �sql_create_column_inline_fk�
sql_delete_fk�sql_delete_procedurer r r+ r. r rV re rm ru rw rd r< �
__classcell__r r r! r r s> ��� T�
�r )r � django.db.backends.base.schemar �!django.db.backends.ddl_referencesr �django.db.backends.utilsr r r r r r �<module> s