integration valid if all checks pass, add settings

Author: Matt Sokoloff

labelbox/client.py

@@ -551,7 +551,8 @@ def create_dataset(self, iam_integration=IAMIntegration._DEFAULT, **kwargs):
                 """mutation validateDatasetPyApi($id: ID!){validateDataset(where: {id : $id}){
                     valid checks{name, success}}}
                 """, {'id': dataset.uid})
-            if not validation_result['validateDataset']['checks'][0]['success']:
+            breakpoint()
+            if not validation_result['validateDataset']['valid']:
                 raise labelbox.exceptions.LabelboxError(
                     f"IAMIntegration {validation_result['validateDataset']['checks']['name']} was not successfully added added to the project."
                 )

labelbox/schema/iam_integration.py

@@ -15,6 +15,11 @@ class IAMIntegration(DbObject):
         is_org_default (boolean)
 
     """
+
+    def __init__(self, client, data):
+        self.settings = data.pop('settings', {})
+        super().__init__(client, data)
+
     _DEFAULT = "DEFAULT"
 
     name = Field.String("name")

labelbox/schema/organization.py

@@ -135,8 +135,15 @@ def get_iam_integrations(self):
         Returns all IAM Integrations for an organization
         """
         res = self.client.execute(
-            """query getAllIntegrationsPyApi { iamIntegrations {%s} } """ %
-            query.results_query_part(Entity.IAMIntegration))
+            """query getAllIntegrationsPyApi { iamIntegrations {
+                %s
+                settings {
+                __typename
+                ... on AwsIamIntegrationSettings {roleArn}
+                ... on GcpIamIntegrationSettings {serviceAccountEmailId readBucket}
+                }
+
+            } } """ % query.results_query_part(Entity.IAMIntegration))
         return [
             Entity.IAMIntegration(self.client, integration_data)
             for integration_data in res['iamIntegrations']