first cut of redis pubsub [redis_pubsub_first_cut]

hjwp · hjwp · commit cbb8db796ed0 · 2019-07-26T15:41:32.000+01:00
diff --git a/src/allocation/redis_pubsub.py b/src/allocation/redis_pubsub.py
@@ -0,0 +1,37 @@
+import json
+import logging
+from dataclasses import asdict
+import redis
+
+from allocation import config, orm, services, unit_of_work
+
+logger = logging.getLogger(__name__)
+
+r = redis.Redis(**config.get_redis_host_and_port())
+
+
+def main():
+    orm.start_mappers()
+    pubsub = r.pubsub(ignore_subscribe_messages=True)
+    pubsub.subscribe('change_batch_quantity')
+
+    for m in pubsub.listen():
+        handle_change_batch_quantity(m)
+
+
+def handle_change_batch_quantity(m):
+    logging.debug('handling %s', m)
+    data = json.loads(m['data'])
+    services.change_batch_quantity(
+        ref=data['batchref'], qty=data['qty'],
+        uow=unit_of_work.SqlAlchemyUnitOfWork(),
+    )
+
+
+def publish(channel, event):
+    logging.debug('publishing: channel=%s, event=%s', channel, event)
+    r.publish(channel, json.dumps(asdict(event)))
+
+
+if __name__ == '__main__':
+    main()
