[graphql-pydantic-converter] Add concatenate_queries function render multiple queries together

Author: Jozef Volak

utils/graphql-pydantic-converter/CHANGELOG.md

@@ -9,4 +9,5 @@
 - Generating of pydantic dataclasses for response parsing
 
 # 0.1.2
-- Ignore generating of private graphql schema objects
+- Ignore generating of private graphql schema objects
+- Add graphql_pydantic_converter.graphql_types.concatenate_queries function

utils/graphql-pydantic-converter/graphql_pydantic_converter/graphql_types.py

@@ -8,6 +8,7 @@
 
 if TYPE_CHECKING:
     from typing import Any
+    from typing import Union
 
 
 def generate_type(depth: int) -> str:
@@ -329,3 +330,8 @@ def render(self) -> str:
         if variable:
             variable = f' ( {variable} )'
         return f'{{ { name }{variable} {{ {payload} }} }}'
+
+
+def concatenate_queries(queries: list[Union[Query, Mutation]]) -> str:
+    merged_query = ''.join(query.render()[1:-1] for query in queries)
+    return f'{{ {merged_query} }}'

utils/graphql-pydantic-converter/pyproject.toml

@@ -19,7 +19,7 @@ packages = [{ include = "graphql_pydantic_converter" }]
 name = "graphql-pydantic-converter"
 description = "Convert pydantic schema to pydantic datamodel and build request from it"
 authors = ["Jozef Volak <[email protected]>"]
-version = '0.1.1'
+version = '0.1.2'
 readme = ["README.md", "CHANGELOG.md"]
 keywords = ["graphql", "pydantic"]
 license = "Apache 2.0"

utils/graphql-pydantic-converter/tests/test_graphql_generator.py

@@ -23,6 +23,7 @@
 from render_models import TagAnd
 from render_models import TagOr
 
+import graphql_pydantic_converter.graphql_types
 from graphql_pydantic_converter.graphql_types import ENUM
 from graphql_pydantic_converter.graphql_types import Input
 from graphql_pydantic_converter.schema_converter import GraphqlJsonParser
@@ -196,3 +197,67 @@ def test_parse_response(self) -> None:
 
         assert 'cli_device_import' == response.data.blueprints.edges[0].node.name
 
+    def test_multiple_query(self) -> None:
+
+        query_first = SearchPoolsByTagsQuery(
+            tags=TagOr(
+                matchesAny=[
+                    TagAnd(
+                        matchesAll=[
+                            'root_pool'
+                        ]
+                    )
+                ]
+            ),
+            payload=ResourcePoolConnection(
+                edges=ResourcePoolEdge(
+                    node=ResourcePool(
+                        Name=True,
+                        id=True,
+                        AllocationStrategy=AllocationStrategy(
+                            Name=True
+                        )
+                    )
+                )
+            )
+        )
+
+        query_second = SchedulesQuery(
+            payload=ScheduleConnection(
+                pageInfo=PageInfoSchedule(
+                    hasNextPage=True,
+                    hasPreviousPage=True,
+                    startCursor=True,
+                    endCursor=True
+                ),
+                edges=ScheduleEdge(
+                    node=Schedule(
+                        name=True,
+                        cronString=True,
+                        enabled=True
+                    ),
+                )
+            ),
+            after='aaa',
+            first=10,
+            filter=SchedulesFilterInput(
+                workflowName='TEST_A',
+                workflowVersion='1'
+            )
+        )
+
+        queries = [
+            query_first,
+            query_second
+        ]
+
+        reference = '{  SearchPoolsByTags ( tags: { matchesAny: [  { matchesAll: [ "root_pool" ] }  ]  } ) ' \
+            '{ edges { node { AllocationStrategy { Description Lang Name Script id } ' \
+            'Name PoolProperties PoolType id } } totalCount }  schedules ( after: "aaa", first: 10, filter: ' \
+            '{ workflowName: "TEST_A" , workflowVersion: "1"  } ) { edges { node { name enabled parallelRuns ' \
+            'workflowName workflowVersion cronString workflowContext fromDate toDate status } cursor } ' \
+            'pageInfo { hasNextPage hasPreviousPage startCursor endCursor } totalCount }  }'
+
+        merged_query = graphql_pydantic_converter.graphql_types.concatenate_queries(queries)
+        assert reference == merged_query
+