Merge pull request #188 from sarahbx/pod_disruption_budget

Add PodDisruptionBudget class

Author: myakove

ocp_resources/pod_disruption_budget.py

@@ -0,0 +1,52 @@
+from ocp_resources.resource import NamespacedResource
+
+
+class PodDisruptionBudget(NamespacedResource):
+    """
+    PodDisruptionBudget object
+    """
+
+    api_group = NamespacedResource.ApiGroup.POLICY
+
+    def __init__(
+        self,
+        name=None,
+        namespace=None,
+        client=None,
+        min_available=None,
+        max_unavailable=None,
+        selector=None,
+        teardown=True,
+        yaml_file=None,
+    ):
+        super().__init__(
+            name=name,
+            namespace=namespace,
+            client=client,
+            teardown=teardown,
+            yaml_file=yaml_file,
+        )
+        self.min_available = min_available
+        self.max_unavailable = max_unavailable
+        self.selector = selector
+
+    def to_dict(self):
+        res = super().to_dict()
+        if self.yaml_file:
+            return res
+
+        update_dict = {
+            "spec": {
+                "selector": self.selector,
+            },
+        }
+
+        if self.min_available is not None:
+            update_dict["spec"]["minAvailable"] = self.min_available
+
+        if self.max_unavailable is not None:
+            update_dict["spec"]["maxUnavailable"] = self.max_unavailable
+
+        res.update(update_dict)
+
+        return res

ocp_resources/resource.py

@@ -300,6 +300,7 @@ class ApiGroup:
         OPERATORS_COREOS_COM = "operators.coreos.com"
         OS_TEMPLATE_KUBEVIRT_IO = "os.template.kubevirt.io"
         PACKAGES_OPERATORS_COREOS_COM = "packages.operators.coreos.com"
+        POLICY = "policy"
         PROJECT_OPENSHIFT_IO = "project.openshift.io"
         RBAC_AUTHORIZATION_K8S_IO = "rbac.authorization.k8s.io"
         RIPSAW_CLOUDBULLDOZER_IO = "ripsaw.cloudbulldozer.io"