File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/testing/__pycache__/schema.cpython-310.pyc
o
���gq � @ s� d dl mZ d dlZddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
l
m Z dd l
mZ d d
gZi Zd dd �Zdd� Z dd
� Zdd� ZG dd� d�ZG dd� d�ZG dd� d�Zd d � Zd d � ZdS )!� )� annotationsN� )�config)�
exclusions� )�event)�schema)�types)�
mapped_column)� OrderedDict�Table�Column�return� schema.Tablec s� � fdd�t � �D �}� �t� t�tjd�r5d� vr4d� vr4d� vr4d|v s+d|v r0d � d< n)d
� d< n$t�tjd �rYd � vrYd
� vrYd� vrYd|v sPd|v rUd � d < nd
� d < tj| i � ��S )z8A schema.Table wrapper/hook for dialect-specific tweaks.c �" i | ]
}|� d �r|� �|��qS ��test_��
startswith�pop��.0�k��kw� �D/usr/local/lib/python3.10/dist-packages/sqlalchemy/testing/schema.py�
<dictcomp> � " zTable.<locals>.<dictcomp>�mysql� mysql_engine�
mysql_type�
autoload_with�
test_needs_fk�test_needs_acid�InnoDB�MyISAM�mariadb�mariadb_engine� mariadb_type) �list�update�
table_optionsr �againstr �_currentr r )�argsr � test_optsr r r r s$
�
c O s t t| |�S )z>An orm.mapped_column wrapper/hook for dialect-specific tweaks.)�_schema_column�_orm_mapped_column�r/ r r r r r
? s r
c O s t tj| |�S )z9A schema.Column wrapper/hook for dialect-specific tweaks.)r1 r r
r3 r r r r
E s c s� � fdd�t � �D �}tjj�t�sdd� |D �}| |i � ��}| tju r(|}n|j}|�dd�r]� �dd�r]|j d u rD|j
d u rDd|_ d|j d< t
�tjd �r]d
d � }tj|d |dd
� |S )Nc r r r r r r r r L r z"_schema_column.<locals>.<dictcomp>c S s g | ]
}t |tj�s|�qS r )�
isinstancer �
ForeignKey)r �argr r r �
<listcomp>O s z"_schema_column.<locals>.<listcomp>�test_needs_autoincrementF� primary_keyT�oraclec S s2 | � tjttjj|jd | j d �dd�� d S )N�_�_seqT)�optional)� _init_itemsr �Sequence�_truncate_namer �db�dialect�name)�c�tblr r r �add_seqe s ���z _schema_column.<locals>.add_seq�after_parent_attach)� propagate)r* r � requirements�foreign_key_ddl�enabled_for_configr r
�column�get�default�server_default�
autoincrement�infor r- r. r �listen)�factoryr/ r r0 � construct�colrF r r r r1 K s"
�
r1 c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) �eq_type_affinitya� Helper to compare types inside of datastructures based on affinity.
E.g.::
eq_(
inspect(connection).get_columns("foo"),
[
{
"name": "id",
"type": testing.eq_type_affinity(sqltypes.INTEGER),
"nullable": False,
"default": None,
"autoincrement": False,
},
{
"name": "data",
"type": testing.eq_type_affinity(sqltypes.NullType),
"nullable": True,
"default": None,
"autoincrement": False,
},
],
)
c C s t �|�| _d S �N)�sqltypes� to_instance�target��selfr[ r r r �__init__� s zeq_type_affinity.__init__c C s | j j|ju S rX �r[ �_type_affinity�r] �otherr r r �__eq__� � zeq_type_affinity.__eq__c C s | j j|juS rX r_ ra r r r �__ne__� rd zeq_type_affinity.__ne__N��__name__�
__module__� __qualname__�__doc__r^ rc re r r r r rW s s
rW c @ rV ) �eq_compile_typez,similar to eq_type_affinity but uses compilec C �
|| _ d S rX �r[ r\ r r r r^ � �
zeq_compile_type.__init__c C s | j |�� kS rX �r[ �compilera r r r rc � rd zeq_compile_type.__eq__c C s | j |�� kS rX ro ra r r r re � rd zeq_compile_type.__ne__Nrf r r r r rk � �
rk c @ rV ) �eq_clause_elementz3Helper to compare SQL structures based on compare()c C rl rX rm r\ r r r r^ � rn zeq_clause_element.__init__c C s | j �|�S rX �r[ �comparera r r r rc � s zeq_clause_element.__eq__c C s | j �|� S rX rs ra r r r re � rd zeq_clause_element.__ne__Nrf r r r r rr � rq rr c C sF t |�| jkr!|dt| jd d�� d tt|�d �dd � S |S )Nr � r; �@ r )�len�max_identifier_length�max�hex�hash)rB rC r r r r@ � s ���r@ c s� t � }d � fdd� }i � t� fdd��}t| tf|||d��}z t�d�j�dd�}W n tt fy6 Y nw |d ur>||_
|S )
Nc sP || _ || _| | j|< | � |< t| j|| � |r&| | j|< t| j|| � d S d S rX )rC �value� __members__�setattr� __class__)r] rC r| �alias��value_to_memberr r r^ � s
�z pep435_enum.<locals>.__init__c s � | S rX r )�clsr| r� r r rM � s zpep435_enum.<locals>.get)r} r^ rM r rg �__main__rX ) r � classmethod�type�object�sys� _getframe� f_globalsrM �AttributeError�
ValueErrorrh )rC r} r^ rM �someenum�moduler r� r � pep435_enum� s$
��r� )r r )�
__future__r r� � r r r r r rY �ormr
r2 �utilr �__all__r, r r
r1 rW rk rr r@ r� r r r r �<module> s(
%(%