if os.geteuid() != 0:
msg = "[sudo] password for %u to continue:"
ret = subprocess.check_call("sudo -v -p '%s'" % msg, shell=True)
return ret
if __name__=='__main__':
try:
print(banner)
if prompt_sudo() != 0:
print("You need SUDO for this!")
exit()
st_dsp()
answer = input("$ ")
if int(answer) == 1:
aes_q()
so = input("$ ")
if so == 'A':
AES.eAES()
print("Your file has been encrypted!")
p = input("Where do you want to save your file?")
os.system('mv dataout.txt.aes' + ' ' + str(p))
os.system('rm -rf' + str(path))
else:
AES.dAES()
print("Your file has been decrypted!")
elif int(answer) == 2:
# This one to be edited.
pass
except:
print("Theres an Error! Check the path && Unvalid inputs!")
