diff options
author | Ben Bangert <ben@groovie.org> | 2017-05-31 14:48:07 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-05-31 14:48:07 -0700 |
commit | c2db1e99a35fcb81610781a5b69ac3748a5b8d90 (patch) | |
tree | c44808fc4a14dbc1ef29470ef851662e9bf50142 /kazoo/recipe | |
parent | 4f0a742599433be2d07cc989716d6af0d3e4f57a (diff) | |
parent | b7aff3fb63625b992fc36df937ff545ec89155fe (diff) | |
download | kazoo-c2db1e99a35fcb81610781a5b69ac3748a5b8d90.tar.gz |
Merge pull request #397 from paperlessreceipts/counter-pre-post-values
Counter: expose the previous and changed values.
Diffstat (limited to 'kazoo/recipe')
-rw-r--r-- | kazoo/recipe/counter.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/kazoo/recipe/counter.py b/kazoo/recipe/counter.py index 2299742..a2768a3 100644 --- a/kazoo/recipe/counter.py +++ b/kazoo/recipe/counter.py @@ -34,10 +34,14 @@ class Counter(object): counter += 2 counter -= 1 counter.value == 1 + counter.pre_value == 2 + counter.post_value == 1 counter = zk.Counter("/float", default=1.0) counter += 2.0 counter.value == 3.0 + counter.pre_value == 1.0 + counter.post_value == 3.0 """ def __init__(self, client, path, default=0): @@ -53,6 +57,8 @@ class Counter(object): self.default = default self.default_type = type(default) self._ensured_path = False + self.pre_value = None + self.post_value = None def _ensure_node(self): if not self._ensured_path: @@ -79,12 +85,15 @@ class Counter(object): return self def _inner_change(self, value): - data, version = self._value() - data = repr(data + value).encode('ascii') + self.pre_value, version = self._value() + post_value = self.pre_value + value + data = repr(post_value).encode('ascii') try: self.client.set(self.path, data, version=version) except BadVersionError: # pragma: nocover + self.post_value = None raise ForceRetryError() + self.post_value = post_value def __add__(self, value): """Add value to counter.""" |