Use redis to transfer newly mapped data [not tested]

1 files changed, 17 insertions, 0 deletions

diff --git a/Code/get_mapped_data.py b/Code/get_mapped_data.py
new file mode 100644
index 0000000..d1a4146
--- /dev/null
+++ b/Code/get_mapped_data.py
@@ -0,0 +1,17 @@
+from configure import MAPPED_RDATA_DIR
+from configure import REDIS_CHANNEL
+from log import write_log_file
+import redis # install redis on Ubuntu using this command: sudo apt install python3-redis
+import json
+import os
+
+r = redis.Redis(host='118.25.96.118', port=6379, 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 = os.path.join(MAPPED_RDATA_DIR, quant_data['filename'])
+    write_log_file('[get_mapped_data.py] Save %s.' % (filename), UPDATE_NETWORK_LOG_FILE))
+    with open(filename, 'w') as f:
+        f.write(quant_data['data'])