flake8 cleanup

Author: jell-o-fishi

examples/bugs/issue_290/client.py

@@ -1,24 +1,17 @@
 import asyncio
 import logging
-import sys
 from asyncio import Event
 from typing import AsyncGenerator, Tuple
 
-import aiohttp
-from rsocket.transports.asyncwebsockets_transport import websocket_client
-
-from rsocket.transports.aiohttp_websocket import TransportAioHttpClient
-
 from reactivestreams.publisher import Publisher
 from reactivestreams.subscriber import Subscriber
 from reactivestreams.subscription import Subscription
 from rsocket.extensions.helpers import route, composite, authenticate_simple
 from rsocket.extensions.mimetypes import WellKnownMimeTypes
-from rsocket.helpers import single_transport_provider
 from rsocket.payload import Payload
 from rsocket.rsocket_client import RSocketClient
 from rsocket.streams.stream_from_async_generator import StreamFromAsyncGenerator
-from rsocket.transports.tcp import TransportTCP
+from rsocket.transports.asyncwebsockets_transport import websocket_client
 
 
 def sample_publisher(wait_for_requester_complete: Event,
@@ -64,8 +57,8 @@ def on_complete(self):
         logging.info('Completed from server on channel')
         self._wait_for_responder_complete.set()
 
-async def request_channel(client: RSocketClient):
 
+async def request_channel(client: RSocketClient):
     channel_completion_event = Event()
     requester_completion_event = Event()
     payload = Payload(b'The first item in the stream', composite(
@@ -84,9 +77,9 @@ async def request_channel(client: RSocketClient):
 
 
 async def application(serve_port: int):
-        async with websocket_client('http://localhost:%s/rsocket' % serve_port,
-                                    metadata_encoding=WellKnownMimeTypes.MESSAGE_RSOCKET_COMPOSITE_METADATA,) as client:
-            await request_channel(client)
+    async with websocket_client('http://localhost:%s/rsocket' % serve_port,
+                                metadata_encoding=WellKnownMimeTypes.MESSAGE_RSOCKET_COMPOSITE_METADATA, ) as client:
+        await request_channel(client)
 
 
 async def command():
@@ -95,4 +88,4 @@ async def command():
 
 
 if __name__ == '__main__':
-    asyncio.run(command())
+    asyncio.run(command())

examples/bugs/issue_290/server.py

@@ -22,7 +22,7 @@ def publish(self, data):
         print(f"ChannelPublisher - publish: {data}")
         print(f"ChannelPublisher - publish subscriber: {self.subscriber}")
 
-        if (self.subscriber is not None):
+        if self.subscriber is not None:
             self.subscriber.on_next(Payload(data.encode('utf-8')))
 
     def subscribe(self, subscriber):
@@ -36,6 +36,7 @@ def subscribe(self, subscriber):
 class ChannelSubscriber(DefaultSubscriber):
 
     def __init__(self, publisher: ChannelPublisher):
+        super().__init__()
         self.subscription = None
         self.publisher = publisher
 
@@ -106,4 +107,4 @@ async def start_server():
 
 
 if __name__ == '__main__':
-    asyncio.run(start_server())
+    asyncio.run(start_server())

examples/django_channels/client_example.py

@@ -29,21 +29,21 @@ async def connect_to_django_channels(url: str, ssl_context: Optional[ssl.SSLCont
     async with websockets.connect(url, ssl=ssl_context) as websocket:
         # Create a transport using the websocket connection
         transport = WebsocketsTransport()
-        
+
         # Start the transport handler
         handler_task = asyncio.create_task(transport.handler(websocket))
-        
+
         try:
             # Create an RSocket client using the transport
             async with RSocketClient(single_transport_provider(transport)) as client:
                 # Send a request-response
                 payload = Payload(b'Hello from RSocket client')
                 response = await client.request_response(payload)
-                
+
                 print(f"Received response: {response.data.decode()}")
-                
+
                 # You can add more interactions here
-                
+
         finally:
             # Clean up the handler task
             handler_task.cancel()
@@ -61,7 +61,7 @@ async def main(url: str, secure: bool):
     Connect to a Django Channels RSocket server.
     """
     logging.basicConfig(level=logging.INFO)
-    
+
     if secure and not url.startswith('wss://'):
         url = 'wss://' + url.removeprefix('ws://')
         ssl_context = ssl.create_default_context()
@@ -70,9 +70,9 @@ async def main(url: str, secure: bool):
         ssl_context.verify_mode = ssl.CERT_NONE
     else:
         ssl_context = None
-    
+
     await connect_to_django_channels(url, ssl_context)
 
 
 if __name__ == '__main__':
-    main()
+    main()

examples/django_channels/server_example.py

@@ -32,15 +32,13 @@ async def request_response(self, payload: Payload):
 # Create a consumer using the factory
 RSocketConsumer = rsocket_consumer_factory(handler_factory=Handler)
 
-
 # Django Channels routing configuration
 application = ProtocolTypeRouter({
     'websocket': URLRouter([
         path('rsocket', RSocketConsumer.as_asgi()),
     ]),
 })
 
-
 """
 To use this in a Django project:
 $ python3 -c 'import channels; import daphne; print(channels.__version__, daphne.__version__)'
@@ -76,4 +74,4 @@ async def request_response(self, payload: Payload):
 websocket_urlpatterns = [
     path('rsocket', RSocketConsumer.as_asgi()),
 ]
-"""
+"""

tests/rsocket/test_frame.py

@@ -11,7 +11,7 @@
                            MetadataPushFrame, PayloadFrame, LeaseFrame, ResumeOKFrame, KeepAliveFrame,
                            serialize_with_frame_size_header, RequestStreamFrame, RequestChannelFrame, ParseError,
                            parse_or_ignore, Frame, RequestFireAndForgetFrame, serialize_prefix_with_frame_size_header,
-                           parse_header_native, Header, is_frame_to_ignore, is_fragmentable_frame, new_frame_fragment)
+                           parse_header_native, Header, is_frame_to_ignore, is_fragmentable_frame)
 from rsocket.frame_parser import FrameParser
 from tests.rsocket.helpers import data_bits, build_frame, bits