Add from_script classmethod to get ScriptExpression by name

selfcustody · Mar 5, 2022 · ac53811 · ac53811
1 parent 17f4b45
commit ac53811
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/urtypes/crypto/output.py b/src/urtypes/crypto/output.py
@@ -36,6 +36,13 @@ def __init__(self, tag, expression):
     def __eq__(self, o):
         return self.tag == o.tag and self.expression == o.expression
 
+    @classmethod
+    def from_script(cls, script):
+        for script_expression in SCRIPT_EXPRESSION_TAG_MAP.values():
+            if script == script_expression.expression:
+                return script_expression
+        raise ValueError("unknown script")
+
 
 SCRIPT_EXPRESSION_TAG_MAP = {
     307: ScriptExpression(307, "addr"),