added schemes to support new ui link options + fix api per new schemes

Change-Id: I768233f13749d01d436c24c764cb903ca19d3d2a
Signed-off-by: Koren Lev <korenlev@gmail.com>

1 files changed, 9 insertions, 10 deletions

diff --git a/app/api/responders/resource/messages.py b/app/api/responders/resource/messages.py
index 367bd34..ff64fbc 100644
--- a/app/api/responders/resource/messages.py
+++ b/app/api/responders/resource/messages.py
@@ -15,16 +15,15 @@ from bson.objectid import ObjectId
 
 
 class Messages(ResponderBase):
-    def __init__(self):
-        super().__init__()
-        self.ID = "id"
-        self.COLLECTION = 'messages'
-        self.PROJECTION = {
-            self.ID: True,
-            "environment": True,
-            "source_system": True,
-            "level": True
-        }
+
+    COLLECTION = 'messages'
+    ID = "id"
+    PROJECTION = {
+        ID: True,
+        "environment": True,
+        "source_system": True,
+        "level": True
+    }
 
     def on_get(self, req, resp):
         self.log.debug("Getting messages from messages")