o ۰i @sddlZddlmZmZmZmZmZmZmZm Z ddl m Z  dde dee de fddZ d e dee fd d Zd d ZddZdde fddZdS)N)ListTupleUnionOptionalCallableAnyDictSet)CodeTypecode_objoverwite_fn_namereturncCs|j|j}|jd|}dd|D}ddl}|j|j@r,|d|j||d7}|j|j@r@|d|j||d7}d|}|durK|n|j }d|d |d }|S) NcSs&g|]}|ds |n|ddqS). comp_arg_) startswithreplace).0xr?/home/ubuntu/.local/lib/python3.10/site-packages/depyf/utils.py s z*get_function_signature..r*z**z, zdef (z): ) co_argcountco_kwonlyargcount co_varnamesinspectco_flags CO_VARARGSappendCO_VARKEYWORDSjoinco_name)r r normal_arg_count arg_namesrargs_strfn_nameheaderrrrget_function_signatures    r)codecCs2|g}|jD]}t|t|r|t|q|SN) co_consts isinstancetypeextendcollect_all_code_objects)r* code_objectsconstrrrr0s  r0c CsRddl}z |j|ddWdSty(}z|j|sWYd}~dSd}~ww)NrT)exist_ok)osmakedirsOSErrorpathisdir)r7r4errrsafe_create_directory&s r:cCsjddl} t|dr|j}n&t|dr|j}nt||jr!|j}nt|dr1t|jdr1|jj}n |Sq)aA callable object `fn` might have a __code__ attribute, which is a code object. However, `fn` might not be the owner of the code object. Only the code owner can change the code object. This function returns the owner of the code object. An example: class A: def func(self): return 1 a = A() `a.func.__code__` is read-only. `A.func.__code__` is writable. We can change the code object via `a.func.__func__.__code__`. rNT__func__ __wrapped____call__) functoolshasattrr;r<r-partialfuncr=)fnr>rrrget_code_owner1s     rCrBc Cshddl}ddlm}z ||j|d}W|S|y3}zt||d}|d}WYd}~|Sd}~ww)Nr)DecompilationError)r z 'Failed to decompile.' )depyfdepyf.decompilerrD Decompiler decompiler))rBr rErDdecompiled_source_coder9r(rrrdecompile_ensureQs  rJr+)distypingrrrrrrrr typesr strr)r0r:rCrJrrrrs(