diff --git a/src/core/grade_notifier.py b/src/core/grade_notifier.py index ab5dbe9..b7bdff0 100644 --- a/src/core/grade_notifier.py +++ b/src/core/grade_notifier.py @@ -66,6 +66,7 @@ redacted_print_err = None client = None state = None +__unsafe = False ''' Sends a text message via Twilio @@ -247,7 +248,9 @@ def start_notifier(): cursor.execute(f'UPDATE Users SET lastUpdated = NOW() WHERE id={__id};') - decrypted_password = decrypt(encrypted_password, '../../private/keys/private.pem') + decrypted_password = if (__unsafe) encrypted_password \ + else decrypt(encrypted_password, '../../private/keys/private.pem') + api = CUNYFirstAPI(username, decrypted_password, school.upper()) api.login() @@ -285,8 +288,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(): @@ -294,11 +298,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 846a68e..d8f9717 100644 --- a/src/core/initializegn.py +++ b/src/core/initializegn.py @@ -101,6 +101,8 @@ def parse(): # Development parser.add_argument('--enable_phone') + parser.add_argument('--unsafe') # Development + return parser.parse_args() @@ -126,8 +128,10 @@ def main(): ) return - password = decrypt(encrypted_password, '../../private/keys/private.pem') - + + password = if (args.unsafe) encrypted_password \ + else decrypt(encrypted_password, '../../private/keys/private.pem') + api = cunyfirstapi.CUNYFirstAPI(username, password) api.login() if api.is_logged_in():