Commit b8a5a14a authored by Pieter-Jan Moreels's avatar Pieter-Jan Moreels
Browse files

more abstraction

parent 133229b4
Loading
Loading
Loading
Loading
+8 −6
Original line number Diff line number Diff line
@@ -183,15 +183,16 @@ def isSingleMaster(user):

# Query Functions
# Generic data
def getCVEs(limit=False, query=[], skip=0, cves=None):
def getCVEs(limit=False, query=[], skip=0, cves=None, collection=None):
  col=colCVE if not collection else db[collection]
  if type(query) == dict: query=[query]
  if type(cves) == list: query.append({"id": {"$in": cves}})
  if len(query) == 0:
    cve=colCVE.find().sort("Modified", -1).limit(limit).skip(skip)
    cve=col.find().sort("Modified", -1).limit(limit).skip(skip)
  elif len(query)  == 1:
    cve=colCVE.find(query[0]).sort("Modified", -1).limit(limit).skip(skip)
    cve=col.find(query[0]).sort("Modified", -1).limit(limit).skip(skip)
  else:
    cve=colCVE.find({"$and": query}).sort("Modified", -1).limit(limit).skip(skip)
    cve=col.find({"$and": query}).sort("Modified", -1).limit(limit).skip(skip)
  return sanitize(cve)

def getCVEsNewerThan(dt):
@@ -200,8 +201,9 @@ def getCVEsNewerThan(dt):
def getCVEIDs(limit=-1):
  return [x["id"] for x in colCVE.find().limit(limit).sort("Modified", -1)]

def getCVE(id):
  return sanitize(colCVE.find_one({"id": id}))
def getCVE(id, collection=None):
  col=colCVE if not collection else db[collection]
  return sanitize(col.find_one({"id": id}))

def getCPE(id):
  return sanitize(colCPE.find_one({"id": id}))