Run ruff with updated config

Author: JordonPhillips

packages/aws-event-stream/src/aws_event_stream/_private/__init__.py

@@ -1,7 +1,6 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 from smithy_core.schemas import Schema
-
 from smithy_core.traits import EventPayloadTrait
 
 INITIAL_REQUEST_EVENT_TYPE = "initial-request"

packages/aws-event-stream/src/aws_event_stream/_private/deserializers.py

@@ -11,6 +11,7 @@
 )
 from smithy_core.schemas import Schema
 from smithy_core.shapes import ShapeType
+from smithy_core.traits import EventHeaderTrait
 from smithy_core.utils import expect_type
 
 from ..events import HEADERS_DICT, Event
@@ -20,7 +21,6 @@
     INITIAL_RESPONSE_EVENT_TYPE,
     get_payload_member,
 )
-from smithy_core.traits import EventHeaderTrait
 
 logger = logging.getLogger(__name__)
 

packages/aws-event-stream/src/aws_event_stream/_private/serializers.py

@@ -15,15 +15,15 @@
     SpecificShapeSerializer,
 )
 from smithy_core.shapes import ShapeType
+from smithy_core.traits import ErrorTrait, EventHeaderTrait, MediaTypeTrait
 
-from ..events import EventHeaderEncoder, EventMessage, HEADER_VALUE, Short, Byte, Long
+from ..events import HEADER_VALUE, Byte, EventHeaderEncoder, EventMessage, Long, Short
 from ..exceptions import InvalidHeaderValue
 from . import (
     INITIAL_REQUEST_EVENT_TYPE,
     INITIAL_RESPONSE_EVENT_TYPE,
     get_payload_member,
 )
-from smithy_core.traits import ErrorTrait, EventHeaderTrait, MediaTypeTrait
 
 logger = logging.getLogger(__name__)
 

packages/aws-event-stream/src/aws_event_stream/aio/__init__.py

@@ -3,29 +3,27 @@
 import asyncio
 import logging
 from collections.abc import Callable
+from typing import Protocol
 
-from smithy_core.aio.interfaces import AsyncByteStream
+from smithy_core.aio.interfaces import AsyncByteStream, AsyncWriter
+from smithy_core.aio.interfaces.eventstream import EventPublisher, EventReceiver
 from smithy_core.codecs import Codec
 from smithy_core.deserializers import DeserializeableShape, ShapeDeserializer
 from smithy_core.exceptions import ExpectationNotMetException
 from smithy_core.serializers import SerializeableShape
-from smithy_core.aio.interfaces.eventstream import EventPublisher, EventReceiver
-from smithy_core.aio.interfaces import AsyncWriter
 
-from .._private.serializers import EventSerializer as _EventSerializer
 from .._private.deserializers import EventDeserializer as _EventDeserializer
+from .._private.serializers import EventSerializer as _EventSerializer
 from ..events import Event, EventHeaderEncoder, EventMessage
 from ..exceptions import EventError
 
-from typing import Protocol
-
 logger = logging.getLogger(__name__)
 
 
 class EventSigner(Protocol):
     """A signer to manage credentials and EventMessages for an Event Stream lifecyle."""
 
-    def sign_event(
+    async def sign_event(
         self,
         *,
         event_message: EventMessage,
@@ -50,7 +48,7 @@ def __init__(
 
     async def send(self, event: E) -> None:
         if self._closed:
-            raise IOError("Attempted to write to closed stream.")
+            raise OSError("Attempted to write to closed stream.")
         logger.debug("Preparing to publish event: %s", event)
         event.serialize(self._serializer)
         result = self._serializer.get_result()
@@ -60,19 +58,18 @@ async def send(self, event: E) -> None:
             )
         if self._signer is not None:
             encoder = self._serializer.event_header_encoder_cls
-            result = await self._signer.sign_event(  # type: ignore
+            result = await self._signer.sign_event(
                 event_message=result,
                 event_encoder_cls=encoder,
             )
 
-        assert isinstance(result, EventMessage)
         encoded_result = result.encode()
         try:
             logger.debug("Publishing serialized event: %s", result)
             await self._writer.write(encoded_result)
         except Exception as e:
             await self.close()
-            raise IOError("Failed to write to stream.") from e
+            raise OSError("Failed to write to stream.") from e
 
     async def close(self) -> None:
         if self._closed:
@@ -111,7 +108,7 @@ async def receive(self) -> E | None:
         except Exception as e:
             await self.close()
             if not isinstance(e, EventError):
-                raise IOError("Failed to read from stream.") from e
+                raise OSError("Failed to read from stream.") from e
             raise
 
         if event is None:

packages/aws-event-stream/tests/unit/_private/__init__.py

@@ -4,6 +4,7 @@
 from dataclasses import dataclass
 from typing import Any, ClassVar, Literal, Self
 
+from aws_event_stream.events import Byte, EventMessage, Long, Short
 from smithy_core.deserializers import ShapeDeserializer
 from smithy_core.exceptions import SmithyException
 from smithy_core.prelude import (
@@ -20,15 +21,13 @@
 from smithy_core.serializers import ShapeSerializer
 from smithy_core.shapes import ShapeID, ShapeType
 from smithy_core.traits import (
+    ErrorTrait,
     EventHeaderTrait,
     EventPayloadTrait,
-    ErrorTrait,
     RequiredTrait,
     StreamingTrait,
 )
 
-from aws_event_stream.events import Byte, EventMessage, Long, Short
-
 EVENT_HEADER_TRAIT = EventHeaderTrait()
 EVENT_PAYLOAD_TRAIT = EventPayloadTrait()
 ERROR_TRAIT = ErrorTrait("client")

packages/aws-event-stream/tests/unit/_private/test_deserializers.py

@@ -4,14 +4,13 @@
 from typing import Any
 
 import pytest
-from smithy_core.aio.types import AsyncBytesReader
-from smithy_core.deserializers import DeserializeableShape
-from smithy_json import JSONCodec
-
-from aws_event_stream.aio import AWSEventReceiver
 from aws_event_stream._private.deserializers import EventDeserializer
+from aws_event_stream.aio import AWSEventReceiver
 from aws_event_stream.events import Event, EventMessage
 from aws_event_stream.exceptions import UnmodeledEventError
+from smithy_core.aio.types import AsyncBytesReader
+from smithy_core.deserializers import DeserializeableShape
+from smithy_json import JSONCodec
 
 from . import (
     EVENT_STREAM_SERDE_CASES,

packages/aws-event-stream/tests/unit/_private/test_serializers.py

@@ -3,13 +3,12 @@
 from typing import Any
 
 import pytest
-from smithy_core.serializers import SerializeableShape
-from smithy_core.aio.types import AsyncBytesProvider
-from smithy_json import JSONCodec
-
-from aws_event_stream.aio import AWSEventPublisher
 from aws_event_stream._private.serializers import EventSerializer
+from aws_event_stream.aio import AWSEventPublisher
 from aws_event_stream.events import EventMessage
+from smithy_core.aio.types import AsyncBytesProvider
+from smithy_core.serializers import SerializeableShape
+from smithy_json import JSONCodec
 
 from . import EVENT_STREAM_SERDE_CASES, INITIAL_REQUEST_CASE, INITIAL_RESPONSE_CASE
 

packages/aws-event-stream/tests/unit/test_events.py

@@ -6,8 +6,6 @@
 from io import BytesIO
 
 import pytest
-from smithy_core.aio.types import AsyncBytesReader
-
 from aws_event_stream.events import (
     MAX_HEADER_VALUE_BYTE_LENGTH,
     MAX_HEADERS_LENGTH,
@@ -31,6 +29,7 @@
     InvalidIntegerValue,
     InvalidPayloadLength,
 )
+from smithy_core.aio.types import AsyncBytesReader
 
 EMPTY_MESSAGE = (
     (

packages/aws-sdk-signers/src/aws_sdk_signers/__init__.py

@@ -4,14 +4,15 @@
 such as AioHTTP, Curl, Postman, Requests, urllib3, etc."""
 
 from __future__ import annotations
+
 import importlib.metadata
 
 from ._http import URI, AWSRequest, Field, Fields
 from ._identity import AWSCredentialIdentity
 from ._io import AsyncBytesReader
 from .signers import (
-    AsyncSigV4Signer,
     AsyncEventSigner,
+    AsyncSigV4Signer,
     SigV4Signer,
     SigV4SigningProperties,
 )
@@ -20,14 +21,14 @@
 __version__ = importlib.metadata.version("aws-sdk-signers")
 
 __all__ = (
-    "AsyncBytesReader",
-    "AsyncSigV4Signer",
-    "AsyncEventSigner",
+    "URI",
     "AWSCredentialIdentity",
     "AWSRequest",
+    "AsyncBytesReader",
+    "AsyncEventSigner",
+    "AsyncSigV4Signer",
     "Field",
     "Fields",
     "SigV4Signer",
     "SigV4SigningProperties",
-    "URI",
 )

packages/aws-sdk-signers/src/aws_sdk_signers/_http.py

@@ -39,7 +39,7 @@ def __init__(
         kind: interfaces_http.FieldPosition = interfaces_http.FieldPosition.HEADER,
     ):
         self.name = name
-        self.values: list[str] = [val for val in values] if values is not None else []
+        self.values: list[str] = list(values) if values is not None else []
         self.kind = kind
 
     def add(self, value: str) -> None:
@@ -114,7 +114,7 @@ def __init__(
         :param encoding: The string encoding to be used when converting the ``Field``
         name and value from ``str`` to ``bytes`` for transmission.
         """
-        init_fields = [fld for fld in initial] if initial is not None else []
+        init_fields = list(initial) if initial is not None else []
         init_field_names = [self._normalize_field_name(fld.name) for fld in init_fields]
         fname_counter = Counter(init_field_names)
         repeated_names_exist = (