Test whether redis works

1 files changed, 15 insertions, 0 deletions

diff --git a/Code/test_redis_subscribe.py b/Code/test_redis_subscribe.py
new file mode 100644
index 0000000..f38249d
--- /dev/null
+++ b/Code/test_redis_subscribe.py
@@ -0,0 +1,15 @@
+from configure import REDIS_CHANNEL
+import redis # install redis on Ubuntu using this command: sudo apt install python3-redis
+import json
+import os
+
+redis_password = os.getenv('REDIS_PASSWORD', '123456')
+redis_host = os.getenv('REDIS_HOST', '127.0.0.1')
+r = redis.Redis(host=redis_host, port=6379, password=redis_password, db=0)
+pubsub = r.pubsub(ignore_subscribe_messages=True)
+pubsub.subscribe(REDIS_CHANNEL)
+
+for message in pubsub.listen():
+    quant_data = json.loads(message['data'])
+    filename = quant_data['filename']
+    print('[test_redis_subscribe.py] %s - %s.' % (filename, quant_data))