1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
|
from operator import setitem
from typing import Any, Callable, Optional, Tuple
from testfixtures import not_there
# Should be Literal[setattr, getattr] but Python 3.8 only.
Setter = Callable[[Any, str, Any], None]
class Resolved:
def __init__(self, container: Any, setter: Setter, name: str, found: Any):
self.container: Any = container
self.setter: Setter = setter
self.name: str = name
self.found: Any = found
def key(self) -> Tuple[int, Setter, str]:
return id(self.container), self.setter, self.name
def __repr__(self):
return f'<Resolved: {self.found}>'
def resolve(dotted_name: str, container: Optional[Any] = None, sep: str = '.') -> Resolved:
names = dotted_name.split(sep)
used = names.pop(0)
if container is None:
found = __import__(used)
container = found
else:
assert not used, 'Absolute traversal not allowed when container supplied'
used = ''
found = container
setter = None
name = None
for name in names:
container = found
used += '.' + name
try:
found = getattr(found, name)
setter = setattr
except AttributeError:
try:
if sep != '.':
raise ImportError
__import__(used)
except ImportError:
setter = setitem
try:
found = found[name] # pragma: no branch
except KeyError:
found = not_there # pragma: no branch
except TypeError:
try:
name = int(name)
except ValueError:
setter = setattr
found = not_there
else:
found = found[name] # pragma: no branch
else:
found = getattr(found, name)
setter = getattr
if found is not_there:
break
return Resolved(container, setter, name, found)
class _Reference:
@classmethod
def classmethod(cls): # pragma: no cover
pass
@staticmethod
def staticmethod(cls): # pragma: no cover
pass
class_type = type(_Reference)
classmethod_type = type(_Reference.classmethod)
|