diff options
| author | Bruno Renié <brutasse@gmail.com> | 2014-08-29 10:14:54 +0200 |
|---|---|---|
| committer | Mark Roberts <wizzat@fb.com> | 2014-09-03 09:55:45 -0700 |
| commit | d6f08d31e233c5d297cbaa04750b145b8eb908f5 (patch) | |
| tree | 229531d90bffa7d96926374fb55b2dbc70e96deb | |
| parent | aad256372bafd637f0a7d66a4271da114f8f6cb0 (diff) | |
| download | kafka-python-d6f08d31e233c5d297cbaa04750b145b8eb908f5.tar.gz | |
Use built-in next()
| -rw-r--r-- | kafka/partitioner.py | 2 | ||||
| -rw-r--r-- | kafka/producer.py | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/kafka/partitioner.py b/kafka/partitioner.py index 5287cef..695dd6f 100644 --- a/kafka/partitioner.py +++ b/kafka/partitioner.py @@ -43,7 +43,7 @@ class RoundRobinPartitioner(Partitioner): if self.partitions != partitions: self._set_partitions(partitions) - return self.iterpart.next() + return next(self.iterpart) class HashedPartitioner(Partitioner): diff --git a/kafka/producer.py b/kafka/producer.py index e1f4a22..fb83da6 100644 --- a/kafka/producer.py +++ b/kafka/producer.py @@ -258,9 +258,9 @@ class SimpleProducer(Producer): if self.random_start: num_partitions = len(self.client.topic_partitions[topic]) for _ in xrange(random.randint(0, num_partitions-1)): - self.partition_cycles[topic].next() + next(self.partition_cycles[topic]) - return self.partition_cycles[topic].next() + return next(self.partition_cycles[topic]) def send_messages(self, topic, *msg): partition = self._next_partition(topic) |