diff --git a/lib/cpeguesser.py b/lib/cpeguesser.py index 45d20e2..cbc9789 100644 --- a/lib/cpeguesser.py +++ b/lib/cpeguesser.py @@ -2,11 +2,16 @@ # -*- coding: utf-8 -*- import redis +from dynaconf import Dynaconf +# Configuration +settings = Dynaconf( + settings_files=['../config/settings.yaml'] +) class CPEGuesser: def __init__(self): - self.rdb = redis.Redis(host='127.0.0.1', port=6379, db=8, decode_responses=True) + self.rdb = redis.Redis(host=settings.redis.host, port=settings.redis.port, db=8, decode_responses=True) def guessCpe(self, words): k = []