Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion infrahub_sdk/ctl/generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,6 @@ async def run(
infrahub_node=InfrahubNode,
)
await generator._init_client.schema.all(branch=generator.branch_name)
await generator.process_nodes(data=data)
await generator.run(identifier=generator_config.name, data=data)

else:
Expand Down
5 changes: 2 additions & 3 deletions infrahub_sdk/generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,16 +97,15 @@
update_group=True,
subscribers=self.subscribers,
)
unpacked = data.get("data") or data
await self.process_nodes(data=unpacked)
return data
return data.get("data") or data

Check warning on line 100 in infrahub_sdk/generator.py

View check run for this annotation

Codecov / codecov/patch

infrahub_sdk/generator.py#L100

Added line #L100 was not covered by tests

async def run(self, identifier: str, data: Optional[dict] = None) -> None:
"""Execute the generator after collecting the data from the GraphQL query."""

if not data:
data = await self.collect_data()
unpacked = data.get("data") or data
await self.process_nodes(data=unpacked)

Check warning on line 108 in infrahub_sdk/generator.py

View check run for this annotation

Codecov / codecov/patch

infrahub_sdk/generator.py#L108

Added line #L108 was not covered by tests

async with self._init_client.start_tracking(
identifier=identifier, params=self.params, delete_unused_nodes=True, group_type="CoreGeneratorGroup"
Expand Down
Loading