From f35a7dacbebae1fad36afaab5a66ff1379f604f4 Mon Sep 17 00:00:00 2001 From: Lan Hui Date: Thu, 31 Jul 2025 20:35:23 +0800 Subject: Test redis tomorrow morning --- Code/download_and_map.py | 4 +++- Code/get_mapped_data.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) (limited to 'Code') diff --git a/Code/download_and_map.py b/Code/download_and_map.py index bceeb82..5ad19cf 100644 --- a/Code/download_and_map.py +++ b/Code/download_and_map.py @@ -388,7 +388,9 @@ def reverse_lines(fname, fname2): def publish(mapped_data_directory): - r = redis.Redis(host='118.25.96.118', port=6379, db=0) + redis_host = os.getenv('REDIS_HOST', '127.0.0.1') + redis_password = os.getenv('REDIS_PASSWORD', '123456') + r = redis.Redis(host=redis_host, port=6379, password=redis_password, db=0) for fname in glob.glob('%s/*_quant.txt' % (mapped_data_directory.rstrip('/'))): file_basename = os.path.basename(fname) with open(fname, 'r') as f: diff --git a/Code/get_mapped_data.py b/Code/get_mapped_data.py index d1a4146..ea125a0 100644 --- a/Code/get_mapped_data.py +++ b/Code/get_mapped_data.py @@ -5,7 +5,9 @@ import redis # install redis on Ubuntu using this command: sudo apt install pyth import json import os -r = redis.Redis(host='118.25.96.118', port=6379, db=0) +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) -- cgit v1.2.1