File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/sql/__pycache__/naming.cpython-310.pyc
o
���g� �
@ s" d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z ddl
m Z dd l
m Z dd
l
m
Z
dd l
mZ dd l
mZ dd
l
mZ ddl
mZ ddl
mZ ddlmZ ddlmZ G dd� d�Zedede dedediZdd� Zd d � Ze�ed �d d � �Ze�e
d!�e�ed!�d"d#� ��ZdS )$z5Establish constraint and index naming conventions.
� )� annotationsN� )�events)�
_NONE_NAME)�conv)�CheckConstraint)�Column)�
Constraint)�ForeignKeyConstraint)�Index)�PrimaryKeyConstraint)�Table)�UniqueConstraint� )�event)�excc @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�ConventionDictc C s* || _ t|t�| _|| _|| _|j| _d S �N)�const�
isinstancer
�_is_fk�table�
convention�name� _const_name)�selfr r r � r �@/usr/local/lib/python3.10/dist-packages/sqlalchemy/sql/naming.py�__init__" s
zConventionDict.__init__c C s | j jS r )r r �r r r r �_key_table_name) s z ConventionDict._key_table_namec C sp | j r z| jj| }W n
ty Y dS w t|j|�S t| jj�}z|| }W n
ty2 Y dS w t||�S )N� )r r �elements�
IndexError�getattr�parent�list�columns)r �idx�attrname�fk�cols�colr r r � _column_X, s � �
zConventionDict._column_Xc C s2 | j d tfv r t�d��t| j t�sd | j_| j S )NzcNaming convention including %(constraint_name)s token requires that constraint is explicitly named.)r r r �InvalidRequestErrorr r r r r r r r �_key_constraint_name= s � z#ConventionDict._key_constraint_namec C � | � |d�S )N�key�r- �r r( r r r �_key_column_X_keyH s z ConventionDict._key_column_X_keyc C r0 )Nr r2 r3 r r r �_key_column_X_nameN � z!ConventionDict._key_column_X_namec C r0 )N�
_ddl_labelr2 r3 r r r �_key_column_X_labelQ r6 z"ConventionDict._key_column_X_labelc C s> | j jd }|j�d�}t|�dkr|\}}}|S |\}}|S )Nr �.� )r r" �target_fullname�split�len)r r* �refs� refschema�reftable�refcolr r r �_key_referred_table_nameT s
�z'ConventionDict._key_referred_table_namec C s | j j| }|jjS r )r r" �columnr )r r( r* r r r �_key_referred_column_X_name] s z*ConventionDict._key_referred_column_X_namec
C s, || j v r| j | | j| j�S t| d| �r t| d| �� S t�d|�}|r�|�d�}|�d�}|rz| jr:| jj }nt
| jj �}g }t |�D ]%\}}d|�
d| d� }z |�t| |�|�� W qF tyk t|��w |�d�rsdnd } | �|�S d|�
|d� }t|�}t| |�r�t| |�|�S t|��)
Nz_key_%sz.*_?column_(\d+)(_?N)?_.+r r �_key_�0�X�_r! )r r r �hasattrr$ �re�match�groupr r" r& r' � enumerate�replace�append�AttributeError�KeyError�
startswith�join�int)
r r1 � col_templater( � multiples�elems�tokens�elem�attr�sepr r r � __getitem__d s6
�
zConventionDict.__getitem__N)
�__name__�
__module__� __qualname__r r r- r/ r4 r5 r8 rB rD r\ r r r r r ! s r �ix�pk�ck�uqr* c C sH |j D ] }|tv rt| | v r| t| S || v r!| | S qd S r )�__mro__� _prefix_dict)�dict_r1 �super_r r r �_get_convention� s
�rh c C s~ |j }t|jt| ��}t| jt�r| jS |d ur7t| jt�s7| jd u s,d|v s,| jtu r7t|t| ||j� �S |tu r=d S d S )N�constraint_name) �metadatarh �naming_convention�typer r r r r )r r rj r r r r �_constraint_name_for_table� s$
�
���rm �'_sa_event_column_added_to_pk_constraintc C s2 | j r| j}d | _t| |�}|r|| _d S d S d S r )�_implicit_generatedr r rm )�
pk_constraintr, r �newnamer r r � _column_added_to_pk_constraint� s
�rr �after_parent_attachc sh t |t�rt�|d� fdd�� d S t |t�r0t � jt�s"� jtu r$d S t� |�}|r2|� _d S d S d S )Nrs c s
t � |�S r )�_constraint_name)r, r �r r r �<lambda>� s
z"_constraint_name.<locals>.<lambda>) r r r �listenr
r r r rm )r r rq r ru r rt � s
�
�rt ) �__doc__�
__future__r rJ r! r �baser r" r �schemar r r r
r r r
r r r r re rh rm � listens_forrr rt r r r r �<module> s@ e�
�