diff options
author | andy <andy@whiskeymedia.com> | 2011-07-11 01:20:48 -0700 |
---|---|---|
committer | andy <andy@whiskeymedia.com> | 2011-07-11 01:20:48 -0700 |
commit | 37fb01edf06673cd56421497128daee3fafbd2d9 (patch) | |
tree | 44cb0f77cd61fdefc3f9323e2d7dd0e6661059cc /redis/client.py | |
parent | 53c928d44acd3d1fbcb3896cadad0bde4671987a (diff) | |
parent | be1e7498ff195681f1cf760b764d657161715968 (diff) | |
download | redis-py-37fb01edf06673cd56421497128daee3fafbd2d9.tar.gz |
Merge branch 'master' into watch
Conflicts:
CHANGES
Diffstat (limited to 'redis/client.py')
-rw-r--r-- | redis/client.py | 42 |
1 files changed, 25 insertions, 17 deletions
diff --git a/redis/client.py b/redis/client.py index 8b8e7a8..a055f22 100644 --- a/redis/client.py +++ b/redis/client.py @@ -81,8 +81,9 @@ def zset_score_pairs(response, **options): """ if not response or not options['withscores']: return response + score_cast_func = options.get('score_cast_func', float) it = iter(response) - return zip(it, imap(float, it)) + return zip(it, imap(score_cast_func, it)) def int_or_none(response): if response is None: @@ -821,27 +822,31 @@ class Redis(object): """ return self._zaggregate('ZINTERSTORE', dest, keys, aggregate) - def zrange(self, name, start, end, desc=False, withscores=False): + def zrange(self, name, start, end, desc=False, withscores=False, + score_cast_func=float): """ Return a range of values from sorted set ``name`` between ``start`` and ``end`` sorted in ascending order. ``start`` and ``end`` can be negative, indicating the end of the range. - ``desc`` indicates to sort in descending order. + ``desc`` a boolean indicating whether to sort the results descendingly ``withscores`` indicates to return the scores along with the values. The return type is a list of (value, score) pairs + + ``score_cast_func`` a callable used to cast the score return value """ if desc: return self.zrevrange(name, start, end, withscores) pieces = ['ZRANGE', name, start, end] if withscores: pieces.append('withscores') - return self.execute_command(*pieces, **{'withscores': withscores}) + options = {'withscores': withscores, 'score_cast_func': score_cast_func} + return self.execute_command(*pieces, **options) def zrangebyscore(self, name, min, max, - start=None, num=None, withscores=False): + start=None, num=None, withscores=False, score_cast_func=float): """ Return a range of values from the sorted set ``name`` with scores between ``min`` and ``max``. @@ -851,6 +856,8 @@ class Redis(object): ``withscores`` indicates to return the scores along with the values. The return type is a list of (value, score) pairs + + `score_cast_func`` a callable used to cast the score return value """ if (start is not None and num is None) or \ (num is not None and start is None): @@ -860,7 +867,8 @@ class Redis(object): pieces.extend(['LIMIT', start, num]) if withscores: pieces.append('withscores') - return self.execute_command(*pieces, **{'withscores': withscores}) + options = {'withscores': withscores, 'score_cast_func': score_cast_func} + return self.execute_command(*pieces, **options) def zrank(self, name, value): """ @@ -889,7 +897,8 @@ class Redis(object): """ return self.execute_command('ZREMRANGEBYSCORE', name, min, max) - def zrevrange(self, name, start, num, withscores=False): + def zrevrange(self, name, start, num, withscores=False, + score_cast_func=float): """ Return a range of values from sorted set ``name`` between ``start`` and ``num`` sorted in descending order. @@ -898,14 +907,17 @@ class Redis(object): ``withscores`` indicates to return the scores along with the values The return type is a list of (value, score) pairs + + ``score_cast_func`` a callable used to cast the score return value """ pieces = ['ZREVRANGE', name, start, num] if withscores: pieces.append('withscores') - return self.execute_command(*pieces, **{'withscores': withscores}) + options = {'withscores': withscores, 'score_cast_func': score_cast_func} + return self.execute_command(*pieces, **options) def zrevrangebyscore(self, name, max, min, - start=None, num=None, withscores=False): + start=None, num=None, withscores=False, score_cast_func=float): """ Return a range of values from the sorted set ``name`` with scores between ``min`` and ``max`` in descending order. @@ -915,6 +927,8 @@ class Redis(object): ``withscores`` indicates to return the scores along with the values. The return type is a list of (value, score) pairs + + ``score_cast_func`` a callable used to cast the score return value """ if (start is not None and num is None) or \ (num is not None and start is None): @@ -924,7 +938,8 @@ class Redis(object): pieces.extend(['LIMIT', start, num]) if withscores: pieces.append('withscores') - return self.execute_command(*pieces, **{'withscores': withscores}) + options = {'withscores': withscores, 'score_cast_func': score_cast_func} + return self.execute_command(*pieces, **options) def zrevrank(self, name, value): """ @@ -1128,13 +1143,6 @@ class PubSub(object): pass return self.execute_command('UNSUBSCRIBE', *channels) - def publish(self, channel, message): - """ - Publish ``message`` on ``channel``. - Returns the number of subscribers the message was delivered to. - """ - return self.execute_command('PUBLISH', channel, message) - def listen(self): "Listen for messages on channels this client has been subscribed to" while self.subscription_count: |