[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

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

@@ -39,7 +39,7 @@ async def policy(
         pilot_stamps: list[str] | None = None,
         job_db: JobDB | None = None,
         job_ids: list[int] | None = None,
-        allow_legacy_pilots: bool = False
+        allow_legacy_pilots: bool = False,
     ):
         assert action, "action is a mandatory parameter"
 
@@ -48,10 +48,11 @@ async def policy(
         # To manage pilots, user have to be an admin
         # In some special cases (described with allow_legacy_pilots), we can allow pilots
         if action == ActionType.MANAGE_PILOTS:
-
             # To make it clear, we separate
             is_an_admin = SERVICE_ADMINISTRATOR in user_info.properties
-            is_a_pilot_if_allowed = allow_legacy_pilots and GENERIC_PILOT in user_info.properties
+            is_a_pilot_if_allowed = (
+                allow_legacy_pilots and GENERIC_PILOT in user_info.properties
+            )
 
             if not is_an_admin and not is_a_pilot_if_allowed:
                 raise HTTPException(
@@ -63,7 +64,7 @@ async def policy(
             if GENERIC_PILOT in user_info.properties:
                 raise HTTPException(
                     status_code=status.HTTP_403_FORBIDDEN,
-                    detail="Pilots can't read other pilots info."
+                    detail="Pilots can't read other pilots info.",
                 )
 
         #

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

@@ -3,7 +3,6 @@
 from http import HTTPStatus
 from typing import Annotated
 
-from diracx.core.properties import GENERIC_PILOT
 from fastapi import Body, Depends, HTTPException, Query, status
 
 from diracx.core.exceptions import (
@@ -15,6 +14,7 @@
     PilotFieldsMapping,
     PilotStatus,
 )
+from diracx.core.properties import GENERIC_PILOT
 from diracx.logic.pilots.management import (
     add_jobs_to_pilot as add_jobs_to_pilot_bl,
 )
@@ -68,7 +68,7 @@ async def add_pilot_stamps(
     # TODO: Verify that grid types, sites, destination sites, etc. are valids
     await check_permissions(
         action=ActionType.MANAGE_PILOTS,
-        allow_legacy_pilots=True # dirac-admin-add-pilot
+        allow_legacy_pilots=True,  # dirac-admin-add-pilot
     )
 
     # Prevent someone who stole a pilot X509 to create thousands of pilots at a time
@@ -77,7 +77,7 @@ async def add_pilot_stamps(
         if len(pilot_stamps) != 1:
             raise HTTPException(
                 status_code=status.HTTP_401_UNAUTHORIZED,
-                detail="As a pilot, you can only create yourself."
+                detail="As a pilot, you can only create yourself.",
             )
 
     try:
@@ -205,10 +205,10 @@ async def update_pilot_fields(
     # Ensures stamps validity
     pilot_stamps = [mapping.PilotStamp for mapping in pilot_stamps_to_fields_mapping]
     await check_permissions(
-        action=ActionType.MANAGE_PILOTS, 
-        pilot_db=pilot_db, 
+        action=ActionType.MANAGE_PILOTS,
+        pilot_db=pilot_db,
         pilot_stamps=pilot_stamps,
-        allow_legacy_pilots=True # dirac-admin-add-pilot
+        allow_legacy_pilots=True,  # dirac-admin-add-pilot
     )
 
     # Prevent someone who stole a pilot X509 to modify thousands of pilots at a time
@@ -218,10 +218,9 @@ async def update_pilot_fields(
         if len(pilot_stamps) != 1:
             raise HTTPException(
                 status_code=status.HTTP_401_UNAUTHORIZED,
-                detail="As a pilot, you can only modify yourself."
+                detail="As a pilot, you can only modify yourself.",
             )
 
-
     await update_pilots_fields(
         pilot_db=pilot_db,
         pilot_stamps_to_fields_mapping=pilot_stamps_to_fields_mapping,

diracx-routers/tests/pilots/test_pilot_creation.py

@@ -25,7 +25,7 @@
         "BaseAccessPolicy",
         "PilotAgentsDB",
         "PilotManagementAccessPolicy",
-        "JobDB"
+        "JobDB",
     ]
 )