Reformat and add a script to the makefile

Author: fabienheureux

Makefile

@@ -50,9 +50,12 @@ clean-test: ## remove test and coverage artifacts
 lint: ## check style with flake8
 	flake8 graphql_ws tests
 
+format:
+	black ./graphql_ws ./tests
+
 test: ## run tests quickly with the default Python
 	py.test
-	
+
 
 test-all: ## run tests on every Python version with tox
 	tox

graphql_ws/base.py

@@ -41,7 +41,7 @@ def remove_operation(self, op_id):
 
     def unsubscribe(self, op_id):
         async_iterator = self.remove_operation(op_id)
-        if hasattr(async_iterator, 'dispose'):
+        if hasattr(async_iterator, "dispose"):
             async_iterator.dispose()
         return async_iterator
 

graphql_ws/django/consumers.py

@@ -7,7 +7,6 @@
 
 
 class GraphQLSubscriptionConsumer(AsyncJsonWebsocketConsumer):
-
     async def connect(self):
         self.connection_context = None
         if WS_PROTOCOL in self.scope["subprotocols"]:

graphql_ws/django/routing.py

@@ -15,8 +15,10 @@
 channels_version_tuple = get_version_tuple(channels_version)
 
 
-if channels_version_tuple > (3, 0 , 0):
-    websocket_urlpatterns = [path("subscriptions", GraphQLSubscriptionConsumer.as_asgi())]
+if channels_version_tuple > (3, 0, 0):
+    websocket_urlpatterns = [
+        path("subscriptions", GraphQLSubscriptionConsumer.as_asgi())
+    ]
 else:
     websocket_urlpatterns = [path("subscriptions", GraphQLSubscriptionConsumer)]
 

tests/test_base.py

@@ -91,7 +91,7 @@ def test_observer_data():
         send_error=send_error,
         send_message=send_message,
     )
-    observer.on_next('data')
+    observer.on_next("data")
     assert send_result.called
     assert not send_error.called
 
@@ -107,6 +107,6 @@ def test_observer_exception():
         send_error=send_error,
         send_message=send_message,
     )
-    observer.on_next(TypeError('some bad message'))
+    observer.on_next(TypeError("some bad message"))
     assert send_error.called
     assert not send_result.called

tests/test_base_async.py

@@ -83,18 +83,18 @@ async def test_resolver_with_promise(server):
         connection_context=None, op_id=1, execution_result=result
     )
     assert server.send_message.called
-    assert result.data == {'test': [1, 2]}
+    assert result.data == {"test": [1, 2]}
 
 
 async def test_resolver_with_nested_promise(server):
     server.send_message = AsyncMock()
     result = mock.Mock()
     inner = promise.Promise(lambda resolve, reject: resolve(2))
-    outer = promise.Promise(lambda resolve, reject: resolve({'in': inner}))
+    outer = promise.Promise(lambda resolve, reject: resolve({"in": inner}))
     result.data = {"test": [1, outer]}
     result.errors = None
     await server.send_execution_result(
         connection_context=None, op_id=1, execution_result=result
     )
     assert server.send_message.called
-    assert result.data == {'test': [1, {'in': 2}]}
+    assert result.data == {"test": [1, {"in": 2}]}