Spaces:
Sleeping
Sleeping
| """Import mangling. | |
| See mangling.md for details. | |
| """ | |
| import re | |
| _mangle_index = 0 | |
| class PackageMangler: | |
| """ | |
| Used on import, to ensure that all modules imported have a shared mangle parent. | |
| """ | |
| def __init__(self): | |
| global _mangle_index | |
| self._mangle_index = _mangle_index | |
| # Increment the global index | |
| _mangle_index += 1 | |
| # Angle brackets are used so that there is almost no chance of | |
| # confusing this module for a real module. Plus, it is Python's | |
| # preferred way of denoting special modules. | |
| self._mangle_parent = f"<torch_package_{self._mangle_index}>" | |
| def mangle(self, name) -> str: | |
| assert len(name) != 0 | |
| return self._mangle_parent + "." + name | |
| def demangle(self, mangled: str) -> str: | |
| """ | |
| Note: This only demangles names that were mangled by this specific | |
| PackageMangler. It will pass through names created by a different | |
| PackageMangler instance. | |
| """ | |
| if mangled.startswith(self._mangle_parent + "."): | |
| return mangled.partition(".")[2] | |
| # wasn't a mangled name | |
| return mangled | |
| def parent_name(self): | |
| return self._mangle_parent | |
| def is_mangled(name: str) -> bool: | |
| return bool(re.match(r"<torch_package_\d+>", name)) | |
| def demangle(name: str) -> str: | |
| """ | |
| Note: Unlike PackageMangler.demangle, this version works on any | |
| mangled name, irrespective of which PackageMangler created it. | |
| """ | |
| if is_mangled(name): | |
| first, sep, last = name.partition(".") | |
| # If there is only a base mangle prefix, e.g. '<torch_package_0>', | |
| # then return an empty string. | |
| return last if len(sep) != 0 else "" | |
| return name | |
| def get_mangle_prefix(name: str) -> str: | |
| return name.partition(".")[0] if is_mangled(name) else name | |