def main():
random.seed(40)
for i in range(20):
# 初始化带测试的字符串
message = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" * random.randint(4, 40)
message = list(message)
random.shuffle(message)
message = "".join(message)
print('Test #%s: "%s..."' % (i + 1, message))
for key in range(1, len(message)):
encrypted = transpositionClipher.encryptMessage(key, message)
decrypted = transpositionDecrypt.decryptMessage(key, encrypted)
if message != decrypted:
print("Mismatch with key %s and message %s." % (key, message))
print(decrypted)
sys.exit()
print("Transposition cipher test passed")
def main():
inputFilename = 'frankenstein.txt'
outputFilename = 'frankenstein.encrypted.txt'
myKey = 10
myMode = 'encrypt'
if not os.path.exists(inputFilename):
print("the file %s does not exist.Quiting..." % (inputFilename))
sys.exit()
if os.path.exists(outputFilename):
print("this will overwrite the file %s. (C)ontinue or (Q)uite?" % (outputFilename))
response = raw_input('>')
if not response.lower().startswith('c'):
sys.exit()
fileObj = open(inputFilename)
content = fileObj.read()
fileObj.close()
print('%sing...' % (myMode.title()))
startTime = time.time()
if myMode == 'encrypt':
translated = transpositionClipher.encryptMessage(myKey, content)
elif myMode == 'decrypt':
translated = transpositionDecrypt.decryptMessage(myKey, content)
totalTime = round(time.time() - startTime, 2)
print('%sion time: %s seconds' % (myMode.title(), totalTime))
outputFileObj = open(outputFilename, 'w')
outputFileObj.write(translated)
outputFileObj.close()
print('Done %sing %s (%s, characters).' % (myMode, inputFilename, len(content)))
print('%sed file is %s.' % (myMode.title(), outputFilename))