add special compatible with arcadia

Author: rekby

ydb/_grpc/common/__init__.py

@@ -1,4 +1,5 @@
 import sys
+import importlib.util
 
 import google.protobuf
 from packaging.version import Version
@@ -8,15 +9,24 @@
 # sdk code must always import from ydb._grpc.common
 protobuf_version = Version(google.protobuf.__version__)
 
-if protobuf_version < Version("4.0"):
-    from ydb._grpc.v3 import *  # noqa
-    from ydb._grpc.v3 import protos  # noqa
+# for compatible with arcadia
+if importlib.util.find_spec("ydb.public.api"):
+    from ydb.public.api.grpc import *
+    sys.modules["ydb._grpc.common"] = sys.modules["ydb.public.api.grpc"]
 
-    sys.modules["ydb._grpc.common"] = sys.modules["ydb._grpc.v3"]
-    sys.modules["ydb._grpc.common.protos"] = sys.modules["ydb._grpc.v3.protos"]
+    from ydb.public.api import protos
+    sys.modules["ydb._grpc.common.protos"] = sys.modules["ydb.public.api.protos"]
 else:
-    from ydb._grpc.v4 import *  # noqa
-    from ydb._grpc.v4 import protos  # noqa
+    # common way, outside of arcadia
+    if protobuf_version < Version("4.0"):
+        from ydb._grpc.v3 import *  # noqa
+        sys.modules["ydb._grpc.common"] = sys.modules["ydb._grpc.v3"]
 
-    sys.modules["ydb._grpc.common"] = sys.modules["ydb._grpc.v4"]
-    sys.modules["ydb._grpc.common.protos"] = sys.modules["ydb._grpc.v4.protos"]
+        from ydb._grpc.v3 import protos  # noqa
+        sys.modules["ydb._grpc.common.protos"] = sys.modules["ydb._grpc.v3.protos"]
+    else:
+        from ydb._grpc.v4 import *  # noqa
+        sys.modules["ydb._grpc.common"] = sys.modules["ydb._grpc.v4"]
+
+        from ydb._grpc.v4 import protos  # noqa
+        sys.modules["ydb._grpc.common.protos"] = sys.modules["ydb._grpc.v4.protos"]