[SB] next-pylint (Azure#38860)

* next-pylint

* pylint

* pylint2

* remove diff

* nit

* fix

* kashif

* mypy

* Update sdk/servicebus/azure-servicebus/azure/servicebus/_pyamqp/utils.py

Co-authored-by: Kashif Khan <[email protected]>

* Revert "Update sdk/servicebus/azure-servicebus/azure/servicebus/_pyamqp/utils.py"

This reverts commit 85b80d4.

* fix

* fix

---------

Co-authored-by: Kashif Khan <[email protected]>

Author: l0lawrence

sdk/servicebus/azure-servicebus/azure/servicebus/_base_handler.py

@@ -38,7 +38,6 @@
     from .exceptions import ServiceBusError
 
     try:
-        # pylint:disable=unused-import
         from uamqp import AMQPClient as uamqp_AMQPClientSync
     except ImportError:
         pass
@@ -372,7 +371,6 @@ def _check_live(self):
     def _do_retryable_operation(  # pylint: disable=inconsistent-return-statements
         self, operation: Callable, timeout: Optional[float] = None, **kwargs: Any
     ) -> Any:
-        # pylint: disable=protected-access
         require_last_exception = kwargs.pop("require_last_exception", False)
         operation_requires_timeout = kwargs.pop("operation_requires_timeout", False)
         retried_times = 0
@@ -435,7 +433,7 @@ def _backoff(
         )
         if backoff <= self._config.retry_backoff_max and (
             abs_timeout_time is None or (backoff + time.time()) <= abs_timeout_time
-        ):  # pylint:disable=no-else-return
+        ):
             time.sleep(backoff)
             _LOGGER.info(
                 "%r has an exception (%r). Retrying...",
@@ -517,7 +515,7 @@ def _mgmt_request_response(
                 timeout=timeout,
                 callback=callback,
             )
-        except Exception as exp:  # pylint: disable=broad-except
+        except Exception as exp:
             if isinstance(exp, self._amqp_transport.TIMEOUT_ERROR):
                 raise OperationTimeoutError(error=exp) from exp
             raise

sdk/servicebus/azure-servicebus/azure/servicebus/_common/message.py

@@ -47,7 +47,6 @@
 
 if TYPE_CHECKING:
     try:
-        # pylint:disable=unused-import
         from uamqp import Message, BatchMessage
     except ImportError:
         pass
@@ -59,7 +58,7 @@
 PrimitiveTypes = Union[int, float, bytes, bool, str, uuid.UUID]
 
 
-class ServiceBusMessage(object):  # pylint: disable=too-many-public-methods,too-many-instance-attributes
+class ServiceBusMessage(object):  # pylint: disable=too-many-instance-attributes
     """A Service Bus Message.
 
     :param body: The data to send in a single message.
@@ -671,7 +670,7 @@ def _add(self, add_message: Union[ServiceBusMessage, Mapping[str, Any], AmqpAnno
             raise MessageSizeExceededError(
                 message=f"ServiceBusMessageBatch has reached its size limit: {self.max_size_in_bytes}"
             )
-        self._amqp_transport.add_batch(self, outgoing_sb_message)  # pylint: disable=protected-access
+        self._amqp_transport.add_batch(self, outgoing_sb_message)
         self._size = size_after_add
         self._count += 1
         self._messages.append(outgoing_sb_message)

sdk/servicebus/azure-servicebus/azure/servicebus/_common/mgmt_handlers.py

@@ -17,7 +17,7 @@ def default(status_code, message, description, amqp_transport):  # pylint: disab
     if status_code == 200:
         return message.value
 
-    amqp_transport.handle_amqp_mgmt_error(  # pylint: disable=protected-access
+    amqp_transport.handle_amqp_mgmt_error(
         _LOGGER, "Service request failed.", condition, description, status_code
     )
 
@@ -29,7 +29,7 @@ def session_lock_renew_op(  # pylint: disable=inconsistent-return-statements
     if status_code == 200:
         return message.value
 
-    amqp_transport.handle_amqp_mgmt_error(  # pylint: disable=protected-access
+    amqp_transport.handle_amqp_mgmt_error(
         _LOGGER, "Session lock renew failed.", condition, description, status_code
     )
 
@@ -42,7 +42,7 @@ def message_lock_renew_op(  # pylint: disable=inconsistent-return-statements
         # TODO: will this always be body type ValueType?
         return message.value
 
-    amqp_transport.handle_amqp_mgmt_error(  # pylint: disable=protected-access
+    amqp_transport.handle_amqp_mgmt_error(
         _LOGGER, "Message lock renew failed.", condition, description, status_code
     )
 
@@ -58,7 +58,7 @@ def peek_op(  # pylint: disable=inconsistent-return-statements
     if status_code in [202, 204]:
         return []
 
-    amqp_transport.handle_amqp_mgmt_error(  # pylint: disable=protected-access
+    amqp_transport.handle_amqp_mgmt_error(
         _LOGGER, "Message peek failed.", condition, description, status_code
     )
 

sdk/servicebus/azure-servicebus/azure/servicebus/_common/receiver_mixins.py

@@ -67,14 +67,13 @@ def _populate_attributes(self, **kwargs):
             )
 
     def _get_source(self):
-        # pylint: disable=protected-access
         if self._session:
             session_filter = None if self._session_id == NEXT_AVAILABLE_SESSION else self._session_id
             return self._amqp_transport.create_source(self._entity_uri, session_filter)
         return self._entity_uri
 
     def _check_message_alive(self, message, action):
-        # pylint: disable=no-member, protected-access
+        # pylint: disable=protected-access
         if message._is_peeked_message:
             raise ValueError(
                 f"The operation {action} is not supported for peeked messages."

sdk/servicebus/azure-servicebus/azure/servicebus/_common/tracing.py

@@ -25,7 +25,6 @@
 
 if TYPE_CHECKING:
     try:
-        # pylint:disable=unused-import
         from uamqp import Message as uamqp_Message
     except ImportError:
         uamqp_Message = None
@@ -229,7 +228,7 @@ def get_receive_links(messages: Union[ServiceBusReceivedMessage, Iterable[Servic
 
     trace_messages = (
         messages
-        if isinstance(messages, Iterable)  # pylint:disable=isinstance-second-argument-not-valid-type
+        if isinstance(messages, Iterable)
         else (messages,)
     )
 

sdk/servicebus/azure-servicebus/azure/servicebus/_common/utils.py

@@ -42,7 +42,6 @@
 
 if TYPE_CHECKING:
     try:
-        # pylint:disable=unused-import
         from uamqp import types as uamqp_types
         from uamqp.authentication import JWTTokenAuth as uamqp_JWTTokenAuth
     except ImportError:
@@ -95,6 +94,9 @@ def create_properties(
     :param str user_agent: If specified,
     this will be added in front of the built-in user agent string.
 
+    :keyword amqp_transport: The AMQP transport type.
+    :paramtype amqp_transport: ~azure.servicebus._transport._base.AmqpTransport
+
     :return: The properties to add to the connection.
     :rtype: dict
     """
@@ -131,7 +133,6 @@ def get_renewable_start_time(renewable):
 
 
 def get_renewable_lock_duration(renewable: Union["ServiceBusReceivedMessage", "BaseSession"]) -> datetime.timedelta:
-    # pylint: disable=protected-access
     try:
         return max(renewable.locked_until_utc - utc_now(), datetime.timedelta(seconds=0))
     except AttributeError:

sdk/servicebus/azure-servicebus/azure/servicebus/_pyamqp/_connection.py

@@ -90,7 +90,7 @@ class Connection:  # pylint:disable=too-many-instance-attributes
      and 1 for transport type AmqpOverWebsocket.
     """
 
-    def __init__(  # pylint:disable=too-many-locals,too-many-statements
+    def __init__(  # pylint:disable=too-many-locals
         self,
         endpoint: str,
         *,

sdk/servicebus/azure-servicebus/azure/servicebus/_pyamqp/_decode.py

@@ -3,8 +3,6 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-# pylint: disable=redefined-builtin, import-error
-
 import struct
 import uuid
 import logging

sdk/servicebus/azure-servicebus/azure/servicebus/_pyamqp/_encode.py

@@ -489,8 +489,9 @@ def encode_map(
     count = len(cast(Sized, value)) * 2
     encoded_size = 0
     encoded_values = bytearray()
+    items: Iterable[Any]
     if isinstance(value, dict):
-        items: Iterable[Any] = value.items()
+        items = value.items()
     elif isinstance(value, Iterable):
         items = value
 

sdk/servicebus/azure-servicebus/azure/servicebus/_pyamqp/_message_backcompat.py

@@ -3,8 +3,6 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-
-# pylint: disable=too-many-lines
 from enum import Enum
 from typing import (
     Callable,