Signed-off-by: Jeroen Oudshoorn <oudshoorn.jeroen@gmail.com>
This commit is contained in:
Jeroen Oudshoorn
2023-10-13 00:10:39 +02:00
parent 09639fd5c9
commit 4c7f5b6c9a

View File

@ -16,7 +16,6 @@ def add_parsers(subparsers):
# pwnagotchi plugins search # pwnagotchi plugins search
parser_google_auth = google_subparsers.add_parser('auth', help='Google Authentication') parser_google_auth = google_subparsers.add_parser('auth', help='Google Authentication')
parser_google_auth.add_argument('bool', type=bool, help="This will start the authentication process")
return subparsers return subparsers
@ -33,29 +32,28 @@ def handle_cmd(args):
Parses the arguments and does the thing the user wants Parses the arguments and does the thing the user wants
""" """
if args.plugincmd == 'auth': if args.plugincmd == 'auth':
return auth(args.bool) return auth()
elif args.plugincmd == 'refresh': elif args.plugincmd == 'refresh':
return refresh(args) return refresh(args)
raise NotImplementedError() raise NotImplementedError()
def auth(args): def auth():
if args == "true": # start authentication process
# start authentication process user_input = input("By completing these steps you give pwnagotchi access to your personal Google Drive!\n"
user_input = input("By completing these steps you give pwnagotchi access to your personal Google Drive!\n" "Personal credentials will be stored only locally for automated verification in the future.\n"
"Personal credentials will be stored only locally for automated verification in the future.\n" "No one else but you have access to these.\n"
"No one else but you have access to these.\n" "Do you agree? \n\n[y(es)/n(o)]")
"Do you agree? \n\n[y(es)/n(o)]") if user_input.lower() in ('y', 'yes'):
if user_input.lower() in ('y', 'yes'): try:
try: gauth = GoogleAuth(settings_file="settings.yaml")
gauth = GoogleAuth(settings_file="settings.yaml") print(gauth.GetAuthUrl())
print(gauth.GetAuthUrl()) user_input = input("Please copy this URL into a browser, "
user_input = input("Please copy this URL into a browser, " "complete the verification and then copy/paste the code from addressbar.")
"complete the verification and then copy/paste the code from addressbar.") gauth.Auth(user_input)
gauth.Auth(user_input) gauth.SaveCredentialsFile("credentials.json")
gauth.SaveCredentialsFile("credentials.json") except Exception as e:
except Exception as e: logging.error(f"Error: {e}")
logging.error(f"Error: {e}")
return 0 return 0