Skip to content

Commit 54022f6

Browse files
author
ljaqueme
committed
superseed config with ENV if available
1 parent c852241 commit 54022f6

File tree

2 files changed

+10
-5
lines changed

2 files changed

+10
-5
lines changed

bin/pdns-import.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,12 @@
5151
analyzer_redis_host = os.getenv('D4_ANALYZER_REDIS_HOST', '127.0.0.1')
5252
analyzer_redis_port = int(os.getenv('D4_ANALYZER_REDIS_PORT', 6400))
5353

54-
d4_server = config.get('global', 'd4-server')
55-
r = redis.Redis(host=analyzer_redis_host, port=analyzer_redis_port)
56-
r_d4 = redis.Redis(host=d4_server.split(':')[0], port=d4_server.split(':')[1], db=2)
54+
d4_server, d4_port = config.get('global', 'd4-server').split(':')
55+
host_redis_metadata = os.getenv('D4_REDIS_METADATA_HOST', d4_server)
56+
port_redis_metadata = int(os.getenv('D4_REDIS_METADATA_PORT', d4_port))
5757

58+
r = redis.Redis(host=analyzer_redis_host, port=analyzer_redis_port)
59+
r_d4 = redis.Redis(host=host_redis_metadata, port=port_redis_metadata, db=2)
5860

5961
with open('../etc/records-type.json') as rtypefile:
6062
rtype = json.load(rtypefile)

bin/pdns-ingestion.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,12 @@
4949
analyzer_redis_host = os.getenv('D4_ANALYZER_REDIS_HOST', '127.0.0.1')
5050
analyzer_redis_port = int(os.getenv('D4_ANALYZER_REDIS_PORT', 6400))
5151

52-
d4_server = config.get('global', 'd4-server')
52+
d4_server, d4_port = config.get('global', 'd4-server').split(':')
53+
host_redis_metadata = os.getenv('D4_REDIS_METADATA_HOST', d4_server)
54+
port_redis_metadata = int(os.getenv('D4_REDIS_METADATA_PORT', d4_port))
55+
5356
r = redis.Redis(host=analyzer_redis_host, port=analyzer_redis_port)
54-
r_d4 = redis.Redis(host=d4_server.split(':')[0], port=d4_server.split(':')[1], db=2)
57+
r_d4 = redis.Redis(host=host_redis_metadata, port=port_redis_metadata, db=2)
5558

5659

5760
with open('../etc/records-type.json') as rtypefile:

0 commit comments

Comments
 (0)