Commit c0a5dde5 authored by Pidgey's avatar Pidgey
Browse files

Update freetext to be mongo 3 compatible ( #124 )

parent 555ae97c
Loading
Loading
Loading
Loading
+4 −1
Original line number Diff line number Diff line
@@ -228,7 +228,10 @@ def getCPEMatching(regex, fullSearch=False):
  return lst

def getFreeText(text):
  try: # Before Mongo 3
    return [x["obj"] for x in db.command("text", "cves", search=text)["results"]]
  except: # As of Mongo 3
    return sanitize(colCVE.find({"$text":{"$search":text}}))

def getCAPECFor(cwe):
  return sanitize(colCAPEC.find({'related_weakness': {'$in': [cwe]}}))