Make script more robust

Author: martignoni

bin/changewifisettings.py

@@ -84,23 +84,26 @@ def is_pi3():
 dhcpcd_conf_file = "/etc/dhcpcd.conf"
 
 # New values taken from settings_file.
-with open(settings_file, 'r') as f:
-    # Inject section manually and read all file.
-    # Needed as configparser 3.11 cannot read INI files without sections.
-    config_string = '[wifi]\n' + f.read()
-settings = configparser.ConfigParser()
 try:
-    settings.read_string(config_string)
-    section = settings['wifi']
-    new_channel = section.get('channel', default_channel)
-    new_country = section.get('country', default_country)
-    new_password = section.get('password', default_password)
-    new_ssid = section.get('ssid', default_ssid)
-    password_protected = section.get('passwordprotected', '1')
-    ssid_hidden_state = section.get('ssidhiddenstate', '0')
-    new_static_ip = section.get('ipaddress', default_ip_address)
-except (KeyError, configparser.Error) as e:
-    sys.exit(f"Invalid settings file: {e}")
+    with open(settings_file, 'r') as f:
+        # Inject section manually and read all file.
+        # Needed as configparser 3.11 cannot read INI files without sections.
+        config_string = '[wifi]\n' + f.read()
+    settings = configparser.ConfigParser()
+    try:
+        settings.read_string(config_string)
+        section = settings['wifi']
+        new_channel = section.get('channel', default_channel)
+        new_country = section.get('country', default_country)
+        new_password = section.get('password', default_password)
+        new_ssid = section.get('ssid', default_ssid)
+        password_protected = section.get('passwordprotected', '1')
+        ssid_hidden_state = section.get('ssidhiddenstate', '0')
+        new_static_ip = section.get('ipaddress', default_ip_address)
+    except (KeyError, configparser.Error) as e:
+        sys.exit(f"Invalid settings file: {e}")
+except FileNotFoundError as e:
+    sys.exit(f"Settings file not found: {e.filename}")
 
 # Action functions.