o tBh/J@svUdZddlmZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl mZddlmZddlmZmZmZmZmZmZmZd Zd ed <  dTdUddZdVddZdWddZdWdd ZdWd!d"ZdWd#d$Z dXd'd(Z!dWd)d*Z"dYd-d.Z#    dZd[d5d6Z$d\d8d9Z%dWd:d;Z&  dTd]dBdCZ'  dTd^dEdFZ(d_dGdHZ)d`dJdKZ*dadMdNZ+dadOdPZ,dbdcdRdSZ-dS)dzTStub generator for C modules. The public interface is via the mypy.stubgen module. ) annotationsN) ModuleType)AnyMapping)Final) is_c_module)ArgSig FunctionSig!infer_arg_sig_from_anon_docstringinfer_prop_type_from_docstring&infer_ret_type_sig_from_anon_docstring!infer_ret_type_sig_from_docstringinfer_sig_from_docstring) rCallableClassVarDictIterableIteratorListOptionalTupleUnionr_DEFAULT_TYPING_IMPORTS module_namestrtargetsigsdict[str, str] | None class_sigsreturnNonec Cst|}t|sJ|dtj|}|r#tj|s#t|g}g}t}t |j ddd} | D]\} } t | rNt || | |||d|| q7g} | D]#\} } | drb| drbqSt| rvt|| | | |||d|| qSg} | D],\} } | dr| drq{| |vrt| sttt| ||}| | d|q{t t|}| D]}||q| D]}|d r|r|d r|d ||q|r|r|d |D]}||qt|}t|d }|D] }||d qWddS1swYdS)a8Generate stub for C module. This combines simple runtime introspection (looking for docstrings and attributes with simple builtin types) and signatures inferred from .rst documentation (if given). If directory for target doesn't exist it will be created. Existing stub will be overwritten. z is not a C modulecSs|dSNrxr"r"d/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/mypy/stubgenc.pyAsz,generate_stub_for_c_module..key)importsr__r)rr: classw N) importlib import_modulerospathdirnameisdirmakedirssetsorted__dict__items is_c_functiongenerate_c_function_stubadd startswithendswith is_c_typegenerate_c_type_stubinspectismodulestrip_or_importget_type_fullnametypeappendadd_typing_importopenwrite)rrrrmodulesubdirr) functionsdoner<nameobjtypes variablestype_stroutputlinefiler"r"r%generate_stub_for_c_module+sb               $rYrV list[str]csVg}tD]tfdd|Dr|q|r%dd|dg|S|ddS)zJAdd typing imports for collections/types that occur in the generated stub.c3s |] }td|VqdS)z\b%s\bN)research).0rWrQr"r% ksz$add_typing_import..zfrom typing import , r/N)ranyrIjoin)rVnamesr"r^r%rJgs  rJrRobjectboolcCt|p t|ttuSN)rD isbuiltinrHordrRr"r"r%r=sr=cCs.t|pt|ttjttjttjfvSrg)rDismethoddescriptorrHrindex__add____new__rjr"r"r% is_c_methodws rpcCst|p t|jdvS)N) classmethodclassmethod_descriptor)rDrhrH__name__rjr"r"r%is_c_classmethodsrtcCst|p t|dS)Nfget)rDisdatadescriptorhasattrrjr"r"r% is_c_propertyrxproprcCst|do |jduS)Nfset)rwr{)rzr"r"r%is_c_property_readonlyryr|cCrfrg)rDisclassrHintrjr"r"r%rBrkrBdocstrrQcCs||ddS)Nz(*args, **kwargs) zOverloaded function. r@)rrQr"r"r%)is_pybind11_overloaded_function_docstringryrrMrr)self_var str | None class_namec  Cs|duri}|dur i}|dkr|rdnd} |dvr1||vr1|r1||vr1t|t||| dg} nYt|dd} t| |} | rL| dusDJt| |rL| d=| so|r`||vr`t|t||| d g} n*t|t||d | dg} n|r|r| d j} | r| d j|kr| d t |d | rt | d knd} | r| d| r| D]Q}g}|jD]-}|j|kr|}n|j}|dkrd}|j r|dt|j ||7}|jr|d7}| |q| r| d| dj|d|t|j||dqdSdS)aGenerate stub for a single function or method. The result (always a single line) will be appended to 'output'. If necessary, any required names will be added to 'imports'. The 'class_name' is used to find signature of __init__ or __new__ in 'class_sigs'. N__init__r rror)rQargsret_type__doc__r.)rrz(*args, **kwargs)rr^Fzfrom typing import overload_noner,z = ...z @overloadz$def {function}({args}) -> {ret}: ...r`)functionrret)r r getattrrrinfer_method_siggetrrQinsertrlenrIrHrFdefaultformatrbr)rMrQrRrVr)rrrrrinferredrr is_overloaded signaturesigargarg_defr"r"r%r>s              r>typcs}tfdddDr0tdD] }t|||q|r/tdt|jdd}n:|rF|jdrFt |jdd }n$dvrjd d}|d krbt d dd }n| d ||d krpd }|S)atStrips unnecessary module names from typ. If typ represents a type that is inside module or is a type coming from builtins, remove module declaration from it. Return stripped name of the type. Arguments: typ: name of the type module: in which this type is used imports: list of import statements (may be modified during the call) c3s|]}|vVqdSrgr")r]crr"r%r_sz"strip_or_import..z[,z[\[,\]]z (^|[\[, ]+).z\1rNbuiltinszimport NoneTyper ) rar[splitrFstripsubescapersr@rrindexrI)rrMr) stripped_typesubtyp arg_moduler"rr%rFs" rFcCst|jdkS)Npybind11_static_property)rHrsrjr"r"r%is_static_propertysrstatic_properties rw_properties ro_propertiesreadonlyModuleType | Nonelist[str] | Nonec sdfdd }tr dS|t|dd} | s%t|dd} |t| dd} | s)d } |dur7|dur7t| ||} t|rP|r?d nd } |d | d | dS|re|d|dd| ddS|d| dS)zGenerate property stub using introspection of 'obj'. Try to infer type from docstring, append resulting lines to 'output'. rrrcs2|durt|}|st|}|st|}|SdS)z:Infer property type from docstring or docstring signature.N)r r r )rrr^r"r%infer_prop_type*s z1generate_c_property_stub..infer_prop_typeNrrurz # read-onlyr/z : ClassVar[z] = ...z @propertyzdef z (self) -> : ...r,)rrrr)is_skipped_attributerrFrrI) rQrRrrrrrMr)rrrutrailing_commentr"r^r%generate_c_property_stubs$     rrHc st|d}t|ddd}g} g} g} g} g} t}|D]h\}}t|s*t|rZ||t|sY|dkr>d|vr.r'rorz @classmethodclsself)r)rrrr)rMr)r+z{}: ClassVar[{}] = ...r.pybind11_objectrNc3s|]}t|VqdSrg) issubclass)r]b)baser"r%r_sz'generate_c_type_stub..z(%s)r`c3s |] }tt|VqdSrg)rFrG)r]r)r)rMr"r%r_s r/zclass :r-z r)rr:r<r9rprtr?rrIr>rxrr|rBrCrrFrGrHmrordrsrarbr@)rMrrRrVr)rrobj_dictr<methodsrSrrrrPattrvaluer all_basesbases bases_strrWr")rr)rMr%rCOs               rCcCs|jdt|d|jS)Nr __qualname__) __module__rrsrr"r"r%rGrkrGtuple[int, str]cCs4|dvrd|fS|dr|drd|fSd|fS)ziSort methods in classes in a typical order. I.e.: constructor, normal methods, special methods. rrr*r)r@rAr^r"r"r%rs rrcCs |dS)N__pybind11_module_local_rrr"r"r%is_pybind_skipped_attributes rcCs|dvpt|S)N)__getattribute____str____repr__rr;r __weakref__)rrr"r"r%rs r list[ArgSig]cCsd}|dr|dr|dd}|dvrg}n|dkr$tddg}n|dkr3tddtd dg}ny|d vr>td dg}nn|d krMtd dtd dg}n_|dkrTg}nX|dkr_tddg}nM|dvrjtddg}nB|dvrqg}n;|dkrtddtddg}n,|dkrtddtd dg}n|dkrtddg}n|dkrtddtd dtddg}|durtddtddg}t|pddg|S) Nr*r)hashiternextsizeofcopydeepcopyreduce getinitargsr~floattrunccomplexreabsbytesdirrreversedroundrmentergetitemrmr^setitemrd)delattrrrQsetattrrgetstatesetstatestate).eqneltlegtger?raddrrsubmulrmulmodrmodfloordiv rfloordivtruedivrtruedivdivmodrdivmodpowrpowxorrxororrorandrandlshiftrlshiftrshiftrrshiftcontainsdelitemiaddiand ifloordivilshiftimodimulioripowirshiftisubitruedivixorother)negposinvertrinstanceownerr9 reduce_exprotocolexitrH tracebackz*argsz**kwargsr)r@rAr)rQrrr"r"r%rs@ 0r)NN) rrrrrrrrrr )rVrZrrZ)rRrdrre)rzrrre)rrrQrrre)NNNN)rMrrQrrRrdrVrZr)rZrrrrrrrrrr )rrrMrr)rZrr)rQrrRrdrrZrrZrrZrrerMrr)rrr )rMrrrrRrHrVrZr)rZrrrrrr )rrHrr)rQrrr)rrrrerg)rQrrrrr).r __future__rr2rDos.pathr4r[rSrtypingrrtyping_extensionsrmypy.moduleinspectr mypy.stubdocrr r r r r rr__annotations__rYrJr=rprtrxr|rBrr>rFrrrCrGrrrrr"r"r"r%sR    $  <        c   : {