Redis PubSub wrapper for Python

Recently I've found that there's no reasonable simple and useful Redis pub sub examples around. So, here is my dead simple wrapper how to implement it without any unnecessary overhead.

	class PubSub(object):
	"""
	Very simple Pub/Sub pattern wrapper
	using simplified Redis Pub/Sub functionality.
	Usage (publisher)::
	import redis
	r = redis.Redis()
	q = PubSub(r, "channel")
	q.publish("test data")
	Usage (listener)::
	import redis
	r = redis.Redis()
	q = PubSub(r, "channel")
	def handler(data):
	print "Data received: %r" % data
	q.subscribe(handler)
	"""
	def __init__(self, redis, channel="default"):
	self.redis = redis
	self.channel = channel
	def publish(self, data):
	self.redis.publish(self.channel, json.dumps(data))
	def subscribe(self, handler):
	redis = self.redis.pubsub()
	redis.subscribe(self.channel)
	for data_raw in redis.listen():
	if data_raw['type'] != "message":
	continue
	data = json.loads(data_raw["data"])
	handler(data)

view raw pubsub.py hosted with ❤ by GitHub