From c51acb22350ff3b6cb201aa3f0d8a747c3993c9a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9lian=20GARCIA?= <celian.garcia1@gmail.com>
Date: Sun, 2 Oct 2022 14:05:36 +0200
Subject: [PATCH] fix: restablish mongo repository declaration

---
 cerebro/server.py | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/cerebro/server.py b/cerebro/server.py
index 8f09441..9284b7a 100644
--- a/cerebro/server.py
+++ b/cerebro/server.py
@@ -43,24 +43,19 @@ def _to_uppercase(self, obj: Dict[str, Any]) -> Dict[str, Any]:
         return retval
 
 
-def build_repository(**params) -> Repository:
-    if params["use_mongo"]:
-        return NLPRepositoryMongo(
-            url=params["mongo_url"],
-            database=params["mongo_database"]
-        )
-    else:
-        return NLPRepositoryFake()
-
-
 logger = logging.getLogger()
 
 # Initialize the sanic app
 config = YamlConfig(path="cerebro.yaml")
 app = Sanic(name="cerebro", config=config)
 
-# repository = build_repository(**params)
-repository = NLPRepositoryFake()
+if config["CEREBRO"]["FEATURES"]["USE_MONGO"]:
+    repository = NLPRepositoryMongo(
+        url=["CEREBRO"]["MONGODB"]["URL"],
+        database=["CEREBRO"]["MONGODB"]["DATABASE"]
+    )
+else:
+    repository = NLPRepositoryFake()
 
 app.add_route(HtmlView.as_view(), '/')
 app.add_route(SamplesView.as_view(repository), '/models/<model_id:str>/samples')