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