Added Firebase Authentication to training/ Endpoints

training/training/core/authenticator.py

@@ -0,0 +1,14 @@
+from ninja.security import HttpBearer
+import firebase_admin
+
+
+class FirebaseAuth(HttpBearer):
+    def authenticate(self, request, token):
+        if token is None or not token:
+            return
+        try:
+            firebase_admin.auth.verify_id_token(token)
+        except Exception as e:
+            print(e)
+            return
+        return token

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

@@ -1,18 +1,25 @@
 from django.http import HttpRequest
 from ninja import Router, Schema
+from ninja.security import HttpBearer
 from training.core.dataset import SklearnDatasetCreator
 from training.routes.datasets.default.schemas import DefaultDatasetResponse
 from training.routes.schemas import NotFoundError
+from core.authenticator import FirebaseAuth
 
 router = Router()
 
 
 @router.get(
     "{name}/columns",
     response={200: DefaultDatasetResponse, 404: NotFoundError},
+    auth=FirebaseAuth(),
 )
 def defaultDatasets(request: HttpRequest, 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 core.authenticator import FirebaseAuth
 
 router = Router()
 
 
-@router.post("")
+@router.post("", auth=FirebaseAuth())
 def tabularTrain(request: HttpRequest, tabularParams: TabularParams):
     if tabularParams.default:
         dataCreator = SklearnDatasetCreator.fromDefault(