From d9d28eab41ee738e563f144feffa0a793fe444cc Mon Sep 17 00:00:00 2001 From: Stanislav Shchetinin Date: Wed, 12 Feb 2025 19:22:59 +0300 Subject: [PATCH 1/2] added new stage --- ydb/import_client.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ydb/import_client.py b/ydb/import_client.py index d94294ca..d74c686b 100644 --- a/ydb/import_client.py +++ b/ydb/import_client.py @@ -29,6 +29,7 @@ class ImportProgress(enum.IntEnum): DONE = 4 CANCELLATION = 5 CANCELLED = 6 + CREATE_CHANGEFEEDS = 7 def _initialize_progresses(): From 3decd9321e1dbdba9c35190149138a7e90850080 Mon Sep 17 00:00:00 2001 From: st-shchetinin Date: Thu, 13 Feb 2025 11:28:55 +0300 Subject: [PATCH 2/2] try/except in import_client --- ydb/import_client.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/ydb/import_client.py b/ydb/import_client.py index d74c686b..e967e877 100644 --- a/ydb/import_client.py +++ b/ydb/import_client.py @@ -29,12 +29,14 @@ class ImportProgress(enum.IntEnum): DONE = 4 CANCELLATION = 5 CANCELLED = 6 - CREATE_CHANGEFEEDS = 7 def _initialize_progresses(): for key, value in ydb_import_pb2.ImportProgress.Progress.items(): - _progresses[value] = getattr(ImportProgress, key[len("PROGRESS_") :]) + try: + _progresses[value] = getattr(ImportProgress, key[len("PROGRESS_") :]) + except AttributeError: + pass _initialize_progresses()