Get annotations from Expr (#1332)

5 years ago · 83ef2c6024
parent 4d2b331ac0
commit 83ef2c6024
1 changed files with 3 additions and 0 deletions
--- a/mythril/laser/smt/expression.py
+++ b/mythril/laser/smt/expression.py
@ -54,6 +54,9 @@ class Expression(Generic[T]):
    def __hash__(self) -> int:
        return self.raw.__hash__()

+    def get_annotations(self, annotation: Any):
+        return list(filter(lambda x: isinstance(x, annotation), self.annotations))
+

 G = TypeVar("G", bound=Expression)