openstack/deb-python-cassandra-driver
Code Issues Proposed changes

Check if max_attempts is None

Browse Source
This commit is contained in:
Tom Lin
2015-05-25 16:11:35 +08:00
parent ff1d75a506
commit 7aa8f30375
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
cassandra/policies.py
View File

@@ -509,14 +509,16 @@ class ConstantReconnectionPolicy(ReconnectionPolicy):
""" """
if delay < 0: if delay < 0:
raise ValueError("delay must not be negative") raise ValueError("delay must not be negative")
if max_attempts < 0: if max_attempts is not None and max_attempts < 0:
raise ValueError("max_attempts must not be negative") raise ValueError("max_attempts must not be negative")
self.delay = delay self.delay = delay
self.max_attempts = max_attempts self.max_attempts = max_attempts
def new_schedule(self): def new_schedule(self):
return repeat(self.delay, self.max_attempts) if self.max_attempts:
return repeat(self.delay, self.max_attempts)
return repeat(self.delay)
class ExponentialReconnectionPolicy(ReconnectionPolicy): class ExponentialReconnectionPolicy(ReconnectionPolicy):