fix arcadia compatibility

Author: vgvoleg

ydb/_grpc/common/__init__.py

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

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