add unknown item type to scheme ls

Author: Valeriya Popova

CHANGELOG.md

@@ -1,6 +1,7 @@
 ## 2.12.0 ##
 * Fixed error message while get token from metadata with asyncio iam
 * Add `SnapshotReadOnly` transaction mode support to `session.transaction`
+* Support `TYPE_UNSPECIFIED` item type to scheme ls
 
 ## 2.11.1 ##
 * Regenerate protobuf code from public api protos (some private protobufs was removed)

ydb/scheme.py

@@ -11,7 +11,7 @@ class SchemeEntryType(enum.IntEnum):
     """
     Enumerates all available entry types.
     """
-
+    TYPE_UNSPECIFIED = 0
     DIRECTORY = 1
     TABLE = 2
     PERS_QUEUE_GROUP = 3
@@ -24,6 +24,10 @@ class SchemeEntryType(enum.IntEnum):
     REPLICATION = 16
     TOPIC = 17
 
+    @classmethod
+    def _missing_(cls, value):
+        return cls.TYPE_UNSPECIFIED
+
     @staticmethod
     def is_table(entry):
         """
@@ -299,7 +303,7 @@ def _wrap_scheme_entry(entry_pb, scheme_entry_cls=None, *args, **kwargs):
     return scheme_entry_cls(
         entry_pb.name,
         entry_pb.owner,
-        getattr(SchemeEntryType, _apis.ydb_scheme.Entry.Type.Name(entry_pb.type)),
+        SchemeEntryType(entry_pb.type),
         _wrap_permissions(entry_pb.effective_permissions),
         _wrap_permissions(entry_pb.permissions),
         entry_pb.size_bytes,
@@ -318,7 +322,7 @@ def _wrap_list_directory_response(rpc_state, response):
     message = _apis.ydb_scheme.ListDirectoryResult()
     response.operation.result.Unpack(message)
     children = []
-    supported_items = set([i.value for i in SchemeEntryType])
+    supported_items = set(i.value for i in SchemeEntryType)
     for children_item in message.children:
         if children_item.type not in supported_items:
             continue
@@ -328,12 +332,10 @@ def _wrap_list_directory_response(rpc_state, response):
     return Directory(
         message.self.name,
         message.self.owner,
-        getattr(SchemeEntryType, _apis.ydb_scheme.Entry.Type.Name(message.self.type)),
+        SchemeEntryType(message.self.type),
         _wrap_permissions(message.self.effective_permissions),
         _wrap_permissions(message.self.permissions),
-        tuple(
-            children,
-        ),
+        tuple(children),
     )
 
 

ydb/scheme_test.py

@@ -0,0 +1,16 @@
+from ydb.scheme import SchemeEntryType, _wrap_scheme_entry, _wrap_list_directory_response
+from ydb._apis import ydb_scheme
+
+
+def test_wrap_scheme_entry():
+    assert _wrap_scheme_entry(ydb_scheme.Entry(type=1)).type is SchemeEntryType.DIRECTORY
+    assert _wrap_scheme_entry(ydb_scheme.Entry(type=17)).type is SchemeEntryType.TOPIC
+
+    assert _wrap_scheme_entry(ydb_scheme.Entry()).type is SchemeEntryType.TYPE_UNSPECIFIED
+    assert _wrap_scheme_entry(ydb_scheme.Entry(type=10)).type is SchemeEntryType.TYPE_UNSPECIFIED
+    assert _wrap_scheme_entry(ydb_scheme.Entry(type=1001)).type is SchemeEntryType.TYPE_UNSPECIFIED
+
+
+def test_wrap_list_directory_response():
+    d = _wrap_list_directory_response(None, ydb_scheme.ListDirectoryResponse())
+    assert d.type is SchemeEntryType.TYPE_UNSPECIFIED