Order of retrieved nodes can be disabled

Author: LucasG0

infrahub_sdk/client.py

@@ -51,7 +51,7 @@
 from .schema import InfrahubSchema, InfrahubSchemaSync, NodeSchemaAPI
 from .store import NodeStore, NodeStoreSync
 from .timestamp import Timestamp
-from .types import AsyncRequester, HTTPMethod, SyncRequester
+from .types import AsyncRequester, HTTPMethod, OrderModel, SyncRequester
 from .utils import decode_json, is_valid_uuid
 
 if TYPE_CHECKING:
@@ -540,6 +540,7 @@ async def all(
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
         property: bool = ...,
+        order: OrderModel | None = ...,
     ) -> list[SchemaType]: ...
 
     @overload
@@ -557,6 +558,7 @@ async def all(
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
         property: bool = ...,
+        order: OrderModel | None = ...,
     ) -> list[InfrahubNode]: ...
 
     async def all(
@@ -573,6 +575,7 @@ async def all(
         fragment: bool = False,
         prefetch_relationships: bool = False,
         property: bool = False,
+        order: OrderModel | None = None,
     ) -> list[InfrahubNode] | list[SchemaType]:
         """Retrieve all nodes of a given kind
 
@@ -605,6 +608,7 @@ async def all(
             fragment=fragment,
             prefetch_relationships=prefetch_relationships,
             property=property,
+            order=order,
         )
 
     @overload
@@ -623,6 +627,7 @@ async def filters(
         prefetch_relationships: bool = ...,
         partial_match: bool = ...,
         property: bool = ...,
+        order: OrderModel | None = ...,
         **kwargs: Any,
     ) -> list[SchemaType]: ...
 
@@ -642,6 +647,7 @@ async def filters(
         prefetch_relationships: bool = ...,
         partial_match: bool = ...,
         property: bool = ...,
+        order: OrderModel | None = ...,
         **kwargs: Any,
     ) -> list[InfrahubNode]: ...
 
@@ -660,6 +666,7 @@ async def filters(
         prefetch_relationships: bool = False,
         partial_match: bool = False,
         property: bool = False,
+        order: OrderModel | None = None,
         **kwargs: Any,
     ) -> list[InfrahubNode] | list[SchemaType]:
         """Retrieve nodes of a given kind based on provided filters.
@@ -710,6 +717,7 @@ async def filters(
                 prefetch_relationships=prefetch_relationships,
                 partial_match=partial_match,
                 property=property,
+                order=order,
             )
             query = Query(query=query_data)
             response = await self.execute_graphql(
@@ -1564,6 +1572,7 @@ def all(
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
         property: bool = ...,
+        order: OrderModel | None = ...,
     ) -> list[SchemaTypeSync]: ...
 
     @overload
@@ -1581,6 +1590,7 @@ def all(
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
         property: bool = ...,
+        order: OrderModel | None = ...,
     ) -> list[InfrahubNodeSync]: ...
 
     def all(
@@ -1597,6 +1607,7 @@ def all(
         fragment: bool = False,
         prefetch_relationships: bool = False,
         property: bool = False,
+        order: OrderModel | None = None,
     ) -> list[InfrahubNodeSync] | list[SchemaTypeSync]:
         """Retrieve all nodes of a given kind
 
@@ -1629,6 +1640,7 @@ def all(
             fragment=fragment,
             prefetch_relationships=prefetch_relationships,
             property=property,
+            order=order,
         )
 
     def _process_nodes_and_relationships(
@@ -1682,6 +1694,7 @@ def filters(
         prefetch_relationships: bool = ...,
         partial_match: bool = ...,
         property: bool = ...,
+        order: OrderModel | None = ...,
         **kwargs: Any,
     ) -> list[SchemaTypeSync]: ...
 
@@ -1701,6 +1714,7 @@ def filters(
         prefetch_relationships: bool = ...,
         partial_match: bool = ...,
         property: bool = ...,
+        order: OrderModel | None = ...,
         **kwargs: Any,
     ) -> list[InfrahubNodeSync]: ...
 
@@ -1719,6 +1733,7 @@ def filters(
         prefetch_relationships: bool = False,
         partial_match: bool = False,
         property: bool = False,
+        order: OrderModel | None = None,
         **kwargs: Any,
     ) -> list[InfrahubNodeSync] | list[SchemaTypeSync]:
         """Retrieve nodes of a given kind based on provided filters.
@@ -1769,6 +1784,7 @@ def filters(
                 prefetch_relationships=prefetch_relationships,
                 partial_match=partial_match,
                 property=property,
+                order=order,
             )
             query = Query(query=query_data)
             response = self.execute_graphql(

infrahub_sdk/graphql.py

@@ -2,6 +2,8 @@
 
 from typing import Any
 
+from pydantic import BaseModel
+
 VARIABLE_TYPE_MAPPING = ((str, "String!"), (int, "Int!"), (float, "Float!"), (bool, "Boolean!"))
 
 
@@ -15,6 +17,9 @@ def convert_to_graphql_as_string(value: str | bool | list) -> str:
     if isinstance(value, list):
         values_as_string = [convert_to_graphql_as_string(item) for item in value]
         return "[" + ", ".join(values_as_string) + "]"
+    if isinstance(value, BaseModel):
+        data = value.model_dump()
+        return "{ " + ", ".join(f"{key}: {convert_to_graphql_as_string(val)}" for key, val in data.items()) + " }"
 
     return str(value)
 

infrahub_sdk/node.py

@@ -23,6 +23,7 @@
 
     from .client import InfrahubClient, InfrahubClientSync
     from .schema import AttributeSchemaAPI, MainSchemaTypesAPI, RelationshipSchemaAPI
+    from .types import OrderModel
 
 # pylint: disable=too-many-lines
 
@@ -977,6 +978,7 @@ def generate_query_data_init(
         include: list[str] | None = None,
         exclude: list[str] | None = None,
         partial_match: bool = False,
+        order: OrderModel | None = None,
     ) -> dict[str, Any | dict]:
         data: dict[str, Any] = {
             "count": None,
@@ -985,6 +987,9 @@ def generate_query_data_init(
 
         data["@filters"] = filters or {}
 
+        if order:
+            data["@filters"]["order"] = order
+
         if offset:
             data["@filters"]["offset"] = offset
 
@@ -1176,9 +1181,16 @@ async def generate_query_data(
         prefetch_relationships: bool = False,
         partial_match: bool = False,
         property: bool = False,
+        order: OrderModel | None = None,
     ) -> dict[str, Any | dict]:
         data = self.generate_query_data_init(
-            filters=filters, offset=offset, limit=limit, include=include, exclude=exclude, partial_match=partial_match
+            filters=filters,
+            offset=offset,
+            limit=limit,
+            include=include,
+            exclude=exclude,
+            partial_match=partial_match,
+            order=order,
         )
         data["edges"]["node"].update(
             await self.generate_query_data_node(
@@ -1682,9 +1694,16 @@ def generate_query_data(
         prefetch_relationships: bool = False,
         partial_match: bool = False,
         property: bool = False,
+        order: OrderModel | None = None,
     ) -> dict[str, Any | dict]:
         data = self.generate_query_data_init(
-            filters=filters, offset=offset, limit=limit, include=include, exclude=exclude, partial_match=partial_match
+            filters=filters,
+            offset=offset,
+            limit=limit,
+            include=include,
+            exclude=exclude,
+            partial_match=partial_match,
+            order=order,
         )
         data["edges"]["node"].update(
             self.generate_query_data_node(

infrahub_sdk/schema/main.py

@@ -264,6 +264,7 @@ class BaseSchema(BaseModel):
     icon: str | None = None
     uniqueness_constraints: list[list[str]] | None = None
     documentation: str | None = None
+    order_by: list[str] | None = None
 
     @property
     def kind(self) -> str:

infrahub_sdk/testing/schemas/animal.py

@@ -81,6 +81,7 @@ def schema_cat(self) -> NodeSchema:
             include_in_menu=True,
             inherit_from=[TESTING_ANIMAL],
             display_labels=["name__value", "breed__value", "color__value"],
+            order_by=["name__value"],
             attributes=[
                 Attr(name="breed", kind=AttributeKind.TEXT, optional=False),
                 Attr(name="color", kind=AttributeKind.COLOR, default_value="#555555", optional=True),

infrahub_sdk/types.py

@@ -4,6 +4,8 @@
 from logging import Logger
 from typing import TYPE_CHECKING, Any, Protocol, Union, runtime_checkable
 
+from pydantic import BaseModel
+
 if TYPE_CHECKING:
     import httpx
 
@@ -64,3 +66,7 @@ def exception(self, event: str | None = None, *args: Any, **kw: Any) -> Any:
 
 
 InfrahubLoggers = Union[InfrahubLogger, Logger]
+
+
+class OrderModel(BaseModel):
+    disable: bool | None = None

tests/integration/test_infrahub_client.py

@@ -10,6 +10,7 @@
 from infrahub_sdk.schema import ProfileSchemaAPI
 from infrahub_sdk.testing.docker import TestInfrahubDockerClient
 from infrahub_sdk.testing.schemas.animal import TESTING_ANIMAL, TESTING_CAT, TESTING_DOG, TESTING_PERSON, SchemaAnimal
+from infrahub_sdk.types import OrderModel
 
 if TYPE_CHECKING:
     from infrahub_sdk import InfrahubClient
@@ -59,7 +60,19 @@ async def test_get_all(self, client: InfrahubClient, base_dataset):
         nodes = await client.all(kind=TESTING_CAT)
         assert len(nodes) == 2
         assert isinstance(nodes[0], InfrahubNode)
-        assert sorted([node.name.value for node in nodes]) == ["Bella", "Luna"]
+        assert [node.name.value for node in nodes] == ["Bella", "Luna"]
+
+    async def test_get_all_no_order(self, client: InfrahubClient, base_dataset):
+        nodes = await client.all(kind=TESTING_CAT, order=OrderModel(disable=True))
+        assert len(nodes) == 2
+        assert isinstance(nodes[0], InfrahubNode)
+        assert {node.name.value for node in nodes} == {"Bella", "Luna"}
+
+    async def test_get_filters_no_order(self, client: InfrahubClient, base_dataset):
+        nodes = await client.filters(kind=TESTING_CAT, order=OrderModel(disable=True))
+        assert len(nodes) == 2
+        assert isinstance(nodes[0], InfrahubNode)
+        assert {node.name.value for node in nodes} == {"Bella", "Luna"}
 
     async def test_get_one(self, client: InfrahubClient, base_dataset, cat_luna, person_sophia):
         node1 = await client.get(kind=TESTING_CAT, id=cat_luna.id)