Add options for enabling/disabling sysprep tasks

[snf-image-creator] / image_creator / __init__.py

diff --git a/image_creator/__init__.py b/image_creator/__init__.py
index 817a9c6..afdda63 100644 (file)
--- a/image_creator/__init__.py
+++ b/image_creator/__init__.py
@@ -1,4 +1,4 @@
-# Copyright 2011 GRNET S.A. All rights reserved.
+# Copyright 2012 GRNET S.A. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or
 # without modification, are permitted provided that the following
@@ -32,3 +32,25 @@
 # or implied, of GRNET S.A.
 
 __version__ = '0.1'
+
+
+import image_creator.os_type
+
+
+def get_os_class(distro, osfamily):
+    module = None
+    classname = None
+    try:
+        module = __import__("image_creator.os_type.%s"
+            % distro, fromlist=['image_creator.os_type'])
+        classname = distro.capitalize()
+    except ImportError:
+        module = __import__("image_creator.os_type.%s"
+            % osfamily, fromlist=['image_creator.os_type'])
+        classname = osfamily.capitalize()
+
+    return getattr(module, classname)
+
+
+
+# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :