mirror of
https://github.com/cve-search/cpe-guesser.git
synced 2024-11-15 03:18:28 +00:00
Add missing configuration in lib cpeguesser.py
This commit is contained in:
parent
6be6f0210a
commit
8fc6b9c45e
1 changed files with 6 additions and 1 deletions
|
@ -2,11 +2,16 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
import redis
|
import redis
|
||||||
|
from dynaconf import Dynaconf
|
||||||
|
|
||||||
|
# Configuration
|
||||||
|
settings = Dynaconf(
|
||||||
|
settings_files=['../config/settings.yaml']
|
||||||
|
)
|
||||||
|
|
||||||
class CPEGuesser:
|
class CPEGuesser:
|
||||||
def __init__(self):
|
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):
|
def guessCpe(self, words):
|
||||||
k = []
|
k = []
|
||||||
|
|
Loading…
Reference in a new issue