Slight Performance/Visual Update

Author: haryunio

ciphers/caesar_cipher.py

@@ -2,50 +2,43 @@
 
 def main():
     message = input("Enter message: ")
-    key = int(input("Key [1-26]: "))
-    mode = input("Encrypt or Decrypt [e/d]: ")
+    key     = int(input("Key [1-26]: "))
+    mode    = input("Encrypt or Decrypt [e/d]: ")
 
     if mode.lower().startswith('e'):
         mode = "encrypt"
     elif mode.lower().startswith('d'):
         mode = "decrypt"
 
     translated = encdec(message, key, mode)
-    if mode == "encrypt":
+    if mode ==   "encrypt":
         print("Encryption:", translated)
     elif mode == "decrypt":
         print("Decryption:", translated)
 
 def encdec(message, key, mode):
-    """
-    >>> encdec('Harshil Darji', 12, 'encrypt')
-    'TMDETUX PMDVU'
-
-    >>> encdec('TMDETUX PMDVU', 12, 'decrypt')
-    'HARSHIL DARJI'
-    """
-    message = message.upper()
+    message    = message.upper()
     translated = ""
-    LETTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+    LETTERS    = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
     for symbol in message:
         if symbol in LETTERS:
             num = LETTERS.find(symbol)
-            if mode == "encrypt":
+            if mode ==   "encrypt":
                 num = num + key
             elif mode == "decrypt":
                 num = num - key
 
             if num >= len(LETTERS):
-                num = num - len(LETTERS)
+                num -= len(LETTERS)
             elif num < 0:
-                num = num + len(LETTERS)
+                num += len(LETTERS)
 
-            translated = translated + LETTERS[num]
+            translated += LETTERS[num]
         else:
-            translated = translated + symbol
+            translated += symbol
     return translated
 
 if __name__ == '__main__':
     import doctest
     doctest.testmod()
-    main()
+    main()