Merge pull request #174 fix grpc import workaround template

Author: rekby

ydb/_apis.py

@@ -1,7 +1,8 @@
 # -*- coding: utf-8 -*-
+import typing
+
 # Workaround for good IDE and universal for runtime
-# noinspection PyUnreachableCode
-if False:
+if typing.TYPE_CHECKING:
     from ._grpc.v4 import (
         ydb_cms_v1_pb2_grpc,
         ydb_discovery_v1_pb2_grpc,

ydb/_grpc/grpcwrapper/common_utils.py

@@ -24,8 +24,7 @@
 import ydb.aio
 
 # Workaround for good IDE and universal for runtime
-# noinspection PyUnreachableCode
-if False:
+if typing.TYPE_CHECKING:
     from ..v4.protos import ydb_topic_pb2, ydb_issue_message_pb2
 else:
     from ..common.protos import ydb_topic_pb2, ydb_issue_message_pb2

ydb/_grpc/grpcwrapper/ydb_topic.py

@@ -10,8 +10,7 @@
 from ... import scheme
 
 # Workaround for good IDE and universal for runtime
-# noinspection PyUnreachableCode
-if False:
+if typing.TYPE_CHECKING:
     from ..v4.protos import ydb_scheme_pb2, ydb_topic_pb2
 else:
     from ..common.protos import ydb_scheme_pb2, ydb_topic_pb2

ydb/_grpc/grpcwrapper/ydb_topic_public_types.py

@@ -1,11 +1,11 @@
 import datetime
+import typing
 from dataclasses import dataclass, field
 from enum import IntEnum
 from typing import Optional, List, Union, Dict
 
 # Workaround for good IDE and universal for runtime
-# noinspection PyUnreachableCode
-if False:
+if typing.TYPE_CHECKING:
     from ..v4.protos import ydb_topic_pb2
 else:
     from ..common.protos import ydb_topic_pb2

ydb/_topic_common/common_test.py

@@ -1,4 +1,5 @@
 import asyncio
+import typing
 
 import grpc
 import pytest
@@ -10,9 +11,8 @@
 )
 from .. import issues
 
-# Workaround for good autocomplete in IDE and universal import at runtime
-# noinspection PyUnreachableCode
-if False:
+# Workaround for good IDE and universal for runtime
+if typing.TYPE_CHECKING:
     from ydb._grpc.v4.protos import (
         ydb_status_codes_pb2,
         ydb_topic_pb2,

ydb/_topic_reader/topic_reader_asyncio_test.py

@@ -1,5 +1,6 @@
 import asyncio
 import datetime
+import typing
 from unittest import mock
 
 import pytest
@@ -12,9 +13,8 @@
 from .._grpc.grpcwrapper.ydb_topic import StreamReadMessage, Codec, OffsetsRange
 from .._topic_common.test_helpers import StreamMock, wait_condition, wait_for_fast
 
-# Workaround for good autocomplete in IDE and universal import at runtime
-# noinspection PyUnreachableCode
-if False:
+# Workaround for good IDE and universal for runtime
+if typing.TYPE_CHECKING:
     from .._grpc.v4.protos import ydb_status_codes_pb2
 else:
     from .._grpc.common.protos import ydb_status_codes_pb2

ydb/aio/connection.py

@@ -1,5 +1,6 @@
 import logging
 import asyncio
+import typing
 from typing import Any, Tuple, Callable, Iterable
 import collections
 import grpc
@@ -24,8 +25,8 @@
 from ydb.settings import BaseRequestSettings
 from ydb import issues
 
-# Workaround for good IDE and universal runtime
-if False:
+# Workaround for good IDE and universal for runtime
+if typing.TYPE_CHECKING:
     from ydb._grpc.v4 import ydb_topic_v1_pb2_grpc
 else:
     from ydb._grpc.common import ydb_topic_v1_pb2_grpc

ydb/credentials.py

@@ -1,5 +1,7 @@
 # -*- coding: utf-8 -*-
 import abc
+import typing
+
 import six
 from . import tracing, issues, connection
 from . import settings as settings_impl
@@ -9,8 +11,7 @@
 import time
 
 # Workaround for good IDE and universal for runtime
-# noinspection PyUnreachableCode
-if False:
+if typing.TYPE_CHECKING:
     from ._grpc.v4.protos import ydb_auth_pb2
     from ._grpc.v4 import ydb_auth_v1_pb2_grpc
 else:

ydb/export.py

@@ -1,12 +1,12 @@
 import enum
+import typing
 
 from . import _apis
 
 from . import settings_impl as s_impl
 
 # Workaround for good IDE and universal for runtime
-# noinspection PyUnreachableCode
-if False:
+if typing.TYPE_CHECKING:
     from ._grpc.v4.protos import ydb_export_pb2
     from ._grpc.v4 import ydb_export_v1_pb2_grpc
 else:

ydb/import_client.py

@@ -1,12 +1,12 @@
 import enum
+import typing
 
 from . import _apis
 
 from . import settings_impl as s_impl
 
 # Workaround for good IDE and universal for runtime
-# noinspection PyUnreachableCode
-if False:
+if typing.TYPE_CHECKING:
     from ._grpc.v4.protos import ydb_import_pb2
     from ._grpc.v4 import ydb_import_v1_pb2_grpc
 else: