fix: previous commit also applies for Python 2

knipknap · Nov 28, 2017 · 04d9571 · 04d9571
1 parent bac2765
commit 04d9571
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Exscript/util/crypt.py b/Exscript/util/crypt.py
@@ -369,9 +369,10 @@ def otp(password, seed, sequence):
             raise ValueError('seed composition')
     if sequence < 0:
         raise ValueError('sequence')
-    if sys.version_info[0] >= 3:
-        seed = seed.encode('utf-8')
-        password = password.encode('utf-8')
+
+    # Pycryptodome only supports byte strings.
+    seed = seed.encode('utf-8')
+    password = password.encode('utf-8')
 
     # Discard the first <sequence> keys
     thehash = MD4.new(seed + password).digest()