fix: Removing os endpoints

Author: Robin-Van-de-Merghel

diracx-routers/src/diracx/routers/pilots/debug.py

@@ -1,94 +1,23 @@
 from __future__ import annotations
 
-from os import getenv
 from typing import Annotated
 
 from fastapi import (
     Depends,
-    HTTPException,
-    status,
 )
 
-from diracx.core.exceptions import (
-    PilotAlreadyExistsError,
-    PilotNotFoundError,
-)
-
-from ..dependencies import (
-    PilotAgentsDB,
-)
 from ..fastapi_classes import DiracxRouter
 from ..utils.users import AuthorizedUserInfo, verify_dirac_access_token
 from .access_policies import RegisteredPilotAccessPolicyCallable
 
 router = DiracxRouter(require_auth=False)
 
-# Debug only route
-# Keep it?
-# TODO: Add to the env?
-if getenv("production") is None:
-
-    @router.get("/info")
-    async def get_pilot_info(
-        check_permissions: RegisteredPilotAccessPolicyCallable,
-        user_info: Annotated[AuthorizedUserInfo, Depends(verify_dirac_access_token)],
-    ):
-        await check_permissions()
-
-        return user_info
-
-    @router.post("/register-pilot")
-    async def add_new_pilot(
-        pilot_db: PilotAgentsDB,
-        vo: str = "default-vo",
-        initial_job_id: int = 0,
-        current_job_id: int = 0,
-        benchmark: float = 0.0,
-        pilot_job_reference: str = "Unknown",
-        pilot_stamp: str = "",
-        status: str = "Unknown",
-        status_reason: str = "Unknown",
-        queue: str = "Unknown",
-        grid_site: str = "Unknown",
-        destination_site: str = "NotAssigned",
-        grid_type: str = "LCG",
-        submission_time: str | None = None,  # ?
-        last_update_time: str | None = None,  # = now?
-        accounting_sent: bool = False,
-    ):
-        return {
-            "id": await pilot_db.register_new_pilot(
-                initial_job_id=initial_job_id,
-                current_job_id=current_job_id,
-                pilot_job_reference=pilot_job_reference,
-                pilot_stamp=pilot_stamp,
-                destination_site=destination_site,
-                queue=queue,
-                grid_site=grid_site,
-                vo=vo,
-                grid_type=grid_type,
-                benchmark=benchmark,
-                submission_time=submission_time,
-                last_update_time=last_update_time,
-                status=status,
-                status_reason=status_reason,
-                accounting_sent=accounting_sent,
-            )
-        }
 
-    @router.post("/add-credentials")
-    async def add_credentials(pilot_db: PilotAgentsDB, pilot_id: int):
-        try:
-            return {"secret": await pilot_db.add_pilot_credentials(pilot_id=pilot_id)}
-        except PilotNotFoundError as e:
-            raise HTTPException(
-                status_code=status.HTTP_404_NOT_FOUND, detail=e.detail
-            ) from e
-        except PilotAlreadyExistsError as e:
-            raise HTTPException(
-                status_code=status.HTTP_409_CONFLICT, detail=e.detail
-            ) from e
+@router.get("/info")
+async def get_pilot_info(
+    check_permissions: RegisteredPilotAccessPolicyCallable,
+    user_info: Annotated[AuthorizedUserInfo, Depends(verify_dirac_access_token)],
+):
+    await check_permissions()
 
-    @router.post("/get-pilots")
-    async def get_pilots(pilot_db: PilotAgentsDB):
-        return await pilot_db.get_pilots()
+    return user_info