Python Configuration.getCSMongoConnection示例

示例#1

# -*- coding: utf-8 -*-
#
# Interface for CVE-Search
#  > Written for MongoDB
#
# Copyright (c) 2015 	Pieter-Jan Moreels
#
# Software is free software released under the "Original BSD license"

# Imports
import re

from lib.Config import Configuration as conf

# Variables
csdb = conf.getCSMongoConnection()
cves = csdb.cves


# Get
def getCVEInfo(cve):
    info = cves.find_one({"id": cve})
    if info: info.pop("_id")
    try:
        info["cvss"] = float(info["cvss"])
    except:
        pass
    return info


def getVulns(cpes, limit):