Skip to content

Commit

Permalink
fix SIM113 ruff issue
Browse files Browse the repository at this point in the history
  • Loading branch information
msaipraneeth committed Jun 7, 2024
1 parent 7bedd7a commit 2c69f2d
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions cmem_plugin_kafka/kafka_handlers.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,10 +73,8 @@ def send_messages(self, data: Response) -> None:
if not self._kafka_producer:
return
messages = self._split_data(data)
count = 0
for message in messages:
for count, message in enumerate(messages):
self._kafka_producer.process(message)
count += 1
if count % 10 == 0:
self._kafka_producer.poll(0)
self.update_report()
Expand Down Expand Up @@ -191,12 +189,10 @@ def _aggregate_data(self) -> Generator:
raise ValueError("Kafka consumer is None")
try:
yield b"["
count = 0
for message in self._kafka_consumer.poll():
for count, message in enumerate(self._kafka_consumer.poll()):
if count > 0:
yield b","
yield get_message_with_json_wrapper(message).encode()
count += 1
yield b"]"
except json.decoder.JSONDecodeError as ex:
raise ValueError("Kafka Message is not in expected format ") from ex
Expand Down

0 comments on commit 2c69f2d

Please sign in to comment.