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

Merge pull request #327 from thoslin/thoslin-patch-1

Browse Source 
Check if max_attempts is None for ConstantReconnectionPolicy
This commit is contained in:
Adam Holmberg
2015-05-26 08:23:08 -05:00
parent 49c3549610 7aa8f30375
commit ccd7e460e2
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:
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")
self.delay = delay
self.max_attempts = max_attempts
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):