diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-02-06 11:57:49 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-02-09 10:34:19 -0800 |
| commit | fb03c69544221a25e925b8c6af45e44ac6944b2a (patch) | |
| tree | 3a4f99544c58863cd7ded04c88d4344179e75745 /kafka/partitioner/hashed.py | |
| parent | bb731eb89cc58c2e6799ff2223c47011594986f1 (diff) | |
| download | kafka-python-fb03c69544221a25e925b8c6af45e44ac6944b2a.tar.gz | |
make partitions arg optional in *Partitioner.partition, defaulting to self.partitions from init
Diffstat (limited to 'kafka/partitioner/hashed.py')
| -rw-r--r-- | kafka/partitioner/hashed.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/kafka/partitioner/hashed.py b/kafka/partitioner/hashed.py index 587a3de..fb5e598 100644 --- a/kafka/partitioner/hashed.py +++ b/kafka/partitioner/hashed.py @@ -5,7 +5,9 @@ class HashedPartitioner(Partitioner): Implements a partitioner which selects the target partition based on the hash of the key """ - def partition(self, key, partitions): + def partition(self, key, partitions=None): + if not partitions: + partitions = self.partitions size = len(partitions) idx = hash(key) % size |