chore: regenerate client

Author: fstagni

diracx-client/src/diracx/client/_generated/aio/operations/_operations.py

@@ -2451,7 +2451,7 @@ async def add_pilot_stamps(self, body: Union[_models.BodyPilotsAddPilotStamps, I
     async def delete_pilots(
         self,
         *,
-        pilot_stamps: Optional[List[str]] = None,
+        pilot_stamps: Optional[list[str]] = None,
         age_in_days: Optional[int] = None,
         delete_only_aborted: bool = False,
         **kwargs: Any
@@ -2622,7 +2622,7 @@ async def update_pilot_fields(
     @distributed_trace_async
     async def get_pilot_jobs(
         self, *, pilot_stamp: Optional[str] = None, job_id: Optional[int] = None, **kwargs: Any
-    ) -> List[int]:
+    ) -> list[int]:
         """Get Pilot Jobs.
 
         Endpoint only for admins, to get jobs of a pilot.
@@ -2646,7 +2646,7 @@ async def get_pilot_jobs(
         _headers = kwargs.pop("headers", {}) or {}
         _params = kwargs.pop("params", {}) or {}
 
-        cls: ClsType[List[int]] = kwargs.pop("cls", None)
+        cls: ClsType[list[int]] = kwargs.pop("cls", None)
 
         _request = build_pilots_get_pilot_jobs_request(
             pilot_stamp=pilot_stamp,
@@ -2683,7 +2683,7 @@ async def search(
         per_page: int = 100,
         content_type: str = "application/json",
         **kwargs: Any
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Search.
 
         Retrieve information about pilots.
@@ -2711,7 +2711,7 @@ async def search(
         per_page: int = 100,
         content_type: str = "application/json",
         **kwargs: Any
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Search.
 
         Retrieve information about pilots.
@@ -2738,7 +2738,7 @@ async def search(
         page: int = 1,
         per_page: int = 100,
         **kwargs: Any
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Search.
 
         Retrieve information about pilots.
@@ -2765,9 +2765,10 @@ async def search(
         _params = kwargs.pop("params", {}) or {}
 
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-        cls: ClsType[List[Dict[str, Any]]] = kwargs.pop("cls", None)
+        content_type = content_type if body else None
+        cls: ClsType[list[dict[str, Any]]] = kwargs.pop("cls", None)
 
-        content_type = content_type or "application/json"
+        content_type = content_type or "application/json" if body else None
         _json = None
         _content = None
         if isinstance(body, (IOBase, bytes)):

diracx-client/src/diracx/client/_generated/aio/operations/_patch.py

@@ -6,17 +6,23 @@
 
 Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize
 """
+
 from __future__ import annotations
 
 __all__ = [
     "AuthOperations",
     "JobsOperations",
-    "PilotsOperations",
 ]  # Add all objects you want publicly available to users at this package level
 
 from ....patches.auth.aio import AuthOperations
 from ....patches.jobs.aio import JobsOperations
-from ....patches.pilots.aio import PilotsOperations
+
+try:
+    from ....patches.pilots.aio import PilotsOperations
+
+    __all__.append("PilotsOperations")
+except ImportError:
+    pass
 
 
 def patch_sdk():

diracx-client/src/diracx/client/_generated/models/_models.py

@@ -5,15 +5,13 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-from collections.abc import MutableMapping
 import datetime
 from typing import Any, Optional, TYPE_CHECKING, Union
 
 from .._utils import serialization as _serialization
 
 if TYPE_CHECKING:
     from .. import models as _models
-JSON = MutableMapping[str, Any]
 
 
 class BodyAuthGetOidcToken(_serialization.Model):
@@ -224,12 +222,12 @@ class BodyPilotsAddPilotStamps(_serialization.Model):
     def __init__(
         self,
         *,
-        pilot_stamps: List[str],
+        pilot_stamps: list[str],
         vo: str,
         grid_type: str = "Dirac",
         grid_site: str = "Unknown",
         destination_site: str = "NotAssigned",
-        pilot_references: Optional[Dict[str, str]] = None,
+        pilot_references: Optional[dict[str, str]] = None,
         pilot_status: Optional[Union[str, "_models.PilotStatus"]] = None,
         **kwargs: Any
     ) -> None:
@@ -277,7 +275,7 @@ class BodyPilotsUpdatePilotFields(_serialization.Model):
         "pilot_stamps_to_fields_mapping": {"key": "pilot_stamps_to_fields_mapping", "type": "[PilotFieldsMapping]"},
     }
 
-    def __init__(self, *, pilot_stamps_to_fields_mapping: List["_models.PilotFieldsMapping"], **kwargs: Any) -> None:
+    def __init__(self, *, pilot_stamps_to_fields_mapping: list["_models.PilotFieldsMapping"], **kwargs: Any) -> None:
         """
         :keyword pilot_stamps_to_fields_mapping: (pilot_stamp, pilot_fields) mapping to change.
          Required.
@@ -1706,10 +1704,6 @@ class ValidationError(_serialization.Model):
     :vartype msg: str
     :ivar type: Error Type. Required.
     :vartype type: str
-    :ivar input: Input.
-    :vartype input: any
-    :ivar ctx: Context.
-    :vartype ctx: JSON
     """
 
     _validation = {
@@ -1722,38 +1716,21 @@ class ValidationError(_serialization.Model):
         "loc": {"key": "loc", "type": "[object]"},
         "msg": {"key": "msg", "type": "str"},
         "type": {"key": "type", "type": "str"},
-        "input": {"key": "input", "type": "object"},
-        "ctx": {"key": "ctx", "type": "object"},
     }
 
-    def __init__(
-        self,
-        *,
-        loc: list[Any],
-        msg: str,
-        type: str,
-        input: Optional[Any] = None,
-        ctx: Optional[JSON] = None,
-        **kwargs: Any
-    ) -> None:
+    def __init__(self, *, loc: list[Any], msg: str, type: str, **kwargs: Any) -> None:
         """
         :keyword loc: Location. Required.
         :paramtype loc: list[any]
         :keyword msg: Message. Required.
         :paramtype msg: str
         :keyword type: Error Type. Required.
         :paramtype type: str
-        :keyword input: Input.
-        :paramtype input: any
-        :keyword ctx: Context.
-        :paramtype ctx: JSON
         """
         super().__init__(**kwargs)
         self.loc = loc
         self.msg = msg
         self.type = type
-        self.input = input
-        self.ctx = ctx
 
 
 class VectorSearchSpec(_serialization.Model):

diracx-client/src/diracx/client/_generated/operations/_operations.py

@@ -584,7 +584,7 @@ def build_pilots_add_pilot_stamps_request(**kwargs: Any) -> HttpRequest:
 
 def build_pilots_delete_pilots_request(
     *,
-    pilot_stamps: Optional[List[str]] = None,
+    pilot_stamps: Optional[list[str]] = None,
     age_in_days: Optional[int] = None,
     delete_only_aborted: bool = False,
     **kwargs: Any
@@ -3062,7 +3062,7 @@ def add_pilot_stamps(self, body: Union[_models.BodyPilotsAddPilotStamps, IO[byte
     def delete_pilots(  # pylint: disable=inconsistent-return-statements
         self,
         *,
-        pilot_stamps: Optional[List[str]] = None,
+        pilot_stamps: Optional[list[str]] = None,
         age_in_days: Optional[int] = None,
         delete_only_aborted: bool = False,
         **kwargs: Any
@@ -3231,7 +3231,7 @@ def update_pilot_fields(  # pylint: disable=inconsistent-return-statements
     @distributed_trace
     def get_pilot_jobs(
         self, *, pilot_stamp: Optional[str] = None, job_id: Optional[int] = None, **kwargs: Any
-    ) -> List[int]:
+    ) -> list[int]:
         """Get Pilot Jobs.
 
         Endpoint only for admins, to get jobs of a pilot.
@@ -3255,7 +3255,7 @@ def get_pilot_jobs(
         _headers = kwargs.pop("headers", {}) or {}
         _params = kwargs.pop("params", {}) or {}
 
-        cls: ClsType[List[int]] = kwargs.pop("cls", None)
+        cls: ClsType[list[int]] = kwargs.pop("cls", None)
 
         _request = build_pilots_get_pilot_jobs_request(
             pilot_stamp=pilot_stamp,
@@ -3292,7 +3292,7 @@ def search(
         per_page: int = 100,
         content_type: str = "application/json",
         **kwargs: Any
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Search.
 
         Retrieve information about pilots.
@@ -3320,7 +3320,7 @@ def search(
         per_page: int = 100,
         content_type: str = "application/json",
         **kwargs: Any
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Search.
 
         Retrieve information about pilots.
@@ -3347,7 +3347,7 @@ def search(
         page: int = 1,
         per_page: int = 100,
         **kwargs: Any
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """Search.
 
         Retrieve information about pilots.
@@ -3374,9 +3374,10 @@ def search(
         _params = kwargs.pop("params", {}) or {}
 
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-        cls: ClsType[List[Dict[str, Any]]] = kwargs.pop("cls", None)
+        content_type = content_type if body else None
+        cls: ClsType[list[dict[str, Any]]] = kwargs.pop("cls", None)
 
-        content_type = content_type or "application/json"
+        content_type = content_type or "application/json" if body else None
         _json = None
         _content = None
         if isinstance(body, (IOBase, bytes)):

diracx-client/src/diracx/client/_generated/operations/_patch.py

@@ -6,17 +6,23 @@
 
 Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize
 """
+
 from __future__ import annotations
 
 __all__ = [
     "AuthOperations",
     "JobsOperations",
-    "PilotsOperations",
 ]  # Add all objects you want publicly available to users at this package level
 
 from ...patches.auth.sync import AuthOperations
 from ...patches.jobs.sync import JobsOperations
-from ...patches.pilots.sync import PilotsOperations
+
+try:
+    from ...patches.pilots.sync import PilotsOperations
+
+    __all__.append("PilotsOperations")
+except ImportError:
+    pass
 
 
 def patch_sdk():

diracx-core/src/diracx/core/models/job.py

@@ -6,7 +6,7 @@
 from __future__ import annotations
 
 from enum import StrEnum
-from typing import Literal, Optional
+from typing import Literal
 
 from pydantic import BaseModel, Field, field_validator
 

diracx-core/src/diracx/core/models/search.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from datetime import datetime
 from enum import StrEnum
 
 from pydantic import BaseModel
@@ -24,7 +25,7 @@ class VectorSearchOperator(StrEnum):
 class ScalarSearchSpec(TypedDict):
     parameter: str
     operator: ScalarSearchOperator
-    value: str | int
+    value: str | int | datetime
 
 
 class VectorSearchSpec(TypedDict):

diracx-db/src/diracx/db/sql/dummy/db.py

@@ -3,7 +3,7 @@
 from sqlalchemy import insert
 from uuid_utils import UUID
 
-from diracx.core.models import SearchSpec
+from diracx.core.models.search import SearchSpec
 from diracx.db.sql.utils import BaseSQLDB
 
 from .schema import Base as DummyDBBase

diracx-db/src/diracx/db/sql/pilots/db.py

@@ -76,11 +76,7 @@ async def add_jobs_to_pilot(self, job_to_pilot_mapping: list[dict[str, Any]]):
         """Associate a pilot with jobs.
 
         job_to_pilot_mapping format:
-            ```py
-            job_to_pilot_mapping = [
-                {"PilotID": pilot_id, "JobID": job_id, "StartTime": now}
-            ]
-            ```
+            job_to_pilot_mapping = [{"PilotID": pilot_id, "JobID": job_id, "StartTime": now}]
 
         Raises:
         - PilotNotFoundError if a pilot_id is not associated with a pilot.
@@ -144,7 +140,7 @@ async def update_pilot_fields(
         """Bulk update pilots with a mapping.
 
         pilot_stamps_to_fields_mapping format:
-            ```py
+
             [
                 {
                     "PilotStamp": pilot_stamp,
@@ -157,7 +153,6 @@ async def update_pilot_fields(
                     ...
                 }
             ]
-            ```
 
         The mapping helps to update multiple fields at a time.
 

diracx-db/src/diracx/db/sql/pilots/schema.py

@@ -11,6 +11,7 @@
 )
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
 
+from diracx.core.models.pilot import PilotStatus
 from diracx.db.sql.utils import (
     EnumBackedBool,
     str32,