fixed error (TheAlgorithms#3281)

this will fix code from randomly throwing `SystemExit: The affine cipher becomes weak when key B is set to 0. Choose different key` exception.
infinityxxx · Oct 14, 2020 · 35eefac · 35eefac
1 parent f164e11
commit 35eefac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ciphers/affine_cipher.py b/ciphers/affine_cipher.py
@@ -94,7 +94,7 @@ def get_random_key():
     while True:
         keyA = random.randint(2, len(SYMBOLS))
         keyB = random.randint(2, len(SYMBOLS))
-        if cryptomath.gcd(keyA, len(SYMBOLS)) == 1:
+        if cryptomath.gcd(keyA, len(SYMBOLS)) == 1 and keyB % len(SYMBOLS) != 0:
             return keyA * len(SYMBOLS) + keyB