optimizing mnemonic_to_bytes

selfcustody · Aug 12, 2024 · ef58d03 · ef58d03
1 parent cbb69fd
commit ef58d03
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/krux/bip39.py b/src/krux/bip39.py
@@ -14,14 +14,15 @@ def mnemonic_to_bytes(
         raise ValueError("Invalid recovery phrase")
 
     accumulator = 0
-    for word in words:
-        try:
-            if wordlist is WORDLIST:
+    try:
+        if wordlist is WORDLIST:
+            for word in words:
                 accumulator = (accumulator << 11) + WORDINDEX[word]
-            else:
+        else:
+            for word in words:
                 accumulator = (accumulator << 11) + wordlist.index(word)
-        except Exception:
-            raise ValueError("Word '%s' is not in the dictionary" % word)
+    except Exception:
+        raise ValueError("Word '%s' is not in the dictionary" % word)
 
     entropy_length_bits = len(words) * 11 // 33 * 32
     checksum_length_bits = len(words) * 11 // 33