mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-10-06 13:49:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #46 from Henocks/patch-2

Browse Source 
Slight Performance/Visual Update
This commit is contained in:
Harshil 2016-11-09 19:55:27 +05:30 committed by GitHub
commit 3f505c5b39
1 changed files with 11 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
ciphers/caesar_cipher.py
View File

@ -2,8 +2,8 @@
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"
@ -11,41 +11,34 @@ def main():
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()