U
|
¸ý°d§ ã
|
@ 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 dZedede 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 úLd:\z\workplace\vscode\pyvenv\venv\Lib\site-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 sz | j r<z| jj| }W n tk
|
r, Y dS X t|j|S n:t| jj}z || }W n tk
|
rj Y dS X t||S d 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 tfkrt d¡t| j ts,d | j_| j S )NzcNaming convention including %(constraint_name)s token requires that constraint is explicitly named.)r r r ZInvalidRequestErrorr r r r r r r r Ú_key_constraint_name= s ÿ z#ConventionDict._key_constraint_namec C s | |d¡S )NÚkey©r- ©r r( r r r Ú_key_column_X_keyH s z ConventionDict._key_column_X_keyc C s | |d¡S )Nr r0 r1 r r r Ú_key_column_X_nameN s z!ConventionDict._key_column_X_namec C s | |d¡S )NZ
|
_ddl_labelr0 r1 r r r Ú_key_column_X_labelQ s z"ConventionDict._key_column_X_labelc C s< | j jd }|j d¡}t|dkr0|\}}}n|\}}|S )Nr Ú.é )r r" Ztarget_fullnameÚsplitÚlen)r r* ÚrefsZ refschemaZreftableZrefcolr 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 kr | j | | j| jS t| d| r<t| d| S t d|¡}|r2| d¡}| d¡}|r | jrx| jj }n t
|
| jj }g }t |D ]R\}}d| d| d¡ }z| t| ||¡ W q tk
|
rà t|Y qX q| d¡ròdnd } | |¡S d| |d¡ }t|}t| |r2t| ||S t|d S )
|
Nz_key_%sz.*_?column_(\d+)(_?N)?_.+r r Z_key_Ú0ÚXÚ_r! )r r r Úhasattrr$ ÚreÚmatchÚgroupr r" r& r' Ú enumerateÚreplaceÚappendÚAttributeErrorÚKeyErrorÚ
|
startswithÚjoinÚint)
|
r r/ Z col_templater( Z multiplesÚelemsÚtokensÚelemÚattrÚsepr r r Ú __getitem__d s4
|
|
|
|
|
zConventionDict.__getitem__N) Ú__name__Ú
|
__module__Ú __qualname__r r r- r. r2 r3 r4 r: r< rQ r r r r r ! s r ÚixÚpkZckZuqr* c C sH |j D ]<}|tkr.t| | kr.| t| S || kr| | S qd S r )Ú__mro__Ú _prefix_dict)Zdict_r/ Zsuper_r r r Ú_get_convention s
|
rY c C s~ |j }t|jt| }t| jtr(| jS |d k rnt| jtsn| jd ksXd|ksX| jtkrnt|t| ||j S |tkrzd S d S )NZconstraint_name) ÚmetadatarY Znaming_conventionÚtyper r r r r )r r rZ r r r r Ú_constraint_name_for_table s* ÿ
|
þüûú ÿÿr\ Z'_sa_event_column_added_to_pk_constraintc C s* | j r&| j}d | _t| |}|r&|| _d S r )Z_implicit_generatedr r r\ )Z pk_constraintr, r Únewnamer r r Ú _column_added_to_pk_constraint° s
|
r^ Úafter_parent_attachc s^ t |tr"t |d fdd¡ n8t |trZt jtsB jtkrFd S t |}|rZ| _d S )Nr_ c s
|
t |S r )Ú_constraint_name)r, r ©r r r Ú<lambda>Ì ó z"_constraint_name.<locals>.<lambda>) r r r Úlistenr r r r r\ )r r r] r ra r r` À s
|
|
ý
|
|
r` ) Ú__doc__Ú
|
__future__r rA r! r Úbaser r" r Zschemar r r r
|
r r r r r r r rX rY r\ Z listens_forr^ r` r r r r Ú<module> sJ e û ÿ
|
|
|
|
