Merge pull request #122 from malefice/master

Added RedisMessage support for bytestrings
jrief · Dec 2, 2015 · 1fabfdc · 1fabfdc
2 parents 2977c64 + 5d3f0a8
commit 1fabfdc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ws4redis/redis_store.py b/ws4redis/redis_store.py
@@ -69,6 +69,9 @@ def __new__(cls, value):
                 if value != settings.WS4REDIS_HEARTBEAT:
                     value = value.encode()
                     return super(RedisMessage, cls).__new__(cls, value)
+            elif isinstance(value, bytes):
+                if value != settings.WS4REDIS_HEARTBEAT.encode():
+                    return super(RedisMessage, cls).__new__(cls, value)
             elif isinstance(value, list):
                 if len(value) >= 2 and value[0] == b'message':
                     return super(RedisMessage, cls).__new__(cls, value[2])