Added Firebase Authentication to training/ Endpoints

training/training/core/authenticator.py

@@ -0,0 +1,25 @@
+from django.http import HttpRequest
+from ninja.security import HttpBearer
+from cli import init_firebase
+import firebase_admin.auth
+import logging
+
+logger = logging.getLogger()
+
+
+class FirebaseAuth(HttpBearer):
+    def authenticate(self, request, token):
+        app = init_firebase()
+        if token is None or not token:
+            return
+        try:
+            firebase_admin.auth.verify_id_token(token)
+            firebase_admin.delete_app(app)
+        except Exception as e:
+            logger.info(e)
+            return
+        return token
+
+
+class Request(HttpRequest):
+    auth: str

training/training/routes/datasets/default/columns.py

@@ -1,5 +1,7 @@
 from django.http import HttpRequest
 from ninja import Router, Schema
+from ninja.security import HttpBearer
+from training.core.authenticator import FirebaseAuth, Request
 from training.core.dataset import SklearnDatasetCreator
 from training.routes.datasets.default.schemas import DefaultDatasetResponse
 from training.routes.schemas import NotFoundError
@@ -10,9 +12,14 @@
 @router.get(
     "{name}/columns",
     response={200: DefaultDatasetResponse, 404: NotFoundError},
+    auth=FirebaseAuth(),
 )
-def defaultDatasets(request: HttpRequest, name: str):
+def defaultDatasets(request: Request, name: str):
     if not name in SklearnDatasetCreator.DEFAULT_DATASETS:
         return 404, {"message": "Dataset not found"}
     dataset = SklearnDatasetCreator.getDefaultDataset(name)
-    return 200, {"data": dataset.columns.tolist(), "message": "Success"}
+    return 200, {
+        "data": dataset.columns.tolist(),
+        "message": "Success",
+        "token": request.auth,
+    }

training/training/routes/tabular/tabular.py

@@ -8,11 +8,12 @@
 from training.core.optimizer import getOptimizer
 from training.core.trainer import ClassificationTrainer, RegressionTrainer
 from training.routes.tabular.schemas import TabularParams
+from training.core.authenticator import FirebaseAuth
 
 router = Router()
 
 
-@router.post("")
+@router.post("", auth=FirebaseAuth())
 def tabularTrain(request: HttpRequest, tabularParams: TabularParams):
     if tabularParams.default:
         dataCreator = SklearnDatasetCreator.fromDefault(