o ni(@sUdZddlZddlZddlmZddlmZmZmZm Z m Z iZ ee d<da ee d<eZded efd d Zd e eefd eded dfddZded e ed dfddZdddZded dfddZGdddZGdddZdS)aHImplements a post-import hook mechanism. Styled as per PEP-369. Note that it doesn't cope with modules being reloaded. Note: This file is based on https://github.com/GrahamDumpleton/wrapt/blob/1.12.1/src/wrapt/importer.py and manual backports of later patches up to 1.15.0 in the wrapt repository (with slight modifications). N) find_spec)AnyCallableDictOptionalUnion_post_import_hooksF_post_import_hooks_initnamereturncsdtdtffdd }|S)Nmoduler csFd\}}|d}t|tj|}|D]}t||}q||S)N:.)split __import__sysmodulesgetattr)r module_namefunctionattrscallbackattrr X/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/lib/import_hooks.py import_hook#s   z4_create_import_hook_from_string..import_hook)rr)r rrrr_create_import_hook_from_string"s rhookhook_idcCst|tfr t|}t*tsdatjdttj |d}|dur/t |i ||iWdn1s9wY|durH||dSdS)NTr) isinstancestrr_post_import_hooks_lockr r meta_pathinsertImportHookFinderrgetr setdefaultupdate)rrr r rrrregister_post_import_hook/s  r)cCst5t|}|dur'|dur||d|st|=nt|=WddSWddSWddS1s:wYdSN)r"rr&pop)r rhooksrrrunregister_post_import_hookSs  "r-cCs2t tWddS1swYdSr*)r"rclearrrrr unregister_all_post_import_hooksbs "r/r cCsZt|dd}tt|i}Wdn1swY|D]}|r*||q"dS)N__name__)rr"rr+values)r r r,rrrrnotify_module_loadedms  r2c@sfeZdZdeddfddZdeddfddZd edefd d Zd edefd dZdeddfddZ dS)_ImportHookChainedLoaderloaderr NcCsD||_t|dr |j|_t|dr|j|_t|dr |j|_dSdS)N load_module create_module exec_module)r4hasattr _load_moduler5_create_moduler6 _exec_moduler7)selfr4rrr__init__s    z!_ImportHookChainedLoader.__init__r cCszGddd}t|d|d|fvr!z|j|_Wn ty Ynwt|dddur9t|jdd|ur;|j|j_dSdSdS)Nc@s eZdZdS)z7_ImportHookChainedLoader._set_loader..UNDEFINEDN)r0 __module__ __qualname__rrrr UNDEFINEDsr@ __loader____spec__r4)rr4rAAttributeErrorrB)r<r r@rrr _set_loaders   z$_ImportHookChainedLoader._set_loaderfullnamecCs"|j|}||t||Sr*)r4r5rDr2)r<rEr rrrr9s  z%_ImportHookChainedLoader._load_modulespeccCs |j|Sr*)r4r6)r<rFrrrr:s z'_ImportHookChainedLoader._create_modulecCs"|||j|t|dSr*)rDr4r7r2)r<r rrrr;s   z%_ImportHookChainedLoader._exec_module) r0r>r?rr=rDr!r9r:r;rrrrr3s   r3c @sZeZdZd ddZ ddedeededfdd Z ddedeed edefd d ZdS)r%r NcCs i|_dSr*) in_progress)r<rrrr=s zImportHookFinder.__init__rEpathr3cCst|tvr WddSWdn1swY||jvr&dSd|j|<z&tt|dd}|rDt|tsKt|W|j|=SW|j|=dSW|j|=dS|j|=wNTr4)r"rrGrrr r3)r<rErHr4rrr find_modules$     zImportHookFinder.find_moduletargetcCst|tvr WddSWdn1swY||jvr&dSd|j|<z#t|}t|dd}|rHt|tsH|dusCJt||_|W|j|=S|j|=wrI)r"rrGrrr r3r4)r<rErHrKrFr4rrrrs"      zImportHookFinder.find_specr Nr*)NN) r0r>r?r=r!rrJrrrrrrr%s(  ,r%rL)__doc__r threadingimportlib.utilrtypingrrrrrr__annotations__r boolRLockr"r!rr)r-r/r2r3r%rrrrs.       $  9