PTV/rainbowadn
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove asserts from instrumentation

Browse Source
This commit is contained in:
AF 2022-12-03 21:32:06 +00:00
parent e9fba7b064
commit 844dacf989
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
rainbowadn/instrument/instrumentation.py
View File

@ -13,8 +13,10 @@ class Instrumentation(Generic[IType]):
_wrap: Callable _wrap: Callable
def __init__(self, target, methodname: str): def __init__(self, target, methodname: str):
assert isinstance(methodname, str) if not isinstance(methodname, str):
assert callable(getattr(target, methodname)) raise TypeError('methodname must be str')
if not callable(getattr(target, methodname)):
raise TypeError('target.methodname must be callable')
self.target = target self.target = target
self.methodname = methodname self.methodname = methodname
@ -22,11 +24,13 @@ class Instrumentation(Generic[IType]):
raise NotImplementedError raise NotImplementedError
def __enter__(self: IType) -> IType: def __enter__(self: IType) -> IType:
assert isinstance(self, Instrumentation) if not isinstance(self, Instrumentation):
assert not hasattr(self, '_method') raise TypeError
assert not hasattr(self, '_wrap') if hasattr(self, '_method') or hasattr(self, '_wrap'):
raise RuntimeError
method = getattr(self.target, self.methodname) method = getattr(self.target, self.methodname)
assert callable(method) if not callable(method):
raise TypeError
self._method = method self._method = method
@functools.wraps(method) @functools.wraps(method)
@ -56,11 +60,13 @@ class Instrumentation(Generic[IType]):
self.deinstrument() self.deinstrument()
def enter(self: IType, stack: ExitStack) -> IType: def enter(self: IType, stack: ExitStack) -> IType:
assert isinstance(self, Instrumentation) if not isinstance(self, Instrumentation):
raise TypeError
return stack.enter_context(self) return stack.enter_context(self)
def enter_conditional(self: IType, stack: ExitStack) -> IType | None: def enter_conditional(self: IType, stack: ExitStack) -> IType | None:
assert isinstance(self, Instrumentation) if not isinstance(self, Instrumentation):
raise TypeError
if hasattr(self.target, self.methodname): if hasattr(self.target, self.methodname):
return self.enter(stack) return self.enter(stack)
else: else: