Added Random key generation

This commit is contained in:
Willow Behar
2020-10-16 12:29:56 -05:00
parent 47363c3726
commit e0fc2358a0

View File

@ -5,6 +5,7 @@ Description: A Caesar Cipher coding challenge given by the Crunix Club"
"""
import sys
import random
help_text = """
NAME
@ -26,6 +27,9 @@ help_text = """
"""
def main():
#Help
if "-h" in sys.argv:
print(help_text)
#Get Key
if "-k" in sys.argv:
#Find key flag location
@ -33,12 +37,10 @@ def main():
#Get Key Integer
key = int(sys.argv[key_flag_location + 1])
else:
print("Missing Key")
print(help_text)
if "-h" in sys.argv:
#Help
print(help_text)
elif "-E" in sys.argv:
#Randomly generate a key
key = random.randint(0, 100)
print("Key:", key)
if "-E" in sys.argv:
#Get Encrypt Flag Location
encrypt_flag_location = sys.argv.index("-E")
#Get String to encrypt