1 files changed, 36 insertions, 0 deletions
diff --git a/azurelinuxagent/pa/deprovision/factory.py b/azurelinuxagent/pa/deprovision/factory.py
new file mode 100644
index 0000000..dd01633
--- /dev/null
+++ b/azurelinuxagent/pa/deprovision/factory.py
@@ -0,0 +1,36 @@
+# Copyright 2014 Microsoft Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# Requires Python 2.4+ and Openssl 1.0+
+#
+
+import azurelinuxagent.common.logger as logger
+from azurelinuxagent.common.utils.textutil import Version
+from azurelinuxagent.common.version import DISTRO_NAME, DISTRO_VERSION, \
+                                     DISTRO_FULL_NAME
+
+from .default import DeprovisionHandler
+from .coreos import CoreOSDeprovisionHandler
+from .ubuntu import UbuntuDeprovisionHandler
+
+def get_deprovision_handler(distro_name=DISTRO_NAME, 
+                            distro_version=DISTRO_VERSION,
+                            distro_full_name=DISTRO_FULL_NAME):
+    if distro_name == "ubuntu":
+        return UbuntuDeprovisionHandler()
+    if distro_name == "coreos":
+        return CoreOSDeprovisionHandler()
+
+    return DeprovisionHandler()
+