Prompt for new settings if categories are not found/changes

chadsr · Oct 25, 2018 · 868dcc2 · 868dcc2
1 parent 6026383
commit 868dcc2
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/nordnm/settings.py b/nordnm/settings.py
@@ -5,6 +5,7 @@
 import logging
 import os
 import ipaddress
+import sys
 
 
 class SettingsHandler(object):
@@ -103,9 +104,15 @@ def get_categories(self):
         categories = []
 
         for category in nordapi.VPN_CATEGORIES.keys():
-            category_name = category.replace(' ', '-')
-            if self.settings.getboolean('Categories', category_name):
-                categories.append(category)
+            category_name = category.replace(' ', '-').lower()
+
+            try:
+                if self.settings.getboolean('Categories', category_name):
+                    categories.append(category)
+            except configparser.NoOptionError:
+                self.logger.error("VPN category '%s' not found in '%s'. Update settings and try again." % (category_name, self.path))
+                self.save_new_settings()
+                sys.exit(0)
 
         return categories