OpenSoure_Repos/TheAlgorithms-Python
1
0
Fork 0
mirror of https://hub.njuu.cf/TheAlgorithms/Python.git synced 2023-10-11 13:06:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make decrypt_caesar_with_chi_squared work with upper case letters (#5379)

Browse Source 
* Fixes: #5323

* Fixes: #5323
This commit is contained in:
happiestbee 2021-10-31 06:46:31 -04:00 committed by GitHub
parent 9ac94c09eb
commit f4fd147d03
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
ciphers/decrypt_caesar_with_chi_squared.py
View File

@ -6,7 +6,7 @@ def decrypt_caesar_with_chi_squared(
ciphertext: str, ciphertext: str,
cipher_alphabet: list[str] | None = None, cipher_alphabet: list[str] | None = None,
frequencies_dict: dict[str, float] | None = None, frequencies_dict: dict[str, float] | None = None,
case_sensetive: bool = False, case_sensitive: bool = False,
) -> tuple[int, float, str]: ) -> tuple[int, float, str]:
""" """
Basic Usage Basic Usage
@ -20,7 +20,7 @@ def decrypt_caesar_with_chi_squared(
* frequencies_dict (dict): a dictionary of word frequencies where keys are * frequencies_dict (dict): a dictionary of word frequencies where keys are
the letters and values are a percentage representation of the frequency as the letters and values are a percentage representation of the frequency as
a decimal/float a decimal/float
* case_sensetive (bool): a boolean value: True if the case matters during * case_sensitive (bool): a boolean value: True if the case matters during
decryption, False if it doesn't decryption, False if it doesn't
Returns: Returns:
@ -117,6 +117,9 @@ def decrypt_caesar_with_chi_squared(
>>> decrypt_caesar_with_chi_squared('crybd cdbsxq') >>> decrypt_caesar_with_chi_squared('crybd cdbsxq')
(10, 233.35343938980898, 'short string') (10, 233.35343938980898, 'short string')
>>> decrypt_caesar_with_chi_squared('Crybd Cdbsxq', case_sensitive=True)
(10, 233.35343938980898, 'Short String')
>>> decrypt_caesar_with_chi_squared(12) >>> decrypt_caesar_with_chi_squared(12)
Traceback (most recent call last): Traceback (most recent call last):
AttributeError: 'int' object has no attribute 'lower' AttributeError: 'int' object has no attribute 'lower'
@ -158,7 +161,7 @@ def decrypt_caesar_with_chi_squared(
# Custom frequencies dictionary # Custom frequencies dictionary
frequencies = frequencies_dict frequencies = frequencies_dict
if not case_sensetive: if not case_sensitive:
ciphertext = ciphertext.lower() ciphertext = ciphertext.lower()
# Chi squared statistic values # Chi squared statistic values
@ -172,10 +175,14 @@ def decrypt_caesar_with_chi_squared(
for letter in ciphertext: for letter in ciphertext:
try: try:
# Try to index the letter in the alphabet # Try to index the letter in the alphabet
new_key = (alphabet_letters.index(letter) - shift) % len( new_key = (alphabet_letters.index(letter.lower()) - shift) % len(
alphabet_letters alphabet_letters
) )
decrypted_with_shift += alphabet_letters[new_key] decrypted_with_shift += (
alphabet_letters[new_key].upper()
if case_sensitive and letter.isupper()
else alphabet_letters[new_key]
)
except ValueError: except ValueError:
# Append the character if it isn't in the alphabet # Append the character if it isn't in the alphabet
decrypted_with_shift += letter decrypted_with_shift += letter
@ -184,10 +191,11 @@ def decrypt_caesar_with_chi_squared(
# Loop through each letter in the decoded message with the shift # Loop through each letter in the decoded message with the shift
for letter in decrypted_with_shift: for letter in decrypted_with_shift:
if case_sensetive: if case_sensitive:
letter = letter.lower()
if letter in frequencies: if letter in frequencies:
# Get the amount of times the letter occurs in the message # Get the amount of times the letter occurs in the message
occurrences = decrypted_with_shift.count(letter) occurrences = decrypted_with_shift.lower().count(letter)
# Get the excepcted amount of times the letter should appear based # Get the excepcted amount of times the letter should appear based
# on letter frequencies # on letter frequencies