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

style fix

Browse Source
This commit is contained in:
AF 2023-08-25 09:16:41 +00:00
parent f1692effcf
commit 0794a9392d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
starbot/starbot/instrumentation.py
View File

@ -4,7 +4,7 @@ import functools
from contextlib import ExitStack from contextlib import ExitStack
from typing import Callable, Self from typing import Callable, Self
__all__ = ('Instrumentation',) __all__ = ("Instrumentation",)
class Instrumentation: class Instrumentation:
@ -14,9 +14,9 @@ class Instrumentation:
def __init__(self, target, methodname: str): def __init__(self, target, methodname: str):
if not isinstance(methodname, str): if not isinstance(methodname, str):
raise TypeError('methodname must be str') raise TypeError("methodname must be str")
if not callable(getattr(target, methodname)): if not callable(getattr(target, methodname)):
raise TypeError('target.methodname must be callable') raise TypeError("target.methodname must be callable")
self.target = target self.target = target
self.methodname = methodname self.methodname = methodname
@ -24,7 +24,7 @@ class Instrumentation:
return method(*args, **kwargs) return method(*args, **kwargs)
def __enter__(self) -> Self: def __enter__(self) -> Self:
if hasattr(self, '_method') or hasattr(self, '_wrap'): if hasattr(self, "_method") or hasattr(self, "_wrap"):
raise RuntimeError raise RuntimeError
method = getattr(self.target, self.methodname) method = getattr(self.target, self.methodname)
if not callable(method): if not callable(method):
@ -59,7 +59,7 @@ class Instrumentation:
def enter(self, stack: ExitStack) -> Self: def enter(self, stack: ExitStack) -> Self:
return stack.enter_context(self) return stack.enter_context(self)
def enter_conditional(self, stack: ExitStack) -> Self | None: def enter_conditional(self, stack: ExitStack) -> Self | None:
if hasattr(self.target, self.methodname): if hasattr(self.target, self.methodname):
return self.enter(stack) return self.enter(stack)