techrunner/Coding-Challenges
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added File manipulation

Browse Source
This commit is contained in:
Willow Behar
2020-10-16 13:14:57 -05:00
parent 79f1775324
commit f9f27cc804
1 changed files with 57 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
Willow_Cypher_Coding_Challenge.py
View File

@ -20,13 +20,21 @@ help_text = """
Decrypts String
-h
Prints Help Screen
-EF
Encrypt File
-DF
Decrypt File
-o
Save output to file instead of printing to terminal
AUTHOR
Written by Willow Behar.
LICENSING
GPLv3
"""
def main():
file_output_flag = False
#Help
if "-h" in sys.argv:
print(help_text)
@ -40,13 +48,29 @@ def main():
#Randomly generate a key
key = random.randint(1, 100)
print("Key:", key)
if "-o" in sys.argv:
#Set flag to true
file_output_flag = True
#Get Location of Flag
file_output_location = sys.argv.index("-o")
#Get File Name
file_output_name = sys.argv[file_output_location + 1]
#Open file
file_output_open = open(file_output_name, "w")
if "-E" in sys.argv:
#Get Encrypt Flag Location
encrypt_flag_location = sys.argv.index("-E")
#Get String to encrypt
encrypt_string = str(sys.argv[encrypt_flag_location + 1])
#Encrypt
print(Encrypt(key, encrypt_string))
encrypted_phrase = Encrypt(key, encrypt_string)
if file_output_flag:
file_output_open.write(encrypted_phrase)
else:
print(encrypted_phrase)
elif "-D" in sys.argv:
#Get Decrypt flag location
decrypt_flag_location = sys.argv.index("-D")
@ -54,6 +78,38 @@ def main():
decrypt_string = sys.argv[decrypt_flag_location + 1]
#Decrypt
print(Decrypt(key, decrypt_string))
elif "-EF" in sys.argv:
#Get Encrypt File Flag
encrypt_file_location = sys.argv.index("-EF")
#Get File Location
encrypt_file_name = str(sys.argv[encrypt_file_location + 1])
#Open File
encrypt_file_open = open(encrypt_file_open, "r")
#Read File
encrypt_file_data = encrypt_file_open.read()
#Encrypt File
encrypted_file_data = Encrypt(key, encrypt_file_data)
if file_output_flag:
file_output_open.write(encrypted_file_data)
else:
print(encrypted_file_data)
elif "-DF" in sys.argv:
#Get Decrypt File Flag
decrypt_file_location = sys.argv.index("-DF")
#Get File location
decrypt_file_name = str(sys.argv[decrypt_file_location + 1])
#Open File
decrypt_file_open = open(decrypt_file_name, "r")
#Read File
decrypt_file_data = decrypt_file_open.read()
#Decrypt File
decrypted_file_data = Decrypt(key, decrypt_file_data)
if file_output_flag:
file_output_open.write(decrypted_file_data)
else:
print(decrypted_file_data)
else:
print(help_text)