fix arcadia compatibility

Author: vgvoleg

ydb/_grpc/common/__init__.py

@@ -10,7 +10,15 @@
 protobuf_version = Version(google.protobuf.__version__)
 
 # for compatible with arcadia
-if _utilities.check_module_exists("ydb.public.api"):
+if _utilities.check_module_exists("contrib.ydb.public.api"):
+    from contrib.ydb.public.api.grpc import *  # noqa
+
+    sys.modules["ydb._grpc.common"] = sys.modules["contrib.ydb.public.api.grpc"]
+
+    from contrib.ydb.public.api import protos
+
+    sys.modules["ydb._grpc.common.protos"] = sys.modules["contrib.ydb.public.api.protos"]
+elif _utilities.check_module_exists("ydb.public.api"):
     from ydb.public.api.grpc import *  # noqa
 
     sys.modules["ydb._grpc.common"] = sys.modules["ydb.public.api.grpc"]

ydb/draft/_apis.py

@@ -11,13 +11,22 @@
         ydb_dynamic_config_pb2,
     )
 else:
-    from .._grpc.common.draft import (
-        ydb_dynamic_config_v1_pb2_grpc,
-    )
-
-    from .._grpc.common.draft.protos import (
-        ydb_dynamic_config_pb2,
-    )
+    try:
+        from .._grpc.common.draft import (
+            ydb_dynamic_config_v1_pb2_grpc,
+        )
+
+        from .._grpc.common.draft.protos import (
+            ydb_dynamic_config_pb2,
+        )
+    except ImportError:
+        from .._grpc.common.draft import (
+            ydb_dynamic_config_v1_pb2_grpc,
+        )
+
+        from .._grpc.common.protos.draft import (
+            ydb_dynamic_config_pb2,
+        )
 
 
 ydb_dynamic_config = ydb_dynamic_config_pb2