vastdb 0.0.5.10__py3-none-any.whl → 0.0.5.12__py3-none-any.whl

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.dist-info → vastdb-0.0.5.12.dist-info}/METADATA

@@ -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.dist-info → vastdb-0.0.5.12.dist-info}/RECORD

@@ -148,9 +148,9 @@ vast_flatbuf/tabular/ObjectDetails.py,sha256=5mMZ0iL_iC1cztyyjPeEeJv3ch6Nk1zyCOc
 vast_flatbuf/tabular/S3File.py,sha256=KC9c2oS5-JXwTTriUVFdjOvRG0B54Cq9kviSDZY3NI0,4450
 vast_flatbuf/tabular/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vastdb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vastdb/api.py,sha256=nqsUzGeEPWSuO402dUQp_-6BPdZI0EXjBxc48GifA1M,131408
-vastdb-0.0.5.10.dist-info/LICENSE,sha256=obffan7LYrq7hLHNrY7vHcn2pKUTBUYXMKu-VOAvDxU,11333
-vastdb-0.0.5.10.dist-info/METADATA,sha256=i1r5ba8-9wFSZhHtqAp2lcbhZAZd8CvMDucLdSDZr-c,1370
-vastdb-0.0.5.10.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-vastdb-0.0.5.10.dist-info/top_level.txt,sha256=Vsj2MKtlhPg0J4so64slQtnwjhgoPmJgcG-6YcVAwVc,20
-vastdb-0.0.5.10.dist-info/RECORD,,
+vastdb/api.py,sha256=nJSpwwIP4f_6bGZqDcsgMlSPeqrRuL7-Lgr_lwy3dBs,132142
+vastdb-0.0.5.12.dist-info/LICENSE,sha256=obffan7LYrq7hLHNrY7vHcn2pKUTBUYXMKu-VOAvDxU,11333
+vastdb-0.0.5.12.dist-info/METADATA,sha256=V0va4DHkpsLvmR92TfU6SG3uhn4e2B3isz_UzPqw0YU,1370
+vastdb-0.0.5.12.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+vastdb-0.0.5.12.dist-info/top_level.txt,sha256=Vsj2MKtlhPg0J4so64slQtnwjhgoPmJgcG-6YcVAwVc,20
+vastdb-0.0.5.12.dist-info/RECORD,,

{vastdb-0.0.5.10.dist-info → vastdb-0.0.5.12.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.45.1)
+Generator: bdist_wheel (0.37.1)
 Root-Is-Purelib: true
 Tag: py3-none-any