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!")