U
|
¸ý°dS ã @ sØ d dl mZ d dlZddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
|
l
|
m Z dd l mZ d d gZi Zdddd Zdd Z dd Zdd ZG dd dZG dd dZG dd dZd d Zd d ZdS ) é )Ú annotationsNé )Úconfig)Ú
|
exclusionsé )Úevent)Úschema)Útypes)Ú mapped_column)Ú OrderedDictÚTableÚColumnz schema.Table)Úreturnc s¾ fddt D } t¡ t tjd¡rjd kr²d kr²d kr²d|ksVd|kr`d d< q²d
|
d< nHt tjd ¡r²d kr²d kr²d kr²d|ks d|krªd d < nd
|
d < tj| S )z8A schema.Table wrapper/hook for dialect-specific tweaks.c s" i | ]}| d ¡r| |¡qS ©Ztest_©Ú
|
startswithÚpop©Ú.0Úk©Úkw© úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/schema.pyÚ
|
<dictcomp> s
|
zTable.<locals>.<dictcomp>ZmysqlZ mysql_engineZ
|
mysql_typeZ autoload_withZ test_needs_fkZtest_needs_acidZInnoDBZMyISAMZmariadbZmariadb_engineZ 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.)r" r r r$ r r r r E s c sº fddt D }tjj t¡s2dd |D }| | }| tjkrL|}n|j}| dd¡r¶ dd¡r¶|j d kr|j
|
d krd|_ d|j d< t tjd ¡r¶d
|
d }tj|d |dd |S )Nc s" i | ]}| d ¡r| |¡qS r r r r r r r L s
|
z"_schema_column.<locals>.<dictcomp>c S s g | ]}t |tjs|qS r )Ú
|
isinstancer Z
|
ForeignKey)r Úargr r r Ú
|
<listcomp>O s z"_schema_column.<locals>.<listcomp>Ztest_needs_autoincrementFZ primary_keyTZoraclec S s2 | tjttjj|jd | j d dd¡ d S )NÚ_Z_seqT)Úoptional)Z _init_itemsr ÚSequenceÚ_truncate_namer ÚdbÚdialectÚname)ÚcZtblr r r Úadd_seqf s ÿüÿz _schema_column.<locals>.add_seqZafter_parent_attach)Ú propagate)r r Ú requirementsZforeign_key_ddlZenabled_for_configr r ÚcolumnÚgetÚdefaultZserver_defaultZ autoincrementÚinfor r r r Úlisten)Úfactoryr r r! Z constructÚcolr0 r r r r" K s$
|
|
ÿ
|
|
r" c @ s( 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)ÚsqltypesZ to_instanceÚtarget©Úselfr= r r r Ú__init__ s zeq_type_affinity.__init__c C s | j j|jkS r; ©r= Z_type_affinity©r? Úotherr r r Ú__eq__ s zeq_type_affinity.__eq__c C s | j j|jk S r; rA rB r r r Ú__ne__ s zeq_type_affinity.__ne__N©Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r@ rD rE r r r r r: t s r: c @ s( e Zd ZdZdd Zdd Zdd ZdS ) Úeq_compile_typez,similar to eq_type_affinity but uses compilec C s
|
|| _ d S r; ©r= r> r r r r@ s zeq_compile_type.__init__c C s | j | ¡ kS r; ©r= ÚcompilerB r r r rD s zeq_compile_type.__eq__c C s | j | ¡ kS r; rM rB r r r rE ¢ s zeq_compile_type.__ne__NrF r r r r rK s rK c @ s( e Zd ZdZdd Zdd Zdd ZdS ) Úeq_clause_elementz3Helper to compare SQL structures based on compare()c C s
|
|| _ d S r; rL r> r r r r@ © s zeq_clause_element.__init__c C s | j |¡S r; ©r= ÚcomparerB r r r rD ¬ s zeq_clause_element.__eq__c C s | j |¡ S r; rP rB r r r rE ¯ s zeq_clause_element.__ne__NrF r r r r rO ¦ s rO c C sJ t || jkrB|dt| jd d
d tt|d dd
S |S d S )Nr é r( é@ r )ÚlenZmax_identifier_lengthÚmaxÚhexÚhash)r- r. r r r r+ ³ s ÿþÿr+ c s t }d fdd }i t fdd}t| tf|||d}zt d¡j dd¡}W n tt fk
|
rn Y nX |d k r~||_
|
|S )
|
Nc sL || _ || _| | j|< | |< t| j|| |rH| | j|< t| j|| d S r; )r. ÚvalueÚ __members__ÚsetattrÚ __class__)r? r. rX Úalias©Zvalue_to_memberr r r@ Â s
|
|
z pep435_enum.<locals>.__init__c s | S r; r )ÚclsrX r] r r r4 Î s zpep435_enum.<locals>.get)rY r@ r4 r rG Ú__main__)N) r Ú classmethodÚtypeÚobjectÚsysÚ _getframeÚ f_globalsr4 ÚAttributeErrorÚ
|
ValueErrorrH )r. rY r@ r4 ZsomeenumÚmoduler r] r Ú pep435_enum¾ s"
|
|
ýri )Ú
|
__future__r rc Ú r r r r r r< Zormr
|
r# Úutilr Ú__all__r r r r" r: rK rO r+ ri r r r r Ú<module> s& %)%
|
