TheAlgorithms/Python
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix(mypy): Fix annotations for 13 cipher algorithms (#4278)

Browse Source 
* Initial fix for mypy errors in some cipher algorithms

* fix(mypy): Update type hints

* fix(mypy): Update type hints for enigma_machine2.py

* Update as per the suggestion

Co-authored-by: Christian Clauss <cclauss@me.com>

Co-authored-by: Christian Clauss <cclauss@me.com>
This commit is contained in:
Dhruv Manilawala
2021-03-22 12:29:51 +05:30
committed by GitHub
parent 99a42f2b58
commit 14bcb580d5
13 changed files with 101 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
ciphers/decrypt_caesar_with_chi_squared.py
View File

@@ -1,14 +1,14 @@
#!/usr/bin/env python3
from typing import Tuple
from typing import Optional
def decrypt_caesar_with_chi_squared(
ciphertext: str,
cipher_alphabet: str = None,
frequencies_dict: str = None,
cipher_alphabet: Optional[list[str]] = None,
frequencies_dict: Optional[dict[str, float]] = None,
case_sensetive: bool = False,
) -> Tuple[int, float, str]:
) -> tuple[int, float, str]:
"""
Basic Usage
===========
@@ -123,9 +123,9 @@ def decrypt_caesar_with_chi_squared(
AttributeError: 'int' object has no attribute 'lower'
"""
alphabet_letters = cipher_alphabet or [chr(i) for i in range(97, 123)]
frequencies_dict = frequencies_dict or {}
if frequencies_dict == {}:
# If the argument is None or the user provided an empty dictionary
if not frequencies_dict:
# Frequencies of letters in the english language (how much they show up)
frequencies = {
"a": 0.08497,
@@ -163,7 +163,7 @@ def decrypt_caesar_with_chi_squared(
ciphertext = ciphertext.lower()
# Chi squared statistic values
chi_squared_statistic_values = {}
chi_squared_statistic_values: dict[int, tuple[float, str]] = {}
# cycle through all of the shifts
for shift in range(len(alphabet_letters)):
@@ -215,22 +215,22 @@ def decrypt_caesar_with_chi_squared(
chi_squared_statistic += chi_letter_value
# Add the data to the chi_squared_statistic_values dictionary
chi_squared_statistic_values[shift] = [
chi_squared_statistic_values[shift] = (
chi_squared_statistic,
decrypted_with_shift,
]
)
# Get the most likely cipher by finding the cipher with the smallest chi squared
# statistic
most_likely_cipher = min(
most_likely_cipher: int = min(
chi_squared_statistic_values, key=chi_squared_statistic_values.get
)
) # type: ignore # First argument to `min` is not optional
# Get all the data from the most likely cipher (key, decoded message)
most_likely_cipher_chi_squared_value = chi_squared_statistic_values[
most_likely_cipher
][0]
decoded_most_likely_cipher = chi_squared_statistic_values[most_likely_cipher][1]
(
most_likely_cipher_chi_squared_value,
decoded_most_likely_cipher,
) = chi_squared_statistic_values[most_likely_cipher]
# Return the data on the most likely shift
return (