[i(6ddlZddlZddlZddlmZmZddlmZmZm Z ddl m Z ddl mZddlmZmZmZgdZejeZGd d eZGd d eZGd deZGddeZeZGddeZdS)N)ABCabstractmethod) _getattribute_Pickler whichmodule) ModuleType)Any)demangleget_mangle_prefix is_mangled)ObjNotFoundErrorObjMismatchErrorImporterOrderedImporterceZdZdZdS)rzHRaised when an importer cannot find an object by searching for its name.N__name__ __module__ __qualname____doc__I/var/www/icac/venv/lib/python3.11/site-packages/torch/package/importer.pyrrsRRRRrrceZdZdZdS)rz]Raised when an importer found a different object with the same name as the user-provided one.Nrrrrrrsggggrrc eZdZUdZeeefed<ededefdZ d de dedzde eeffd Z de dedefd Z dS) radRepresents an environment to import modules from. By default, you can figure out what module an object belongs by checking __module__ and importing the result using __import__ or importlib.import_module. torch.package introduces module importers other than the default one. Each PackageImporter introduces a new namespace. Potentially a single name (e.g. 'foo.bar') is present in multiple namespaces. It supports two main operations: import_module: module_name -> module object get_name: object -> (parent module name, name of obj within module) The guarantee is that following round-trip will succeed or throw an ObjNotFoundError/ObjMisMatchError. module_name, obj_name = env.get_name(obj) module = env.import_module(module_name) obj2 = getattr(module, obj_name) assert obj1 is obj2 modules module_namereturncdS)zvImport `module_name` from this environment. The contract is the same as for importlib.import_module. Nrselfrs r import_modulezImporter.import_module6srNobjnamec 6t|rrtjt|Ft |dd}|3 |}t |t r|n#t$rYnwxYwt |dd|j |}t|} |}tj dkr$t|d}nt|\}} n3#t t"t$f$rt'|d|ddwxYw||ur|fSfd} | |\} } } | |\}}}d| d | d |d |d | d |d }t)|)aiGiven an object, return a name that can be used to retrieve the object from this environment. Args: obj: An object to get the module-environment-relative name for. name: If set, use this name instead of looking up __name__ or __qualname__ on `obj`. This is only here to match how Pickler handles __reduce__ functions that return a string, don't use otherwise. Returns: A tuple (parent_module_name, attr_name) that can be used to retrieve `obj` from this environment. Use it like: mod = importer.import_module(parent_module_name) obj = getattr(mod, attr_name) Raises: ObjNotFoundError: we couldn't retrieve `obj by name. ObjMisMatchError: we found a different object with the same name as `obj`. N __reduce__r).z was not found as cJ|}t|}|rt|nd}|rdt|nd}|||fS)Nzthe current Python environmentzthe importer for z'sys_importer')rr r )r$r is_mangled_location importer_namer%r"s r get_obj_infoz'Importer.get_name..get_obj_infoxs###**355K$[11K6!+...5 &D$5k$B$BDDD%  -7 7rz The object provided is from 'z', which is coming from z. However, when we import 'z', it's coming from z@. To fix this, make sure this 'PackageExporter's importer lists z before )rdispatchgettypegetattr isinstancestr Exceptionrrr r#sys version_inforsplit ImportErrorKeyErrorAttributeErrorrr)r"r$r%reducervorig_module_namermoduleobj2_r/obj_module_name obj_locationobj_importer_nameobj2_module_name obj2_locationobj2_importer_namemsgs` ` rget_namezImporter.get_name=s_& >+>>>>   $;;$ $ 8 8 8 8 8 8 Jl4>P>P;-); @ @ @$0 @ @*: @ @P] @ @!  @ @+=  @ @ @ s###s!A)) A65A6;AD0Ec t|dd}||S|jD]A\}}|dks|dks| t ||d|ur|cS2#t $rY>wxYwdS)aFind the module name an object belongs to. This should be considered internal for end-users, but developers of an importer can override it to customize the behavior. Taken from pickle.py, but modified to exclude the search into sys.modules rN__main__ __mp_main__r)r3rcopyitemsrr<)r"r$r%rr@s rrzImporter.whichmodulesc<66  " $(<#4#4#6#6#<#<#>#>   Kz))-//>  ..q1S88&&&&9!    zsA33 B?BN)rrrrdictr5r__annotations__rr#r tuplerJrrrrrrs(#z/ """"     ^ T$T$CT$sTzT$U38_T$T$T$T$ls##rrc2eZdZdZdefdZdededefdZdS) _SysImporterz;An importer that implements the default behavior of Python.rc*tj|SrP) importlibr#r!s rr#z_SysImporter.import_modules&{333rr$r%rcLt|dd}||St||S)Nr)r3_pickle_whichmodule)r"r$r%rs rrz_SysImporter.whichmodules1c<66  " "3---rN)rrrrr5r#r rrrrrUrUs^EE44444.s.#.#......rrUc neZdZdZdZdZd dededzdeeeffdZ d ede fd Z dededefd Z dS) rzA compound importer that takes a list of importers and tries them one at a time. The first importer in the list that returns a result "wins". c.t||_dSrP)list _importers)r"argss r__init__zOrderedImporter.__init__s*.t**rct|ddsdSt|dsdSt|dsdS|jduS)aReturns true iff this module is an empty PackageNode in a torch.package. If you intern `a.b` but never use `a` in your code, then `a` will be an empty module with no source. This can break cases where we are trying to re-package an object after adding a real dependency on `a`, since OrderedImportere will resolve `a` to the dummy package and stop there. See: https://github.com/pytorch/pytorch/pull/71520#issuecomment-1029603769 __torch_package__F__path____file__TN)r3hasattrrc)r"r@s r_is_torchpackage_dummyz&OrderedImporter._is_torchpackage_dummysYv2E:: 5vz** 5vz** 4$&&rNr$r%rc  |jD]`} |||cS#ttf$r2}d|d|d|d|}t|Yd}~Yd}~wwxYwtd|d|d|j)Nz Tried to call get_name with obj z , and name z on z and got zCould not find obj z and name z in any of the importers )r]rJrrlogwarning)r"r$r%importerewarning_messages rrJzOrderedImporter.get_names - -H -((d33333$&67 - - -AsAA $AA*2AA=>AA  O,,,,,,,,  -  a# a a a aPTP_ a a   s#A&(A!!A&rcd}|jD]q}t|tst|d ||}||rU|cS#t $r }|}Yd}~jd}~wwxYw||t |)NzP is not a Importer. All importers in OrderedImporter must inherit from Importer.)r]r4r TypeErrorr#reModuleNotFoundError)r"rlast_errrir@errs rr#zOrderedImporter.import_modules  Hh11 SSS !// <<..v66 &      N%k22 2s*A#A## A9-A44A9cZ|jD]"}|||}|dkr|cS#dS)NrL)r]r)r"r$r%rirs rrzOrderedImporter.whichmodulesK # #H"..sD99Kj(("""")zrrP) rrrrr_rer r5rSrJrr#rrrrrrs 555'''$   C  sTz  U38_     333333*s##rr)rWloggingr7abcrrpicklerrrrYtypesrtypingr _manglingr r r __all__ getLoggerrrgr6rrrrU sys_importerrrrrr{s ######## >>>>>>>>>> R Q Qg!!SSSSSySSShhhhhyhhhOOOOOsOOOd . . . . .8 . . . |~~ DDDDDhDDDDDr