build: Split avro, json, msgpack into separate modules (#176)

So it is possible to import use each of the codecs independently without including all of the optional dependencies.
getsentry · Jan 23, 2023 · 0035f28 · 0035f28
1 parent ff5c88d
commit 0035f28
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/arroyo/processing/strategies/decoder/__init__.py b/arroyo/processing/strategies/decoder/__init__.py
@@ -1,19 +1,17 @@
-from arroyo.processing.strategies.decoder.avro import AvroCodec
+from arroyo.processing.strategies.decoder import avro, json, msgpack
 from arroyo.processing.strategies.decoder.base import (
     Codec,
     DecodedKafkaMessage,
     KafkaMessageDecoder,
     ValidationError,
 )
-from arroyo.processing.strategies.decoder.json import JsonCodec
-from arroyo.processing.strategies.decoder.msgpack import MsgpackCodec
 
 __all__ = [
-    "AvroCodec",
+    "avro",
+    "json",
+    "msgpack",
     "Codec",
     "DecodedKafkaMessage",
-    "JsonCodec",
     "KafkaMessageDecoder",
-    "MsgpackCodec",
     "ValidationError",
 ]
diff --git a/tests/processing/strategies/test_decoder.py b/tests/processing/strategies/test_decoder.py
@@ -11,13 +11,13 @@
 
 from arroyo.backends.kafka import KafkaPayload
 from arroyo.processing.strategies.decoder import (
-    AvroCodec,
     DecodedKafkaMessage,
-    JsonCodec,
     KafkaMessageDecoder,
-    MsgpackCodec,
     ValidationError,
 )
+from arroyo.processing.strategies.decoder.avro import AvroCodec
+from arroyo.processing.strategies.decoder.json import JsonCodec
+from arroyo.processing.strategies.decoder.msgpack import MsgpackCodec
 from arroyo.types import Message, Partition, Topic, Value