o
    [h                     @   sX   d dl mZ d dl mZ ddl mZ ddl mZ ddgZi Zdd Zd	d Zd
d Z	dS )   )config)
exclusions   )event)schemaTableColumnc            	         s   fddt  D } t ttjdr4d vr4d vr4d vr4d|v s+d|v r0d	 d< nd
 d< ttjdry| d }tjjj	j
}dd | D }|D ]*}|j}t|tjr^|jj}n||d }||krx|jdu rpd|_|jdu rxd|_qNtj| i  S )z8A schema.Table wrapper/hook for dialect-specific tweaks.c                    "   i | ]}| d r| |qS Ztest_
startswithpop.0kkw /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/sqlalchemy/testing/schema.py
<dictcomp>      " zTable.<locals>.<dictcomp>mysqlZmysql_engineZ
mysql_typeZautoload_withZtest_needs_fkZtest_needs_acidZInnoDBZMyISAMfirebird    c                 S   s(   g | ]}t |tjr|jD ]}|qqS r   )
isinstancer   r   Zforeign_keys)r   colfkr   r   r   
<listcomp>-   s    
zTable.<locals>.<listcomp>NZCASCADE)listupdatetable_optionsr   againstr   _currentdbdialectZidentifier_preparerZunformat_identifiersZ_colspecr   r   r   tablenameZondeleteZonupdater   )	argsr   	test_optsZ
table_nameunpackZfksr   refr&   r   r   r   r      s6   




c                     s    fddt  D }tjjtsdd | D } tj| i  }|ddrT ddrT|jdu r:|j	du r:d	|_
d	|jd< ttjd
drTdd }tj|d|d	d |S )z9A schema.Column wrapper/hook for dialect-specific tweaks.c                    r	   r
   r   r   r   r   r   r   J   r   zColumn.<locals>.<dictcomp>c                 S   s   g | ]
}t |tjs|qS r   )r   r   Z
ForeignKey)r   argr   r   r   r   M   s    zColumn.<locals>.<listcomp>Ztest_needs_autoincrementFZprimary_keyNTr   oraclec                 S   s2   |  tjttjj|jd | j d dd d S )N_Z_seqT)optional)Z_init_itemsr   Sequence_truncate_namer   r#   r$   r&   )cZtblr   r   r   add_seq^   s   zColumn.<locals>.add_seqZafter_parent_attach)	propagate)r   r   requirementsZforeign_key_ddlZenabled_for_configr   r   getdefaultZserver_defaultZautoincrementinfor   r!   r"   r   listen)r'   r   r(   r   r2   r   r   r   r   G   s   

c                 C   sF   t || jkr!|dt| jd d d tt|d dd   S |S )Nr      r-   @   r   )lenZmax_identifier_lengthmaxhexhash)r$   r&   r   r   r   r0   l   s   r0   N)
 r   r   r   r   __all__r    r   r   r0   r   r   r   r   <module>   s   4%