added get/delete method for mongol db plugin

Author: savon-noir

libnmap/plugins/dbplugin.py

@@ -0,0 +1,17 @@
+#!/usr/bin/env python
+
+class NmapDBPlugin(object):
+    def __init__(self):
+        self.dbname = 'nmapdb'
+        self.store = 'reports'
+
+    def db_insert(self, dict_data):
+        raise NotImplementedError
+    def db_update(self, id):
+        raise NotImplementedError
+    def db_delete(self, id):
+        raise NotImplementedError
+    def db_get(self, id):
+        raise NotImplementedError
+    def db_find(self, key):
+        raise NotImplementedError

libnmap/plugins/mongodb.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 from libnmap.plugins.dbplugin import NmapDBPlugin
 from pymongo import MongoClient
+from bson.objectid import ObjectId
 
 class NmapMongoPlugin(NmapDBPlugin):
     def __init__(self, **kwargs):
@@ -10,3 +11,20 @@ def __init__(self, **kwargs):
 
     def db_insert(self, dict_data):
         self.collection.insert(dict_data)
+
+    def db_get(self, report_id=None):
+        rid = report_id
+        if report_id is not None and isinstance(report_id, str):
+            rid = ObjectId(report_id)
+
+        if isinstance(rid, ObjectId):
+            r = self.collection.find({ '_id': rid })
+        else:
+            r = self.collection.find()
+        return r
+
+    def db_delete(self, report_id=None):
+        if report_id is not None and isinstance(report_id, str):
+            ir = self.collection.remove({ '_id': ObjectId(report_id)})
+        else:
+            r= self.collection.remove({ '_id': report_id })