a 8Sic"@sddlZddlmZmZddlmZmZddlmZddl m Z ddl m Z m Z mZmZmZddlmZmZmZGd d d eZGd d d eZGd ddeZGdddeZeZGdddeZdS)N)ABCabstractmethod) _getattribute_Pickler) whichmodule) ModuleType)AnyListOptionalTupleDict)demangleget_mangle_prefix is_mangledc@seZdZdZdS)ObjNotFoundErrorzHRaised when an importer cannot find an object by searching for its name.N__name__ __module__ __qualname____doc__rrR/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/package/importer.pyr src@seZdZdZdS)ObjMismatchErrorz]Raised when an importer found a different object with the same name as the user-provided one.Nrrrrrrsrc@sheZdZUdZeeefed<eeedddZ d e e ee eefddd Z e eedd d ZdS) ImporteradRepresents 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_module.szImporter.import_moduleNobjnamerc sVdur`|r`tjt|dur`t|dd}|dur`z|}t|trJ|Wnty^Yn0durtt|dddur|j |}t |}z |}t |\}} Wn2t ttfyt|d|ddYn0||ur|fSfdd} | |\} } } | |\}}}d| d | d |d |d | d |d }t|dS)amGiven an object, return a name that can be used to retrieve the object from this environment. Args: obj: An object to get the 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__rz was not found as .csPdus J|}t|}|r,t|nd}|rBdt|nd}|||fS)Nzthe current Python environmentzthe importer for z'sys_importer')rrr)r#rZ is_mangled_locationZ importer_namer$r rr get_obj_infols   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 Exceptionrrrr!r ImportErrorKeyErrorAttributeErrorrr)r r#r$reducervZorig_module_namermoduleobj2_r)Zobj_module_nameZ obj_locationZobj_importer_nameZobj2_module_nameZ obj2_locationZobj2_importer_namemsgrr(rget_name6sT       zImporter.get_namec Cst|dd}|dur|S|jD]V\}}|dks&|dks&|durHq&z t||d|urf|WSWq&tyzYq&0q&dS)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-rcopyitemsrr3)r r#r$rr6rrrrs"  zImporter.whichmodule)N)rrrrr r/r__annotations__rr!rr r r:rrrrrrs  Qrc@s0eZdZdZedddZeeedddZdS) _SysImporterz;An importer that implements the default behavior of Python.)rcCs t|SN) importlibr!rrrrr!sz_SysImporter.import_moduler"cCs t||SrA)_pickle_whichmodule)r r#r$rrrrsz_SysImporter.whichmoduleN)rrrrr/r!rrrrrrr@sr@c@sBeZdZdZddZddZeedddZe eed d d Z d S) OrderedImporterzA 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||_dSrA)list _importers)r argsrrr__init__szOrderedImporter.__init__cCs6t|ddsdSt|dsdSt|ds,dS|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 Z__torch_package__F__path____file__TN)r-hasattrrJ)r r6rrr_is_torchpackage_dummys   z&OrderedImporter._is_torchpackage_dummyrc Csd}|jD]h}t|ts&t|dz"||}||r@Wq |WStyp}z|}WYd}~q d}~00q |dur|nt|dS)NzP is not a Importer. All importers in OrderedImporter must inherit from Importer.)rFr.r TypeErrorr!rLModuleNotFoundError)r rZlast_errimporterr6errrrrr!s      zOrderedImporter.import_moduler"cCs,|jD] }|||}|dkr|SqdS)Nr;)rFr)r r#r$rOrrrrrs    zOrderedImporter.whichmoduleN) rrrrrHrLr/rr!rrrrrrrDs rD)rBabcrrpicklerrrrCtypesrtypingrr r r r _manglingrrrr0rrrr@ sys_importerrDrrrrs