File: //usr/lib/python3/dist-packages/django/db/backends/__pycache__/ddl_references.cpython-310.pyc
o
�a� � @ s� d Z ddlmZ G dd� d�ZG dd� de�ZG dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�ZG dd� de�Z G dd� de�Z
G dd� de�Z dS )z~
Helpers to manipulate deferred DDL statements that might need to be adjusted or
discarded within when executing a migration.
� )�deepcopyc @ �@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� Referencez0Base class that defines the reference interface.c C � dS )zU
Return whether or not this instance references the specified table.
F� ��self�tabler r �C/usr/lib/python3/dist-packages/django/db/backends/ddl_references.py�references_table � zReference.references_tablec C r )zV
Return whether or not this instance references the specified column.
Fr �r r �columnr r r
�references_column r zReference.references_columnc C r )zI
Rename all references to the old_name to the new_table.
Nr �r � old_table� new_tabler r r
�rename_table_references r z!Reference.rename_table_referencesc C r )zL
Rename all references to the old_column to the new_column.
Nr �r r �
old_column�
new_columnr r r
�rename_column_references r z"Reference.rename_column_referencesc C s d| j jt| �f S )Nz<%s %r>)� __class__�__name__�str�r r r r
�__repr__# � zReference.__repr__c C s t d��)Nz>Subclasses must define how they should be converted to string.)�NotImplementedErrorr r r r
�__str__& s zReference.__str__N)
r �
__module__� __qualname__�__doc__r r r r r r r r r r
r s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �Tablez Hold a reference to a table.c C � || _ || _d S �N)r �
quote_name)r r r& r r r
�__init__- �
zTable.__init__c C s
| j |kS r% �r r r r r
r 1 s
zTable.references_tablec C s | j |kr
|| _ d S d S r% r) r r r r
r 4 s
�z Table.rename_table_referencesc C s | � | j�S r% )r& r r r r r
r 8 � z
Table.__str__N)r r r! r" r' r r r r r r r
r# * s r# c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � TableColumnsz9Base class for references to multiple columns of a table.c C r$ r% �r �columns)r r r- r r r
r' ? r( zTableColumns.__init__c C s | j |ko || jv S r% r, r
r r r
r C r z TableColumns.references_columnc C s8 | j |krt| j�D ]\}}||kr|| j|< q
d S d S r% )r � enumerater- )r r r r �indexr r r r
r F s
��z%TableColumns.rename_column_referencesN)r r r! r" r' r r r r r r
r+ < s
r+ c s* e Zd ZdZd� fdd� Zdd� Z� ZS )�Columnsz(Hold a reference to one or many columns.r c � || _ || _t� �||� d S r% )r& � col_suffixes�superr' )r r r- r& r2 �r r r
r' P � zColumns.__init__c �* �fdd�� d� � fdd�t�j�D ��S )Nc sF � � | �}z� j| }|rd�||�}W |S W |S ty" Y |S w �Nz{} {})r& r2 �format�
IndexError�r �idx�col�suffixr r r
�col_strV s
� ��z Columns.__str__.<locals>.col_str�, c 3 � � | ]
\}}� ||�V qd S r% r ��.0r; r �r> r r
� <genexpr>` � � z"Columns.__str__.<locals>.<genexpr>��joinr. r- r r �r> r r
r U s
zColumns.__str__)r �r r r! r" r' r �
__classcell__r r r4 r
r0 M s r0 c s( e Zd ZdZ� fdd�Zdd� Z� ZS )� IndexNamez"Hold a reference to an index name.c r1 r% )r= �create_index_namer3 r' )r r r- r= rL r4 r r
r' f r5 zIndexName.__init__c C s | � | j| j| j�S r% )rL r r- r= r r r r
r k r zIndexName.__str__rI r r r4 r
rK c s rK c s& e Zd Zd� fdd� Zdd� Z� ZS )� IndexColumnsr c s || _ t� �||||� d S r% )� opclassesr3 r' )r r r- r& r2 rN r4 r r
r' p s zIndexColumns.__init__c r6 )Nc sT d� � �| �� j| �}z� j| }|r d� ||�}W |S W |S ty) Y |S w r7 )r8 r& rN r2 r9 r: r r r
r> u s
� ��z%IndexColumns.__str__.<locals>.col_strr? c 3 r@ r% r rA rC r r
rD � rE z'IndexColumns.__str__.<locals>.<genexpr>rF r r rH r
r t s zIndexColumns.__str__)r r )r r r! r' r rJ r r r4 r
rM o s rM c sX e Zd ZdZ� fdd�Z� fdd�Z� fdd�Z� fdd �Z� fd
d �Zd d
� Z � Z
S )�ForeignKeyNamez'Hold a reference to a foreign key name.c s* t ||�| _|| _|| _t� �||� d S r% )r+ � to_reference�suffix_template�create_fk_namer3 r' )r �
from_table� from_columns�to_table�
to_columnsrQ rR r4 r r
r' � s zForeignKeyName.__init__c s t � �|�p | j�|�S r% )r3 r rP r r4 r r
r � s z ForeignKeyName.references_tablec s t � �||�p
| j�||�S r% )r3 r rP r
r4 r r
r � s �z ForeignKeyName.references_columnc s t � �||� | j�||� d S r% )r3 r rP r r4 r r
r � s z&ForeignKeyName.rename_table_referencesc s$ t � �|||� | j�|||� d S r% )r3 r rP r r4 r r
r � s z'ForeignKeyName.rename_column_referencesc C s. | j | jj| jjd d� }| �| j| j|�S )Nr )rU � to_column)rQ rP r r- rR )r r= r r r
r � s
�zForeignKeyName.__str__) r r r! r" r' r r r r r rJ r r r4 r
rO � s rO c @ r )� Statementz�
Statement template and formatting parameters container.
Allows keeping a reference to a statement without interpolating identifiers
that might have to be adjusted if they're referencing a table or column
that is removed
c K r$ r% ��template�parts)r rZ r[ r r r
r' � r( zStatement.__init__c s t � fdd�| j�� D ��S )Nc 3 s$ � | ]
}t |d �o
|�� �V qdS )r N)�hasattrr �rB �partr) r r
rD � s
� �
�z-Statement.references_table.<locals>.<genexpr>��anyr[ �valuesr r r) r
r � s �zStatement.references_tablec s t � �fdd�| j�� D ��S )Nc 3 s& � | ]}t |d �o|��� �V qdS )r N)r\ r r] �r r r r
rD � s
� �
�z.Statement.references_column.<locals>.<genexpr>r_ r
r rb r
r � s �zStatement.references_columnc C s* | j �� D ]
}t|d�r|�||� qd S )Nr )r[ ra r\ r )r r r r^ r r r
r � s
��z!Statement.rename_table_referencesc C s, | j �� D ]}t|d�r|�|||� qd S )Nr )r[ ra r\ r )r r r r r^ r r r
r � s
��z"Statement.rename_column_referencesc C s | j | j S r% rY r r r r
r � r* zStatement.__str__N)
r r r! r" r' r r r r r r r r r
rX � s rX c s8 e Zd Z� fdd�Z� fdd�Zdd� Zdd� Z� ZS ) � Expressionsc s@ || _ || _|| _dd� | j j�| jg�D �}t� �||� d S )Nc S s g | ]}|j j�qS r )�targetr )rB r< r r r
�
<listcomp>� s z(Expressions.__init__.<locals>.<listcomp>)�compiler� expressions� quote_value�query� _gen_colsr3 r' )r r rg rf rh r- r4 r r
r' � s
zExpressions.__init__c s2 | j |krd S | j�||i�| _t� �||� d S r% )r rg �relabeled_cloner3 r r r4 r r
r � s
z#Expressions.rename_table_referencesc C sb | j |krd S t| j�}g | _| jj�|g�D ]}|jj|kr#||j_| j� |jj� q|| _d S r% )
r r rg r- rf ri rj rd r �append)r r r r rg r< r r r
r � s
z$Expressions.rename_column_referencesc C s* | j �| j�\}}t| j|�}|t|� S r% )rf �compilerg �maprh �tuple)r �sql�paramsr r r
r � s zExpressions.__str__)r r r! r' r r r rJ r r r4 r
rc � s
rc N) r" �copyr r r# r+ r0 rK rM rO rX rc r r r r
�<module> s " "&