1 files changed, 23 insertions, 0 deletions
diff --git a/app/discover/network_agents_list.py b/app/discover/network_agents_list.py
new file mode 100644
index 0000000..c1c1b36
--- /dev/null
+++ b/app/discover/network_agents_list.py
@@ -0,0 +1,23 @@
+###############################################################################
+# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems)   #
+# and others                                                                  #
+#                                                                             #
+# All rights reserved. This program and the accompanying materials            #
+# are made available under the terms of the Apache License, Version 2.0       #
+# which accompanies this distribution, and is available at                    #
+# http://www.apache.org/licenses/LICENSE-2.0                                  #
+###############################################################################
+from utils.mongo_access import MongoAccess
+
+
+class NetworkAgentsList(MongoAccess):
+    def __init__(self):
+        super(NetworkAgentsList, self).__init__()
+        self.list = MongoAccess.db["network_agent_types"]
+
+    def get_type(self, type):
+        matches = self.list.find({"type": type})
+        for doc in matches:
+            doc["_id"] = str(doc["_id"])
+            return doc
+        return {}