U
|
¸ý°dG' ã @ sR d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm
|
Z
|
dd l m Z dd
|
l mZ dd l mZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddl m Z ddl m!Z! ddl mZ dd l"m#Z# dd l"m$Z$ dd l"m%Z% z e&d W n e'k
|
r Y n
|
X e'd G d!d" d"e Z(d#d$d%d&d'Z)d(d)d*d+d,Z*d-d)d*d.d/Z+d-d)d*d0d1Z,d-d)d*d2d3Z-d-d)d*d4d5Z.d-d)d*d6d7Z/d-d)d*d8d9Z0d:d;d*d<d=Z1d>d)d*d?d@Z2dAdBd)dCdDdEZ3dFS )Gz"
|
Mypy plugin for SQLAlchemy ORM.
|
|
é )Ú annotations)ÚCallable)ÚList)ÚOptional)ÚTuple)ÚType)ÚUnion)Únodes)Ú calculate_mro)ÚMroError)ÚBlock)ÚClassDef)ÚGDEF)ÚMypyFile)ÚNameExpr)Ú SymbolTable)ÚSymbolTableNode)ÚTypeInfo)ÚAttributeContext)ÚClassDefContext)ÚDynamicClassDefContext)ÚPlugin)Ú SemanticAnalyzerPluginInterface)Úget_proper_type)ÚInstanceé )Ú
|
decl_class)Únames)Úutilzsqlalchemy-stubszÙThe SQLAlchemy mypy plugin in SQLAlchemy 2.0 does not work with sqlalchemy-stubs or sqlalchemy2-stubs installed, as well as with any other third party SQLAlchemy stubs. Please uninstall all SQLAlchemy stubs packages.c @ s| e Zd ZdddddZdddddZdddd d
|
Zdddd d Zdddd dZdddddZdddddZ dS )ÚSQLAlchemyPluginÚstrz2Optional[Callable[[DynamicClassDefContext], None]])ÚfullnameÚreturnc C s t |¡t jkrtS d S ©N)r Útype_id_for_fullnameZDECLARATIVE_BASEÚ_dynamic_class_hook©Úselfr! © r( úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/ext/mypy/plugin.pyÚget_dynamic_class_hook< s z'SQLAlchemyPlugin.get_dynamic_class_hookz+Optional[Callable[[ClassDefContext], None]]c C s t S r# )Ú_fill_in_decoratorsr& r( r( r) Ú get_customize_class_mro_hookC s z-SQLAlchemyPlugin.get_customize_class_mro_hookc C s\ | |¡}|d k rX|jd k rXt |j¡}|tjkr6tS |tjtjfkrJtS |tj krXt
|
S d S r# ) Úlookup_fully_qualifiedÚnoder Útype_id_for_named_nodeZMAPPED_DECORATORÚ_cls_decorator_hookZAS_DECLARATIVEZAS_DECLARATIVE_BASEÚ_base_cls_decorator_hookZDECLARATIVE_MIXINÚ_declarative_mixin_hook)r' r! ÚsymZtype_idr( r( r) Úget_class_decorator_hookH s
|
|
þ
|
z)SQLAlchemyPlugin.get_class_decorator_hookc C s t |¡t jkrtS d S r# )r r$ ZDECLARATIVE_METAÚ_metaclass_cls_hookr& r( r( r) Úget_metaclass_hook\ s z#SQLAlchemyPlugin.get_metaclass_hookc C s. | |¡}|r*t|jtr*t |j¡r*tS d S r# )r- Ú
|
isinstancer. r r Zhas_declarative_baseÚ_base_cls_hook)r' r! r3 r( r( r) Úget_base_class_hookf s
|
ÿ
|
þ
|
ýz$SQLAlchemyPlugin.get_base_class_hookz,Optional[Callable[[AttributeContext], Type]]c C s | d¡rtS d S )Nz-sqlalchemy.orm.attributes.QueryableAttribute.)Ú
|
startswithÚ_queryable_getattr_hookr& r( r( r) Úget_attribute_hookt s
|
ÿz#SQLAlchemyPlugin.get_attribute_hookr zList[Tuple[int, str, int]])Úfiler" c C s
|
dddgS )N)é
|
úsqlalchemy.orméÿÿÿÿ)r> zsqlalchemy.orm.attributesr@ )r> zsqlalchemy.orm.decl_apir@ r( )r' r= r( r( r) Úget_additional_deps~ s üz$SQLAlchemyPlugin.get_additional_depsN)
|
Ú__name__Ú
|
__module__Ú __qualname__r* r, r4 r6 r9 r< rA r( r( r( r) r ; s
|
|
r r z TypingType[SQLAlchemyPlugin])Úversionr" c C s t S r# )r )rE r( r( r) Úplugin s rF r ÚNone)Úctxr" c C s4 t | t| jtg }| j | j¡|_tt || jj }||_
|
t | j| t j | jdtfd}|dk r¨t|jtr¨t |j¡ tj|jj| jdd t|jg g|_n| j tj¡}|g|_z t| W nD t k
|
r t | jd| j¡ | j tj¡}|g|_d|_ Y nX | j | jt t!|¡ t |¡ dS )zZGenerate a declarative Base class when the declarative_base() function
|
is encountered.Úcls)Z
|
expr_typesNT©Z is_mixin_scanz.Not able to calculate MRO for declarative base)"Ú _add_globalsr Únamer ÚapiZqualified_namer! r r Z
|
cur_mod_idÚinfoÚ_set_declarative_metaclassr Zget_callexpr_kwargÚcallr r7 r. Ú set_is_baser Ú,scan_declarative_assignments_and_apply_typesZdefnr ÚbasesZ
|
named_typer ZNAMED_TYPE_BUILTINS_OBJECTr
|
r ÚfailZfallback_to_anyZadd_symbol_table_noder r )rH rI rN Zcls_argÚobjr( r( r) r% s< ÿ ÿ r% r c C sÒ | j jD ]Ä}t|tjr:t|jtjr:|jjdkr:|j}n t|tjr|jdkr|}nqt|jt r| j
|
j |jj|dd}nq|r|j rt |j}t|tr´|jj d|j |_qt | j
|
d|j |j ¡ qd S )NZas_declarative_baseZmappedT)Zsuppress_errorsÚ.zClass decorator called %s(), but we can't tell if it's from an ORM registry. Please annotate the registry assignment, e.g. my_registry: registry = registry())rI Z
|
decoratorsr7 r ZCallExprZcalleeÚ
|
MemberExprrL Úexprr rM Zlookup_qualifiedr. r Útyper r! r rT )rH Ú decoratorÚtargetr3 Zsym_typer( r( r) r+ ´ s@
|
|
ÿ þ
|
ý
|
ÿþ ÿ
|
|
|
|
ýúr+ c C s t | t| jtjst| jj}t|tjr<t|jtj s@tt
|
|jj }t|t rht |j ¡t jksltt | j| j¡ d S r# )rK r7 Úreasonr rW ÚAssertionErrorrX ZRefExprr. ZVarr rY r r r/ ZREGISTRYr rR rI rM )rH rX Z node_typer( r( r) r0 ê s ÿþr0 c C s> t | | j}t| j| t | jj¡ tj|| jdd d S ©NTrJ ) rK rI rO rM r rQ rN r rR )rH rI r( r( r) r1 û s ÿr1 c C s. t | t | jj¡ tj| j| jdd d S r^ )rK r rQ rI rN r rR rM ©rH r( r( r) r2 s ÿr2 c C s t | jj¡ d S r# )r rQ rI rN r_ r( r( r) r5 s r5 c C s t | t | j| j¡ d S r# )rK r rR rI rM r_ r( r( r) r8 s r8 r r c C s | j S r# )Zdefault_attr_typer_ r( r( r) r; s r; z.Union[ClassDefContext, DynamicClassDefContext]c C s t | ddd¡ dS )z`Add __sa_DeclarativeMeta and __sa_Mapped symbol to the global space
|
for all class defs
|
|
r? ZMappedZ __sa_MappedN)r Z
|
add_globalr_ r( r( r) rK s rK r r )rM Ú
|
target_clsr" c C s@ |j }| d¡}|d k r$t|jts(tt|jg |_|_d S )Nz'sqlalchemy.orm.decl_api.DeclarativeMeta) rN Z lookup_fully_qualified_or_noner7 r. r r] r Zdeclared_metaclassZmetaclass_type)rM r` rN r3 r( r( r) rO ( s ÿrO N)4Ú__doc__Ú
|
__future__r Útypingr r r r r Z
|
TypingTyper Zmypyr Zmypy.mror
|
r Z
|
mypy.nodesr r r r r r r r Z mypy.pluginr r r r r Z
|
mypy.typesr r Ú r r r Ú
|
__import__Ú ImportErrorr rF r% r+ r0 r1 r2 r5 r8 r; rK rO r( r( r( r) Ú<module> s` ÿ N'6
|
