add typing to aiokafka/cluster.py

Makefile

@@ -6,15 +6,19 @@ KAFKA_VERSION?=2.8.1
 DOCKER_IMAGE=aiolibs/kafka:$(SCALA_VERSION)_$(KAFKA_VERSION)
 DIFF_BRANCH=origin/master
 FORMATTED_AREAS=\
+	aiokafka/cluster.py \
 	aiokafka/codec.py \
+	aiokafka/conn.py \
 	aiokafka/coordinator/ \
 	aiokafka/errors.py \
 	aiokafka/helpers.py \
 	aiokafka/structs.py \
 	aiokafka/util.py \
 	aiokafka/protocol/ \
 	aiokafka/record/ \
+	tests/test_cluster.py \
 	tests/test_codec.py \
+	tests/test_conn.py \
 	tests/test_helpers.py \
 	tests/test_protocol.py \
 	tests/test_protocol_object_conversion.py \

aiokafka/abc.py

@@ -1,4 +1,5 @@
 import abc
+from typing import Dict
 
 
 class ConsumerRebalanceListener(abc.ABC):
@@ -103,7 +104,7 @@ class AbstractTokenProvider(abc.ABC):
     """
 
     @abc.abstractmethod
-    async def token(self):
+    async def token(self) -> str:
         """
         An async callback returning a :class:`str` ID/Access Token to be sent to
         the Kafka client. In case where a synchronous callback is needed,
@@ -122,7 +123,7 @@ def _token(self):
                     # The actual synchronous token callback.
         """
 
-    def extensions(self):
+    def extensions(self) -> Dict[str, str]:
         """
         This is an OPTIONAL method that may be implemented.
 

aiokafka/client.py

@@ -1,5 +1,6 @@
 import asyncio
 import contextlib
+import enum
 import logging
 import random
 import time
@@ -45,7 +46,7 @@ class ConnectionGroup:
     COORDINATION = 1
 
 
-class CoordinationType:
+class CoordinationType(enum.IntEnum):
     GROUP = 0
     TRANSACTION = 1