Quick vsphere fix

rubrik_polaris/common/graphql/query_core_event_series_list.graphql

@@ -36,4 +36,4 @@ query RubrikPolarisSDKRequest($after: String, $filters: ActivitySeriesFilter, $f
         }
     }
 
-}
+}

rubrik_polaris/compute/vsphere.py

@@ -23,22 +23,11 @@
 Collection of functions that manipulate vSphere compute components
 """
 
+def get_compute_vsphere(self, filter=None):
+    """Retrieves all VMware VM object details and implements gql filter (Under development)
 
-def get_compute_object_ids_vsphere(self, match_all=True, **kwargs):
-    """Retrieves all vSphere objects that match query
-
-    Arguments:
-        match_all {bool} -- Set to false to match ANY defined criteria
-        kwargs {} -- Any top level object from the get_compute_ec2 call
-    """
-    try:
-        return self._get_object_ids_instances(self.get_instances_vsphere(), kwargs, match_all=match_all)
-    except Exception:
-        raise
-
-
-def get_compute_vsphere(self):
-    """Retrieves all VMware VM object details (Under development)
+    Args:
+        filter (dict): Limit of results to retrieve. Uses GQL filter (ex- [{"field": "NAME", "texts": ["100gb-centos-001"]}])
 
     Returns:
         dict: details of VMware VM objects
@@ -47,11 +36,11 @@ def get_compute_vsphere(self):
         RequestException: If the query to Polaris returned an error
     """
     try:
-        query_name = "compute_vmware_vsphere"
+        query_name = "compute_vmware_vsphere_list"
         # self._validate(
         #     query_name=query_name
         # )
-        variables = {"filter": [], "first": 500}
+        variables = {"filter": filter, "first": 500}
         return self._query(query_name, variables)
     except Exception:
-        raise
+        raise

rubrik_polaris/rubrik_polaris.py

@@ -54,7 +54,7 @@ class PolarisClient:
         submit_compute_restore_ec2
     from .compute.azurevm import get_compute_object_ids_azure, get_compute_azure, submit_compute_restore_azure
     from .compute.gce import get_compute_object_ids_gce, get_compute_gce, submit_compute_restore_gce
-    from .compute.vsphere import get_compute_vsphere, get_compute_object_ids_vsphere
+    from .compute.vsphere import get_compute_vsphere
     from .storage.ebs import get_storage_object_ids_ebs, get_storage_ebs
     from .common.graphql import get_enum_values
     from .cluster import get_cdm_cluster_location, get_cdm_cluster_connection_status

sample/polaris_client.py

@@ -95,6 +95,8 @@
 
 ### Returns all objectIDs matching arbitrary available inputs. ec2 tags have special treatment
 # pp.pprint(rubrik.get_compute_object_ids_ec2(tags = {"Name": "Puppet Master"}))
+# pp.pprint(rubrik.get_compute_object_ids_vsphere(name="O365-003"))
+# pp.pprint(rubrik.get_compute_vsphere(filter=[{"field": "NAME", "texts": ["100gb-centos-001"]}]))
 # pp.pprint(rubrik.get_compute_object_ids_azure(region = "EastUS2"))
 # pp.pprint(rubrik.get_compute_object_ids_gce(region = "us-west1"))