kobai-sdk 0.3.5rc1__py3-none-any.whl → 0.3.5rc4__py3-none-any.whl

kobai/mobi_config.py

@@ -11,7 +11,7 @@ class MobiSettings(BaseSettings):
     mobi_password: str = "admin"
     cookies: str = ""
     use_cookies: bool = False
-    verify_ssl = False
+    verify_ssl: bool = False
 
     catalog_name: str = "http://mobi.com/catalog-local"
     default_tenant_id: str = "00000000-0000-0000-0000-000000000000"

kobai/tenant_client.py

@@ -73,7 +73,7 @@ class TenantClient:
         """
         self._init_post_auth_success(access_token)
 
-    def use_access_token(self, access_token: str, id_token: str = None,  tenant_id: str = None):
+    def use_access_token(self, access_token: str, id_token: str = None,  tenant_id: str = None, token_provider: str = None):
 
         """
         Authenticate the TenantClient with the Kobai instance. Returns nothing, but stores bearer token in client.
@@ -88,7 +88,8 @@ class TenantClient:
             "tenantName" : self.tenant_name,
             "tenantId" : tenant_id,
             "idToken" : id_token,
-            "accessToken" : access_token
+            "accessToken" : access_token,
+            "tokenProvider" : token_provider
         }
 
         response = self.api_client._TenantAPI__run_post(
@@ -218,13 +219,13 @@ class TenantClient:
         else:
             return True
 
-    def spark_generate_genie_views(self, domains = None, concepts = None, not_concepts=None, enforce_map=True):
+    def spark_generate_genie_views(self, domains = None, concepts = None, not_concepts=None, enforce_map=True, local=False):
 
         """
         Use the Spark Client to generate views for this tenant required to populate a Genie Data Room.
         """
 
-        tables = self.__get_view_sql(domains=domains, concepts=concepts, not_concepts=not_concepts, enforce_map=enforce_map)
+        tables = self.__get_view_sql(domains=domains, concepts=concepts, not_concepts=not_concepts, enforce_map=enforce_map, local=False)
         for t in tables:
             #print(t["sql"])
             try:
@@ -325,15 +326,26 @@ class TenantClient:
 # Semantic Profile
 ########################################
 
-    def __get_descriptions(self):
+    #def get_descriptions(self):
+    #    response = self.api_client._TenantAPI__run_get("/episteme-svcs/api/descriptions", params={"schema": self.schema})
+    #    return response.json()
 
-        tenant_config = self.get_tenant_config()
-        descriptions = get_genie_descriptions(self.model_id, tenant_config, self.schema)
-        return descriptions
+    def __get_descriptions(self, local):
 
-    def __get_view_sql(self, domains=None, concepts=None, not_concepts=None, enforce_map=True):
+        if local:
+            tenant_config = self.get_tenant_config()
+            descriptions = get_genie_descriptions(self.model_id, tenant_config, self.schema)
+            return descriptions
+        else:
+            response = self.api_client._TenantAPI__run_get("/episteme-svcs/api/descriptions", params={"schema": self.schema})
+            return response.json()
+        #tenant_config = self.get_tenant_config()
+        #descriptions = get_genie_descriptions(self.model_id, tenant_config, self.schema)
+        #return descriptions
+
+    def __get_view_sql(self, domains=None, concepts=None, not_concepts=None, enforce_map=True, local=False):
         sql_list = []
-        descriptions = self.__get_descriptions()
+        descriptions = self.__get_descriptions(local)
         
         for dom in descriptions["domains"]:
             for con in dom["concepts"]:

{kobai_sdk-0.3.5rc1.dist-info → kobai_sdk-0.3.5rc4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kobai-sdk
-Version: 0.3.5rc1
+Version: 0.3.5rc4
 Summary: A package that enables interaction with a Kobai tenant.
 Author-email: Ryan Oattes <ryan@kobai.io>
 License:                                  Apache License

{kobai_sdk-0.3.5rc1.dist-info → kobai_sdk-0.3.5rc4.dist-info}/RECORD

@@ -5,13 +5,13 @@ kobai/databricks_client.py,sha256=fyqqMly2Qm0r1AHWsQjkYeNsDdH0G1JSgTkF9KJ55qA,21
 kobai/demo_tenant_client.py,sha256=wlNc-bdI2wotRXo8ppUOalv4hYdBlek_WzJNARZV-AE,9293
 kobai/genie.py,sha256=-EbEYpu9xj_3zIXaPdwbNJEAmoeM7nb9qK-h1f_STtM,8061
 kobai/mobi.py,sha256=1pIAdn9qKmg2zm_DnH1qAsgKr-YgH6SJJWXEzSZch38,32787
-kobai/mobi_config.py,sha256=AkrGtwDNMPuwt-udbp_8KsLnLGze4S8-at2UH-zN9YQ,525
+kobai/mobi_config.py,sha256=e_O4S7YjgGOb3xEOzNDx1AELuwMwXP7F3i0kGp65IrI,531
 kobai/ms_authenticate.py,sha256=f7t_BaxK8e-SUWoB6JliBEngF2iduF-COcCZq83H5t0,2297
 kobai/spark_client.py,sha256=opM_F-4Ut5Hq5zZjWMuLvUps9sDULvyPNZHXGL8dW1k,776
 kobai/tenant_api.py,sha256=Q5yuFd9_V4lo3LWzvYEEO3LpDRWFgQD4TlRPXDTGbiE,4368
-kobai/tenant_client.py,sha256=EeJhjdg1iFKZm0L7iSwbg4FEXbJjh6iWmdGLdKb0yC4,43470
-kobai_sdk-0.3.5rc1.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-kobai_sdk-0.3.5rc1.dist-info/METADATA,sha256=0QRyijyJKePuaBrZ1vBYO4yjoC6NVxuUy7Uq9rS1CAU,19864
-kobai_sdk-0.3.5rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kobai_sdk-0.3.5rc1.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
-kobai_sdk-0.3.5rc1.dist-info/RECORD,,
+kobai/tenant_client.py,sha256=ODlkl4SnkQN5ojuZNjh_d1Ianqb_Tv6v7SRdxwatyBw,44158
+kobai_sdk-0.3.5rc4.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+kobai_sdk-0.3.5rc4.dist-info/METADATA,sha256=C3r6lkzXDjvxeKSm9X49Ysdkf688ADyPWMbPa9-Qr5w,19864
+kobai_sdk-0.3.5rc4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+kobai_sdk-0.3.5rc4.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
+kobai_sdk-0.3.5rc4.dist-info/RECORD,,

{kobai_sdk-0.3.5rc1.dist-info → kobai_sdk-0.3.5rc4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any