mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-10-16 18:51:14 +00:00
2d5dd6f132
* MAINT: Used f-string method Updated the code with f-string methods wherever required for a better and cleaner understanding of the code. * Updated files with f-string method * Update rsa_key_generator.py * Update rsa_key_generator.py * Update elgamal_key_generator.py * Update lru_cache.py I don't think this change is efficient but it might tackle the error as the error was due to using long character lines. * Update lru_cache.py * Update lru_cache.py Co-authored-by: cyai <seriesscar@gmail.com> Co-authored-by: Christian Clauss <cclauss@me.com>
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
|
|
from . import transposition_cipher as transCipher
|
|
|
|
|
|
def main() -> None:
|
|
inputFile = "Prehistoric Men.txt"
|
|
outputFile = "Output.txt"
|
|
key = int(input("Enter key: "))
|
|
mode = input("Encrypt/Decrypt [e/d]: ")
|
|
|
|
if not os.path.exists(inputFile):
|
|
print(f"File {inputFile} does not exist. Quitting...")
|
|
sys.exit()
|
|
if os.path.exists(outputFile):
|
|
print(f"Overwrite {outputFile}? [y/n]")
|
|
response = input("> ")
|
|
if not response.lower().startswith("y"):
|
|
sys.exit()
|
|
|
|
startTime = time.time()
|
|
if mode.lower().startswith("e"):
|
|
with open(inputFile) as f:
|
|
content = f.read()
|
|
translated = transCipher.encryptMessage(key, content)
|
|
elif mode.lower().startswith("d"):
|
|
with open(outputFile) as f:
|
|
content = f.read()
|
|
translated = transCipher.decryptMessage(key, content)
|
|
|
|
with open(outputFile, "w") as outputObj:
|
|
outputObj.write(translated)
|
|
|
|
totalTime = round(time.time() - startTime, 2)
|
|
print(("Done (", totalTime, "seconds )"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|