From 8753c0decccac69c1c22b741e63c790cd059d2d5 Mon Sep 17 00:00:00 2001 From: Ehud Adler Date: Tue, 31 Dec 2019 11:12:08 -0500 Subject: [PATCH] unsafe --- src/core/grade_notifier.py | 11 +++++++---- src/core/initializegn.py | 6 +++++- 2 files changed, 12 insertions(+), 5 deletions(-) diff --git a/src/core/grade_notifier.py b/src/core/grade_notifier.py index f515e43..31c3de2 100644 --- a/src/core/grade_notifier.py +++ b/src/core/grade_notifier.py @@ -69,6 +69,7 @@ redacted_print_err = None client = None state = None +__unsafe = False ''' Sends a text message via Twilio @@ -250,7 +251,8 @@ def start_notifier(): cursor.execute(f'UPDATE Users SET lastUpdated = NOW() WHERE id={__id};') - decrypted_password = decrypt(encrypted_password, 'keys/private.pem') + decrypted_password = if (__unsafe) encrypted_password \ + else decrypt(encrypted_password, 'keys/private.pem') api = CUNYFirstAPI(username, decrypted_password, school.upper()) api.login() @@ -289,8 +291,9 @@ def parse(): parser = argparse.ArgumentParser( description='Specify commands for CUNY Grade Notifier Retriever v1.0') + parser.add_argument('--unsafe') # Development parser.add_argument('--prod') # Production - parser.add_argument('--enable_phone') # Development + parser.add_argument('--enable_phone') return parser.parse_args() def initialize_twilio(): @@ -298,11 +301,11 @@ def initialize_twilio(): client = Client(account_sid, auth_token) def main(): - global state + global state, __unsafe args = parse() state = LoginState.determine_state(args) - + __unsafe = True if args.unsafe else False #default is that its safe try: # Only initialize twilio in production # or when specifically asked diff --git a/src/core/initializegn.py b/src/core/initializegn.py index d8a6f10..fbffa8a 100644 --- a/src/core/initializegn.py +++ b/src/core/initializegn.py @@ -106,6 +106,8 @@ def parse(): # Development parser.add_argument('--enable_phone') + parser.add_argument('--unsafe') # Development + return parser.parse_args() @@ -132,7 +134,9 @@ def main(): return - password = decrypt(encrypted_password, 'keys/private.pem') + password = if (args.unsafe) encrypted_password \ + else decrypt(encrypted_password, 'keys/private.pem') + # print(password) api = cunyfirstapi.CUNYFirstAPI(username, password) api.login()