diff --git a/block.py b/block.py new file mode 100644 index 0000000..c3b4efb --- /dev/null +++ b/block.py @@ -0,0 +1,33 @@ +from lib.input import * +from lib.vars import alphabet +from lib.vars import listToString + +if( inputHasKeys(["-k", "-i", "-t"]) ): + in_key = getValueOfKey("-k") + in_txt = getValueOfKey("-i") + in_mode = int(getValueOfKey("-t")) + blockSize = len(in_key) +else: + print("file.py -k {int KEY} -i {string TXT} -t {MODE}") + print("-k: The encryption/decryption key") + print("-i: The text to be encrypted/decrypted") + print("-t: The mode (0=encrypt or 1=decrypt)") + exit() + +def keyToIntTuple(key): + return tuple(map(int, list(key))) + +def applyKeyToBlock(keyT, blockL): # function to apply a key to a block + blockDe = [""] * blockSize + + for i in range(blockSize): + blockDe[i] = blockL[keyT[i]] + + print(blockDe) + return blockDe + +KEY = keyToIntTuple(in_key) # define and make the key a tuple so that we can index it + +applyKeyToBlock(keyToIntTuple(in_key), list(in_txt)) + + diff --git a/caesar.py b/caesar.py index 3740b4c..016daf6 100644 --- a/caesar.py +++ b/caesar.py @@ -26,7 +26,8 @@ for char in txt_list: # loop through all of the chars print("Decrypting char-index: " + str(charindex) + " (" + char + ")") index = index + in_key # shift the alphabet - while( index > alen - 1 ): #cycle through the alphabet + while( index > alen - 1 ): #cycle through the alphabet + print("Alphabet cycle") index = (index + in_key) - (alen - 1) decryp_list[charindex] = alphabet[in_alphabet][index]