vastdb 0.0.5.10__tar.gz → 0.0.5.12__tar.gz

{vastdb-0.0.5.10 → vastdb-0.0.5.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vastdb
-Version: 0.0.5.10
+Version: 0.0.5.12
 Summary: VAST Data SDK
 Home-page: https://github.com/vast-data/vastdb_sdk
 Author: VAST DATA

{vastdb-0.0.5.10 → vastdb-0.0.5.12}/setup.py

@@ -46,7 +46,7 @@ if os.environ.get('VASTDB_APPEND_VERSION_SUFFIX'):
 setup(
     name='vastdb',
     description='VAST Data SDK',
-    version='0.0.5.10' + suffix,
+    version='0.0.5.12' + suffix,
     url='https://github.com/vast-data/vastdb_sdk',
     author='VAST DATA',
     author_email='hello@vastdata.com',

{vastdb-0.0.5.10 → vastdb-0.0.5.12}/vastdb/api.py

@@ -1,6 +1,7 @@
 import array
 import logging
 import struct
+import time
 import urllib.parse
 from collections import defaultdict, namedtuple
 from datetime import datetime
@@ -1016,6 +1017,15 @@ class VastdbApi:
 
             return snapshots, is_truncated, marker
 
+    def _make_sure_schema_exists(self, bucket, schema, timeout_sec=3):
+        start_time = time.time()
+        while time.time() - start_time < timeout_sec:
+            _, schemas, _, _, _ = self.list_schemas(bucket)
+            if schema in (x[0] for x in schemas):
+                return
+            self.create_schema(bucket, schema)
+        raise Exception(f"Failed to find or create schema {schema} in {timeout_sec} seconds.")
+ 
     def create_table(self, bucket, schema, name, arrow_schema=None, txid=0, client_tags=[], expected_retvals=[],
                      create_imports_table=False, use_external_row_ids_allocation=False):
         return self._create_table_internal(bucket=bucket, schema=schema, name=name, arrow_schema=arrow_schema,
@@ -1026,6 +1036,7 @@ class VastdbApi:
     def create_topic(self, bucket, name, topic_partitions, expected_retvals=[],
                      message_timestamp_type=None, retention_ms=None,
                      message_timestamp_after_max_ms=None, message_timestamp_before_max_ms=None):
+        self._make_sure_schema_exists(bucket, KAFKA_TOPICS_SCHEMA_NAME)
         table_props = _encode_table_props(message_timestamp_type=message_timestamp_type,
                                           retention_ms=retention_ms,
                                           message_timestamp_after_max_ms=message_timestamp_after_max_ms,
@@ -1094,6 +1105,8 @@ class VastdbApi:
         # create the table
         return self.create_table(bucket, schema, name, arrow_schema, txid, client_tags, expected_retvals)
 
+    def get_topic_stats(self, bucket, name, expected_retvals=[]):
+        return self.get_table_stats(bucket=bucket, schema=KAFKA_TOPICS_SCHEMA_NAME, name=name, expected_retvals=expected_retvals)
 
     def get_table_stats(self, bucket, schema, name, txid=0, client_tags=[], expected_retvals=[]):
         """
@@ -1127,7 +1140,7 @@ class VastdbApi:
             table_properties = ""
 
         return self.alter_table(bucket=bucket, schema=KAFKA_TOPICS_SCHEMA_NAME, name=name,
-                                table_properties=table_properties, expected_retvals=expected_retvals)
+                                table_properties=table_properties, new_name=new_name, expected_retvals=expected_retvals)
 
     def alter_table(self, bucket, schema, name, txid=0, client_tags=[], table_properties="",
                     new_name="", expected_retvals=[]):

{vastdb-0.0.5.10 → vastdb-0.0.5.12}/vastdb.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vastdb
-Version: 0.0.5.10
+Version: 0.0.5.12
 Summary: VAST Data SDK
 Home-page: https://github.com/vast-data/vastdb_sdk
 Author: VAST DATA