mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-02-17 06:48:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ciphers/caesar_cipher.py with type hints (#3860)

Browse Source 
* Update caesar_cipher.py

improved for conciseness and readability

* Add type hints

Co-authored-by: Dhruv Manilawala <dhruvmanila@gmail.com>
This commit is contained in:
Abdeldjaouad Nusayr Medakene 2020-12-10 18:25:57 +01:00 committed by GitHub
parent 75759fae22
commit 110a740d5d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
ciphers/caesar_cipher.py
View File

@ -1,7 +1,8 @@
from string import ascii_letters
from typing import Dict, Optional
def encrypt(input_string: str, key: int, alphabet=None) -> str:
def encrypt(input_string: str, key: int, alphabet: Optional[str] = None) -> str:
"""
encrypt
=======
@ -79,7 +80,7 @@ def encrypt(input_string: str, key: int, alphabet=None) -> str:
return result
def decrypt(input_string: str, key: int, alphabet=None) -> str:
def decrypt(input_string: str, key: int, alphabet: Optional[str] = None) -> str:
"""
decrypt
=======
@ -144,7 +145,7 @@ def decrypt(input_string: str, key: int, alphabet=None) -> str:
return encrypt(input_string, key, alphabet)
def brute_force(input_string: str, alphabet=None) -> dict:
def brute_force(input_string: str, alphabet: Optional[str] = None) -> Dict[int, str]:
"""
brute_force
===========
@ -193,31 +194,18 @@ def brute_force(input_string: str, alphabet=None) -> dict:
# Set default alphabet to lower and upper case english chars
alpha = alphabet or ascii_letters
# The key during testing (will increase)
key = 1
# The encoded result
result = ""
# To store data on all the combinations
brute_force_data = {}
# Cycle through each combination
while key <= len(alpha):
# Decrypt the message
result = decrypt(input_string, key, alpha)
# Update the data
brute_force_data[key] = result
# Reset result and increase the key
result = ""
key += 1
for key in range(1, len(alpha) + 1):
# Decrypt the message and store the result in the data
brute_force_data[key] = decrypt(input_string, key, alpha)
return brute_force_data
def main():
if __name__ == "__main__":
while True:
print(f'\n{"-" * 10}\n Menu\n{"-" * 10}')
print(*["1.Encrypt", "2.Decrypt", "3.BruteForce", "4.Quit"], sep="\n")
@ -248,7 +236,3 @@ def main():
elif choice == "4":
print("Goodbye.")
break
if __name__ == "__main__":
main()