update improves error handling

Author: loadept

qtheme/cli/kitty.py

@@ -9,13 +9,17 @@ def __init__(self, user_path) -> None:
     def set_terminal_theme(self, theme: str = None):
         try:
             kitty_conf_path = path.join(self.user_path, '.config', 'kitty', 'kitty.conf')
+            kitty_theme_path = path.join(self.user_path, '.config', 'kitty', 'themes')
 
             if theme is None:
                 return
 
             if not path.exists(kitty_conf_path):
                 raise FileNotFoundError('The conf file kitty does not exist.')
 
+            if not path.exists(path.join(kitty_theme_path, theme + '.conf')):
+                raise FileNotFoundError(f'The file "{theme}.conf" does not exist')
+            
             with open(kitty_conf_path, 'r+') as file:
                 lines = file.readlines()
                 for index, line in enumerate(lines):

qtheme/cli/qtile.py

@@ -65,11 +65,12 @@ def set_bar_position(self, position: str = None):
     def set_qtile_theme(self, theme: str = None):
         try:
             full_path = path.join(self.user_path, '.config', 'qtile', 'themes', 'theme_selector.json')
+            qtile_theme_path = path.join(self.user_path, '.config', 'qtile', 'themes')
 
             if theme is None:
                 return
 
-            if not path.exists(path.join(self.user_path, '.config', 'qtile', 'themes', theme + '.json')):
+            if not path.exists(path.join(qtile_theme_path, theme + '.json')):
                 raise FileNotFoundError(f'The file "{theme}.json" does not exist')
 
             with open(full_path, 'r+') as file: