X-Ph<dZddlmZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZmZmZmZmZmZmZddlmZddZddZddZddZdS)zGSimple type inference for decorated functions during semantic analysis.) annotations)ARG_POSCallExpr Decorator ExpressionFuncDefRefExprVar)SemanticAnalyzerInterface) function_type)AnyType CallableType ProperTypeType TypeOfAny TypeVarTypeget_proper_type)has_no_typevarsdecranalyzerr returnNonec V|jjr|jjst t t jgtgdgt t j| d|jj |j_n5t|jjtr|jj|j_dSd}|j D]`}d}t|trAt|jtr'|jjrt!|jjrd}|sd}na|r2t#|j| d|j_|j rt%|j d}|r:t|t r%t t j||j_t)|j d}|rOt#|j| d}|jdj |_ ||j_dSdSdS)aaTry to infer the type of the decorated function. This lets us resolve additional references to decorated functions during type checking. Otherwise the type might not be available when we need it, since module top levels can't be deferred. This basically uses a simple special-purpose type inference engine just for decorators. Nzbuiltins.function)nameTFr) source_any)var is_propertyfunctyperr r special_formr named_typer isinstance decoratorsr noderis_identity_signaturer calculate_return_typefrom_another_anyfind_fixed_callable_returnitems)rrdecorator_preserves_typeexpr preserve_type return_typesigorig_sigs R/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/mypy/semanal_infer.py#infer_decorator_signature_if_simpler1s w  8= '/001  .//##$788W\ CGLL | 4 4 )8=CGL# dG $ $ %DIw)G)G %y~ %"7 "G"G % $  ', $ E  Y%SXx/B/BCV/W/WXX  ~ +CN1,=>>  W:k7;; W#9#=+VVVCGL():;;  %SXx/B/BCV/W/WXXH~a(-CHCGLLL    r.rboolc$t|}t|trk|jtgkrZt|jdt r:t|jt r |jdj|jjkSdS)z?Is type a callable of form T -> T (where T is a type variable)?rF) rr"r arg_kindsr arg_typesrret_typeid)r.s r0r%r%Jsz #  C#|$$:7))C)C cmA& 4 4 :CLR]9^9^ :=#&#,/9 9 5r2r+rProperType | Nonect|trt|jtrR|jj}|t t jSt|trt|j SdSt|jtrt|jjSn)t|trt|jSdS)zReturn the return type if we can calculate it. This only uses information available during semantic analysis so this will sometimes return None because of insufficient information (as type inference hasn't run yet). N)r"r r$rrr r unannotatedrrr7r rr&callee)r+typs r0r&r&Ss$  2 di ) ) 3).C{y4555#|,, 5&s|4444  3 ' ' 3"49>22 2 3 D( # #2$T[111 4r2CallableType | Nonect|tr}t|jtrb|jj}|rTt|t r?t |jr+t|j}t|t r|SnVt|trAt|j }|r+t|j}t|t r|SdS)a=Return the return type, if expression refers to a callable that returns a callable. But only do this if the return type has no type variables. Return None otherwise. This approximates things a lot as this is supposed to be called before type checking when full type information is not available yet. N) r"r r$rrrrr7rrr(r<)r+r=r7ts r0r(r(ks$   di ) ) ().C (c<00(_S\5R5R(.s|<)__doc__ __future__r mypy.nodesrrrrrr r mypy.semanal_sharedr mypy.typeopsr mypy.typesr rrrrrr mypy.typevarsrr1r%r&r(r2r0rIs@MM""""""VVVVVVVVVVVVVVVVVV999999&&&&&&*)))))3333l0r2