Fixup generator

Author: afausti

src/kafkaaggregator/cli.py

@@ -1,11 +1,12 @@
 """Command-line interface for kafkaaggregator."""
 
-__all__ = ["main", "produce", "init_example"]
+__all__ = ["main", "produce", "init_example", "generate_agents"]
 
 import logging
 from pathlib import Path
 
 from faust.cli import AppCommand, option
+from faust.cli.base import argument
 
 from kafkaaggregator.app import app, config
 from kafkaaggregator.config import ExampleConfiguration
@@ -60,6 +61,14 @@ async def init_example(self: AppCommand) -> None:
 
 
 @app.command(
+    argument(
+        "aggregated_topic",
+        type=str,
+        help=(
+            "The aggregated topic to generate the agent for. If not specified "
+            "generate agents for all aggregated topics in the configuration."
+        ),
+    ),
     option(
         "--template-file",
         type=str,
@@ -81,19 +90,11 @@ async def init_example(self: AppCommand) -> None:
         help="Aggregator configuration file.",
         show_default=True,
     ),
-    option(
-        "--aggregated-topic",
-        type=str,
-        help=(
-            "The aggregated topic to generate the agent for. If not specified "
-            "generate agents for all aggregated topics in the configuration."
-        ),
-    ),
 )
 async def generate_agents(
     self: AppCommand,
-    config_file: str,
     aggregated_topic: str,
+    config_file: str,
     template_file: str,
     output_dir: str,
 ) -> None:

src/kafkaaggregator/generator.py

@@ -28,8 +28,6 @@ class AgentGenerator:
         Name of the aggregated topic.
     template_file: str
         Name of the agent Jinja2 template file
-
-
     """
 
     logger = logger
@@ -49,7 +47,6 @@ def __init__(
 
         # Supports the 1 source topic -> 1 aggregated topic case for the moment
         self._source_topic = aggregator_config.source_topics[0]
-        self._template: Template = self._load_template()
         self._window_size_secods = (
             aggregator_config.window_aggregation.window_size_seconds
         )