cpe-guesser/lib/cpeguesser.py

#!/usr/bin/env python3
# -*- 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=settings.redis.host, port=settings.redis.port, db=8, decode_responses=True)

    def guessCpe(self, words):
        k = []
        for keyword in words:
            k.append(f"w:{keyword.lower()}")

        maxinter = len(k)
        cpes = []
        for x in reversed(range(maxinter)):
            ret = self.rdb.sinter(k[x])
            cpes.append(list(ret))
        result = set(cpes[0]).intersection(*cpes)

        ranked = []

        for cpe in result:
            rank = self.rdb.zrank('rank:cpe', cpe)
            ranked.append((rank, cpe))

        return sorted(ranked)