First commit

This commit is contained in:
Bucchino Geoffrey 2022-06-22 20:42:48 +02:00
parent 7bd9325a73
commit c9b432c900
2 changed files with 77 additions and 0 deletions

75
main.py Normal file

@ -0,0 +1,75 @@
# coding: utf-8
from argparse import ArgumentParser
import matplotlib.pyplot as plt
from lorem import text as loremtext
def checkArguments():
args = ArgumentParser(description='Cryptanalyst')
args.add_argument('-f', '--filename', help='Text to analyze')
args.add_argument('-s', '--stats', help='Statistics', action='store_true')
args.add_argument('-d', '--decrypt', help='Decrypt Caesar text', action='store_true')
return args.parse_args()
def readCipherText(filename):
file = []
with open(filename, 'r') as f:
data = f.readlines()
for entry in data:
file.append(entry.strip())
return file
def main():
args = checkArguments()
cipherFile = None
if args.filename:
cipherFile = readCipherText(args.filename)
if args.stats:
stats(cipherFile)
if args.decrypt:
decryptCaesarText(cipherFile)
def stats(file=None):
if file is None:
text = loremtext()
else:
text = file
lettersOccurence = {}
letterUpperToAscii = ord('A') # Get the ASCII code
for i in range (0, 26):
lettersOccurence[chr(i + letterUpperToAscii)] = 0
for line in text:
for letter in line:
letter = letter.upper()
if letter in lettersOccurence:
lettersOccurence[letter] = lettersOccurence[letter] + 1
x = [chr(letter + 65) for letter in range(0, 26)]
y = [lettersOccurence[tmp] for tmp in lettersOccurence]
largeur = 0.5
plt.bar(x, y, largeur)
plt.show()
def decryptCaesarText(cipherFile):
if cipherFile is None:
return
#for lines in cipherFile:
# splitLines = lines.split(' ')
# for word in splitLines:
# if len(word) <= 3:
# print(word)
if __name__ == "__main__":
main()

2
test.txt.crypt Normal file

@ -0,0 +1,2 @@
AJGNNQ YQTNF!
LG UWKU WP OGUUCIG GP ENCKT !!