o ki'@sddlZddlZddlZddlmZmZddlmZmZm Z ddl m Z ddl mZddlmZmZmZgdZeeZGd d d eZGd d d eZGd ddeZGdddeZeZGdddeZdS)N)ABCabstractmethod) _getattribute_Pickler whichmodule) ModuleType)Any)demangleget_mangle_prefix is_mangled)ObjNotFoundErrorObjMismatchErrorImporterOrderedImporterc@eZdZdZdS)r zHRaised when an importer cannot find an object by searching for its name.N__name__ __module__ __qualname____doc__rr`/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/package/importer.pyr r c@r)rz]Raised when an importer found a different object with the same name as the user-provided one.Nrrrrrrrrc @sreZdZUdZeeefed<ededefddZ dde d edBde eeffd d Z de d edefd d 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_namereturncCsdS)zvImport `module_name` from this environment. The contract is the same as for importlib.import_module. Nrselfrrrr import_module3szImporter.import_moduleNobjnamec sjdur0|r0tjt|dur0t|dd}|dur0z |}t|tr%|Wn ty/Ynwdur:t|dddurA|j |}t |}z |}t j dkr_t|d}nt|\}} Wntttfy~t|d|ddw||ur|fSfdd} | |\} } } | |\}}}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 csTdurtd|}t|}|rt|nd}|r#dt|nd}|||fS)Nzname must not be Nonezthe current Python environmentzthe importer for z'sys_importer')AssertionErrorrr r )r r is_mangled_location importer_namer!rrr get_obj_infous   z'Importer.get_name..get_obj_infoz 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 rsys version_inforsplit ImportErrorKeyErrorAttributeErrorr r)rr r!reducervorig_module_namermoduleobj2_r+obj_module_name obj_locationobj_importer_nameobj2_module_name obj2_locationobj2_importer_namemsgrr*rget_name:s`        zImporter.get_namec Cst|dd}|dur |S|jD]*\}}|dks#|dks#|dur$qzt||d|ur3|WSWqty=YqwdS)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)r/rcopyitemsrr8)rr r!rr<rrrrs   zImporter.whichmoduleN)rrrrdictr1r__annotations__rrrtuplerFrrrrrrs $Wrc@s4eZdZdZdefddZdededefdd Zd S) _SysImporterz;An importer that implements the default behavior of Python.rcCs t|SrK) importlibrrrrrrs z_SysImporter.import_moduler r!rcCs"t|dd}|dur |St||S)Nr)r/_pickle_whichmodule)rr r!rrrrrs  z_SysImporter.whichmoduleN)rrrrr1rrrrrrrrOsrOc @sleZdZdZddZddZddededBd eeeffd d Z d ed e fd dZ deded efddZ 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". cGst||_dSrK)list _importers)rargsrrr__init__szOrderedImporter.__init__cCs6t|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)r/hasattrrX)rr<rrr_is_torchpackage_dummys    z&OrderedImporter._is_torchpackage_dummyNr r!rc Cs|jD]2}z |||WSttfy5}zd|d|d|d|}t|WYd}~qd}~wwtd|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 )rSrFr rlogwarning)rr r!importerewarning_messagerrrrFs$ zOrderedImporter.get_namerc Csd}|jD]3}t|tst|dz||}||r Wq|WSty8}z|}WYd}~qd}~ww|dur?|t|)NzP is not a Importer. All importers in OrderedImporter must inherit from Importer.)rSr0r TypeErrorrrZModuleNotFoundError)rrlast_errr]r<errrrrrs$     zOrderedImporter.import_modulecCs,|jD]}|||}|dkr|SqdS)NrG)rSr)rr r!r]rrrrrs  zOrderedImporter.whichmodulerK) rrrrrUrZrr1rNrFrrrrrrrrs$r)rPloggingr3abcrrpicklerrrrQtypesrtypingr _manglingr r r __all__ getLoggerrr[r2r rrrO sys_importerrrrrrs"