Made classes enum.IntEnums where appropriate. (#1053)

Author: jackgene

aiokafka/client.py

@@ -3,6 +3,7 @@
 import logging
 import random
 import time
+from enum import IntEnum
 
 import aiokafka.errors as Errors
 from aiokafka import __version__
@@ -40,12 +41,12 @@
 log = logging.getLogger("aiokafka")
 
 
-class ConnectionGroup:
+class ConnectionGroup(IntEnum):
     DEFAULT = 0
     COORDINATION = 1
 
 
-class CoordinationType:
+class CoordinationType(IntEnum):
     GROUP = 0
     TRANSACTION = 1
 

aiokafka/conn.py

@@ -15,6 +15,7 @@
 import uuid
 import warnings
 import weakref
+from enum import IntEnum
 
 import async_timeout
 
@@ -45,7 +46,7 @@
 SASL_QOP_AUTH = 1
 
 
-class CloseReason:
+class CloseReason(IntEnum):
     CONNECTION_BROKEN = 0
     CONNECTION_TIMEOUT = 1
     OUT_OF_SYNC = 2

aiokafka/metrics/stats/percentiles.py

@@ -1,11 +1,13 @@
+from enum import IntEnum
+
 from aiokafka.metrics.compound_stat import AbstractCompoundStat, NamedMeasurable
 from aiokafka.metrics.measurable import AnonMeasurable
 
 from .histogram import Histogram
 from .sampled_stat import AbstractSampledStat
 
 
-class BucketSizing:
+class BucketSizing(IntEnum):
     CONSTANT = 0
     LINEAR = 1
 

aiokafka/producer/transaction_manager.py

@@ -1,5 +1,5 @@
 from collections import defaultdict, deque, namedtuple
-from enum import Enum
+from enum import Enum, IntEnum
 
 from aiokafka.structs import TopicPartition
 from aiokafka.util import create_future
@@ -16,7 +16,7 @@ class SubscriptionType(Enum):
     USER_ASSIGNED = 4
 
 
-class TransactionResult:
+class TransactionResult(IntEnum):
     ABORT = 0
     COMMIT = 1