service: get_oscmd now raises OSCmdError

projectatomic · Mar 8, 2017 · 8eaedc4 · 8eaedc4
1 parent 4d5c876
commit 8eaedc4
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/commissaire_service/oscmd/__init__.py b/src/commissaire_service/oscmd/__init__.py
@@ -18,6 +18,15 @@
 
 import importlib
 
+from commissaire.errors import CommissaireError
+
+
+class OSCmdError(CommissaireError):
+    """
+    OSCmd specific errors.
+    """
+    pass
+
 
 class OSCmdBase:
     """
@@ -113,5 +122,4 @@ def get_oscmd(os_type):
         module = importlib.import_module(module_name)
         return getattr(module, 'OSCmd')
     except ImportError:
-        # TODO: Make this a specific exception
-        raise Exception('No OSCmd class for {}'.format(os_type))
+        raise OSCmdError('No OSCmd class for {}'.format(os_type))