diff --git a/authenticator.py b/authenticator.py index 802bba6..7ee40db 100644 --- a/authenticator.py +++ b/authenticator.py @@ -24,13 +24,16 @@ def api_auth(user, passwd, host): def main(): """Main logic entrypoint""" - config_path = os.path.join(os.path.expanduser("~"), "config.ini") - config = configparser.ConfigParser() - config.read(config_path) + try: + config_path = os.path.join(os.path.expanduser("~"), "config.ini") + config = configparser.ConfigParser() + config.read(config_path) - API_HOST = config['API']['HOST'] - API_USER = config['API']['USER'] - PASSWORD = config['API']['PASSWORD'] + API_HOST = config['API']['HOST'] + API_USER = config['API']['USER'] + PASSWORD = config['API']['PASSWORD'] + except configparser.Error: + return auth_token = api_auth(API_USER, PASSWORD, API_HOST) print auth_token