Add property flag to control if the GraphQL query should include the property

Author: dgarros

infrahub_sdk/client.py

@@ -333,6 +333,7 @@ async def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> Optional[SchemaType]: ...
 
@@ -351,6 +352,7 @@ async def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> SchemaType: ...
 
@@ -369,6 +371,7 @@ async def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> SchemaType: ...
 
@@ -387,6 +390,7 @@ async def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> Optional[InfrahubNode]: ...
 
@@ -405,6 +409,7 @@ async def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> InfrahubNode: ...
 
@@ -423,6 +428,7 @@ async def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> InfrahubNode: ...
 
@@ -440,6 +446,7 @@ async def get(
         populate_store: bool = False,
         fragment: bool = False,
         prefetch_relationships: bool = False,
+        property: bool = False,
         **kwargs: Any,
     ) -> Union[InfrahubNode, SchemaType, None]:
         branch = branch or self.default_branch
@@ -472,6 +479,7 @@ async def get(
             exclude=exclude,
             fragment=fragment,
             prefetch_relationships=prefetch_relationships,
+            property=property,
             **filters,
         )
 
@@ -535,6 +543,7 @@ async def all(
         exclude: Optional[list[str]] = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
     ) -> list[SchemaType]: ...
 
     @overload
@@ -551,6 +560,7 @@ async def all(
         exclude: Optional[list[str]] = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
     ) -> list[InfrahubNode]: ...
 
     async def all(
@@ -566,6 +576,7 @@ async def all(
         exclude: Optional[list[str]] = None,
         fragment: bool = False,
         prefetch_relationships: bool = False,
+        property: bool = False,
     ) -> Union[list[InfrahubNode], list[SchemaType]]:
         """Retrieve all nodes of a given kind
 
@@ -597,6 +608,7 @@ async def all(
             exclude=exclude,
             fragment=fragment,
             prefetch_relationships=prefetch_relationships,
+            property=property,
         )
 
     @overload
@@ -614,6 +626,7 @@ async def filters(
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
         partial_match: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> list[SchemaType]: ...
 
@@ -632,6 +645,7 @@ async def filters(
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
         partial_match: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> list[InfrahubNode]: ...
 
@@ -649,6 +663,7 @@ async def filters(
         fragment: bool = False,
         prefetch_relationships: bool = False,
         partial_match: bool = False,
+        property: bool = False,
         **kwargs: Any,
     ) -> Union[list[InfrahubNode], list[SchemaType]]:
         """Retrieve nodes of a given kind based on provided filters.
@@ -698,6 +713,7 @@ async def filters(
                 fragment=fragment,
                 prefetch_relationships=prefetch_relationships,
                 partial_match=partial_match,
+                # property=property,
             )
             query = Query(query=query_data)
             response = await self.execute_graphql(
@@ -1551,6 +1567,7 @@ def all(
         exclude: Optional[list[str]] = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
     ) -> list[SchemaTypeSync]: ...
 
     @overload
@@ -1567,6 +1584,7 @@ def all(
         exclude: Optional[list[str]] = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
     ) -> list[InfrahubNodeSync]: ...
 
     def all(
@@ -1582,6 +1600,7 @@ def all(
         exclude: Optional[list[str]] = None,
         fragment: bool = False,
         prefetch_relationships: bool = False,
+        property: bool = False,
     ) -> Union[list[InfrahubNodeSync], list[SchemaTypeSync]]:
         """Retrieve all nodes of a given kind
 
@@ -1613,6 +1632,7 @@ def all(
             exclude=exclude,
             fragment=fragment,
             prefetch_relationships=prefetch_relationships,
+            property=property,
         )
 
     def _process_nodes_and_relationships(
@@ -1665,6 +1685,7 @@ def filters(
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
         partial_match: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> list[SchemaTypeSync]: ...
 
@@ -1683,6 +1704,7 @@ def filters(
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
         partial_match: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> list[InfrahubNodeSync]: ...
 
@@ -1700,6 +1722,7 @@ def filters(
         fragment: bool = False,
         prefetch_relationships: bool = False,
         partial_match: bool = False,
+        property: bool = False,
         **kwargs: Any,
     ) -> Union[list[InfrahubNodeSync], list[SchemaTypeSync]]:
         """Retrieve nodes of a given kind based on provided filters.
@@ -1749,6 +1772,7 @@ def filters(
                 fragment=fragment,
                 prefetch_relationships=prefetch_relationships,
                 partial_match=partial_match,
+                property=property,
             )
             query = Query(query=query_data)
             response = self.execute_graphql(
@@ -1801,6 +1825,7 @@ def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> Optional[SchemaTypeSync]: ...
 
@@ -1819,6 +1844,7 @@ def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> SchemaTypeSync: ...
 
@@ -1837,6 +1863,7 @@ def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> SchemaTypeSync: ...
 
@@ -1855,6 +1882,7 @@ def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> Optional[InfrahubNodeSync]: ...
 
@@ -1873,6 +1901,7 @@ def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> InfrahubNodeSync: ...
 
@@ -1891,6 +1920,7 @@ def get(
         populate_store: bool = ...,
         fragment: bool = ...,
         prefetch_relationships: bool = ...,
+        property: bool = ...,
         **kwargs: Any,
     ) -> InfrahubNodeSync: ...
 
@@ -1908,6 +1938,7 @@ def get(
         populate_store: bool = False,
         fragment: bool = False,
         prefetch_relationships: bool = False,
+        property: bool = False,
         **kwargs: Any,
     ) -> Union[InfrahubNodeSync, SchemaTypeSync, None]:
         branch = branch or self.default_branch
@@ -1940,6 +1971,7 @@ def get(
             exclude=exclude,
             fragment=fragment,
             prefetch_relationships=prefetch_relationships,
+            property=property,
             **filters,
         )