File: //lib/python3/dist-packages/django/db/backends/sqlite3/__pycache__/introspection.cpython-310.pyc
o
�a�O � @ s� d dl Z d dlmZ d dlZd dlmZmZmZ d dl m
Z
d dl m Z edej
d �Ze d�Zd d
� ZG d d � d �ZG d
d� de�ZdS )� N)�
namedtuple)�BaseDatabaseIntrospection� FieldInfo� TableInfo)�Index)�_lazy_re_compiler )�pk�has_json_constraintz&^\s*(?:var)?char\s*\(\s*(\d+)\s*\)\s*$c C s t �| �}|r
t|d �S dS )z8 Extract the size number from a "varchar(11)" type name � N)�
field_size_re�search�int)�name�m� r �J/usr/lib/python3/dist-packages/django/db/backends/sqlite3/introspection.py�get_field_size s
r c @ s� e Zd Zi dd�dd�dd�dd�dd�d d
�d d
�d d
�dd�dd�dd�dd�dd�dd�dd�dd �d d �d d d!��Zd"d#� Zd$S )%�FlexibleFieldLookupDict�bool� BooleanField�boolean�smallint�SmallIntegerFieldzsmallint unsigned�PositiveSmallIntegerField� smallintegerr
� IntegerField�integer�bigint�BigIntegerFieldzinteger unsigned�PositiveIntegerFieldzbigint unsigned�PositiveBigIntegerField�decimal� DecimalField�real�
FloatField�text� TextField�char� CharField�varchar�blob� BinaryField�date� DateField�
DateTimeField� TimeField)�datetime�timec C s" |� � �dd�d �� }| j| S )N�(r
r )�lower�split�strip�base_data_types_reverse)�self�keyr r r � __getitem__4 s
z#FlexibleFieldLookupDict.__getitem__N)�__name__�
__module__� __qualname__r6 r9 r r r r r sP �������� �
� � �
������ r c s� e Zd Ze� Z� fdd�Zdd� Zdd� Zd dd �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Z� ZS ) �DatabaseIntrospectionc s. t � �||�}|jr|dv rdS |jrdS |S )N> r r r � AutoField� JSONField)�super�get_field_typer r )r7 � data_type� description�
field_type�� __class__r r rA < s z$DatabaseIntrospection.get_field_typec C s |� d� dd� |�� D �S )z>Return a list of table and view names in the current database.z�
SELECT name, type FROM sqlite_master
WHERE type in ('table', 'view') AND NOT name='sqlite_sequence'
ORDER BY namec S s" g | ]
}t |d |d d ��qS )r r
)r )�.0�rowr r r �
<listcomp>N s " z8DatabaseIntrospection.get_table_list.<locals>.<listcomp>)�execute�fetchall)r7 �cursorr r r �get_table_listF s
z$DatabaseIntrospection.get_table_listc s� |� d| jj�|� � |�� }| �||�� t� �| jjjr<|D ]}|d }d| }|� d||g�� � }|r;��
|� q � �fdd�|D �S )zi
Return a description of the table with the DB-API cursor.description
interface.
zPRAGMA table_info(%s)r
z%%json_valid("%s")%%z�
SELECT sql
FROM sqlite_master
WHERE
type = 'table' AND
name = %s AND
sql LIKE %s
c sH g | ] \}}}}}}t ||d t|�d d | |� �|�|dk|�v � �qS )Nr
)r r �get)rG �cidr rB �notnull�defaultr ��
collations� json_columnsr r rI g s ���z?DatabaseIntrospection.get_table_description.<locals>.<listcomp>) rJ �
connection�ops�
quote_namerK �_get_column_collations�set�features�can_introspect_json_field�fetchone�add)r7 rL �
table_name�
table_info�line�column�json_constraint_sqlr r rR r �get_table_descriptionP s&
��
� �z+DatabaseIntrospection.get_table_descriptionr c C s | � ||�}||d�gS )N)�tablera )�get_primary_key_column)r7 rL r^ � table_fields�pk_colr r r �
get_sequenceso s z#DatabaseIntrospection.get_sequencesc C st i }|� d|g� |�� \}}|dkr|S ||�d�d |�d�� }|�d�D ]�}|�� }|�d�r4q(t�d|tj �}|s?q(d d
� |�
� D �\} }
|�d �r_t� d |tj �}|d �d
�} n |�� d �d
�} |� d| g� |� � d } | d �� }
|
�d�|
�d�}}|
|d |� }
|
�d�D ]#}|�� }|�d�r�q�|�dd�d �d
�}||
kr�|| f|| < nq�q(|S )z�
Return a dictionary of {field_name: (field_name_other_table, other_table)}
representing all relationships to the given table.
�USELECT sql, type FROM sqlite_master WHERE tbl_name = %s AND type IN ('table', 'view')�viewr2 r
�)�,�UNIQUEz$references (\S*) ?\(["|]?(.*)["|]?\)c S s g | ]}|� d ��qS )�"�r5 �rG �sr r r rI � s z7DatabaseIntrospection.get_relations.<locals>.<listcomp>z FOREIGN KEYz FOREIGN KEY\s*\(([^\)]*)\).*rn r z1SELECT sql FROM sqlite_master WHERE tbl_name = %s� )
rJ r\ �index�rindexr4 r5 �
startswith�rer �I�groups�matchrK )r7 rL r^ � relations�
create_sql�
table_type�results�
field_descr rd ra �
field_name�result�other_table_results�li�ri�
other_desc�
other_namer r r �
get_relationss sJ �
��z#DatabaseIntrospection.get_relationsc C s� g }|� d|dg� |�� d �� }||�d�d |�d�� }t|�d��D ]'\}}|�� }|�d�r5q't� d |tj
�}|s@q'|� t d
d � |�
� D ��� q'|S ) z�
Return a list of (column_name, referenced_table_name, referenced_column_name)
for all key columns in given table.
z?SELECT sql FROM sqlite_master WHERE tbl_name = %s AND type = %srd r r2 r
rk rl rm z("(.*)".*references (.*) \(["|](.*)["|]\)c s s � | ]}|� d �V qdS )rn Nro rp r r r � <genexpr>� � � z8DatabaseIntrospection.get_key_columns.<locals>.<genexpr>)rJ r\ r5 rs rt � enumerater4 ru rv r rw �append�tuplerx )r7 rL r^ � key_columnsr} � field_indexr~ r r r r �get_key_columns� s
z%DatabaseIntrospection.get_key_columnsc C s� |� d|g� |�� }|du rtd| ��|\}}|dkr dS ||�d�d |�d�� }|�d�D ] }|�� }t�d |�}|rP|d rJ|d S |d
S q2dS ) z>Return the column name of the primary key for the given table.ri NzTable %s does not existrj r2 r
rk rl z1(?:(?:["`\[])(.*)(?:["`\]])|(\w+)).*PRIMARY KEY.*� ) rJ r\ �
ValueErrorrs rt r4 r5 rv ry ) r7 rL r^ rH r{ r| �
fields_sqlr~ r r r r re � s$ � �z,DatabaseIntrospection.get_primary_key_columnc
C sd i }|� d| jj�|� � |�� D ] }|d d� \}}}}} |gdd|| fddd�|d| < q|S )NzPRAGMA foreign_key_list(%s)� F��columns� primary_key�unique� foreign_key�checkrs zfk_%d)rJ rU rV rW rK )
r7 rL r^ � constraintsrH �id_�_rd �from_�tor r r � _get_foreign_key_constraints� s �z2DatabaseIntrospection._get_foreign_key_constraintsc C s� d }d }d }d }d}g }d} g }
d} |D �]!}|� tjjd�r$| d7 } n"|� tjjd�r8| d8 } | dk r7 �n n| dkrF|� tjjd�rF n�|d u rU|� tjjd�}|rUq|r�|d u rz|jtjjtjjfv rj|j}n|jtjjj j
krz|jdd� }|� tjjd �r�d
}| } n_|r�| | kr�|r�d}q|jtjjtjjfv r�|� |j� nB|jtjjj j
kr�|� |jdd� � n.|d u r�|jtjjtjjfv r�|j}n|jtjjj j
kr�|jdd� }|� tjjd �r�|g}|� tjjd �r�d
} | }
q| �r6|
| kr�|
r�d} q|jtjjtjjfv �r|j|v �r|
� |j� q|jtjjj j
k�r6|jdd� |v �r6|
� |jdd� � q|�rCd
|dd ddd �nd }|
�rQd
|
ddd dd
�nd }||||fS )NFr r2 r
rk rl �
CONSTRAINT���rm T�CHECK)r� r� r� r� r� rs )r� r� r� r� r� rs ) ry �sqlparse�tokens� Punctuation�Keyword�ttype�Name�value�Literal�String�Symbolr� )r7 r� r� �token�is_constraint_definitionr �constraint_namer� �unique_columnsr� �
check_columns� braces_deep�unique_braces_deep�check_braces_deep�unique_constraint�check_constraintr r r �&_parse_column_or_constraint_definition� s�
�� �������� z<DatabaseIntrospection._parse_column_or_constraint_definitionc C s� t �|�d }i }d}dd� |�� D �}|D ] }|�t jjd�r" nq | �||�\}} }
} | rA|r7| ||< n
|d7 }| |d| < |
rT|rJ|
||< n
|d7 }|
|d| < | �t jjd�r_ |S q$) Nr c s s � | ]}|j s|V qd S �N)�
is_whitespace)rG r� r r r r� Q r� zADatabaseIntrospection._parse_table_constraints.<locals>.<genexpr>r2 Tr
z__unnamed_constraint_%s__rk )r� �parse�flattenry r� r� r� ) r7 �sqlr� � statementr� �unnamed_constrains_indexr� r� r� r� r� � end_tokenr r r �_parse_table_constraintsK s0 �
�z.DatabaseIntrospection._parse_table_constraintsc C s� i }z|� d| jj�|�f ��� d }W n ty Y nw dd� | �||�D �}|�| �||�� |� d| jj�|� � |� � D ]m}|dd� \}}} |� d| jj�|� � |�� p_d \}
|
sdqC|� d
| jj�|� � |� � D ] \} } }
||vr�g d t
| �dd d d
�||< || d � |
� qt|| d r�t j
|| d< | �|
�}|dur�||| d< qC| �||�}|r�|gd d dd d d
�|d< |�| �||�� |S )zu
Retrieve any constraints or keys (unique, pk, fk, check, index) across
one or more columns.
z<SELECT sql FROM sqlite_master WHERE type='table' and name=%sr c S s h | ]}|j �qS r )r �rG �infor r r � <setcomp>z s z8DatabaseIntrospection.get_constraints.<locals>.<setcomp>zPRAGMA index_list(%s)N� z<SELECT sql FROM sqlite_master WHERE type='index' AND name=%sr� zPRAGMA index_info(%s)FTr� r� rs �type�orders� __primary__)rJ rU rV rW r\ � TypeErrorrc �updater� rK r r� r �suffix�_get_index_columns_ordersre r� )r7 rL r^ r� � table_schemar� rH �numberrs r� r� �
index_rank� column_rankra r� � pk_columnr r r �get_constraintsi sj ��� � ��
�
�
�z%DatabaseIntrospection.get_constraintsc C sP t �|�d }|D ] }t|t jj�r%t|��d��d�}dd� |D � S q d S )Nr �()�, c S s g | ] }|� d �r d nd�qS )�DESC�ASC)�endswithr� r r r rI � s zCDatabaseIntrospection._get_index_columns_orders.<locals>.<listcomp>)r� r� �
isinstancer� � Parenthesis�strr5 r4 )r7 r� r� r� r� r r r r� � s �z/DatabaseIntrospection._get_index_columns_ordersc
C s� |� d|g��� }|s
i S |d }tt�|�d d ��d��d�}i }|D ],}|dd � �� }|d �d�} t|�D ]\}
} | dkrK||
d } nq;d } | || < q&|S ) Nzn
SELECT sql
FROM sqlite_master
WHERE type = 'table' AND name = %s
r r� r� r� r
rn �COLLATE)rJ r\ r� r� r� r5 r4 r� )
r7 rL r^ rH r� r� rS ra r� � column_namers r� � collationr r r rX � s* ��" �
z,DatabaseIntrospection._get_column_collations)r )r: r; r< r �data_types_reverserA rM rc rh r� r� re r� r� r� r� r� rX �
__classcell__r r rE r r= 9 s
: Z Or= )rv � collectionsr r� �%django.db.backends.base.introspectionr r �
BaseFieldInfor �django.db.modelsr �django.utils.regex_helperr �_fieldsr r r r= r r r r �<module> s