refactor: use generator comprehension instead of list comprehension

Co-Authored-By: Aaron <AJ> Steers <[email protected]>

Author: devin-ai-integration[bot]

airbyte_cdk/entrypoint.py

@@ -141,54 +141,53 @@ def run(self, parsed_args: argparse.Namespace) -> Iterable[str]:
                 )
                 if cmd == "spec":
                     message = AirbyteMessage(type=Type.SPEC, spec=source_spec)
-                    yield from [
+                    yield from (
                         self.airbyte_message_to_string(queued_message)
                         for queued_message in self._emit_queued_messages(self.source)
-                    ]
+                    )
                     yield self.airbyte_message_to_string(message)
+                elif (
+                    cmd == "discover"
+                    and not parsed_args.config
+                    and not self.source.check_config_against_spec
+                ):
+                    empty_config: dict[str, Any] = {}
+                    yield from (
+                        self.airbyte_message_to_string(queued_message)
+                        for queued_message in self._emit_queued_messages(self.source)
+                    )
+                    yield from map(
+                        AirbyteEntrypoint.airbyte_message_to_string,
+                        self.discover(source_spec, empty_config),
+                    )
                 else:
-                    if (
-                        cmd == "discover"
-                        and not parsed_args.config
-                        and not self.source.check_config_against_spec
-                    ):
-                        empty_config: dict[str, Any] = {}
-                        yield from [
-                            self.airbyte_message_to_string(queued_message)
-                            for queued_message in self._emit_queued_messages(self.source)
-                        ]
+                    raw_config = self.source.read_config(parsed_args.config)
+                    config = self.source.configure(raw_config, temp_dir)
+
+                    yield from (
+                        self.airbyte_message_to_string(queued_message)
+                        for queued_message in self._emit_queued_messages(self.source)
+                    )
+                    if cmd == "check":
+                        yield from map(
+                            AirbyteEntrypoint.airbyte_message_to_string,
+                            self.check(source_spec, config),
+                        )
+                    elif cmd == "discover":
+                        yield from map(
+                            AirbyteEntrypoint.airbyte_message_to_string,
+                            self.discover(source_spec, config),
+                        )
+                    elif cmd == "read":
+                        config_catalog = self.source.read_catalog(parsed_args.catalog)
+                        state = self.source.read_state(parsed_args.state)
+
                         yield from map(
                             AirbyteEntrypoint.airbyte_message_to_string,
-                            self.discover(source_spec, empty_config),
+                            self.read(source_spec, config, config_catalog, state),
                         )
                     else:
-                        raw_config = self.source.read_config(parsed_args.config)
-                        config = self.source.configure(raw_config, temp_dir)
-
-                        yield from [
-                            self.airbyte_message_to_string(queued_message)
-                            for queued_message in self._emit_queued_messages(self.source)
-                        ]
-                        if cmd == "check":
-                            yield from map(
-                                AirbyteEntrypoint.airbyte_message_to_string,
-                                self.check(source_spec, config),
-                            )
-                        elif cmd == "discover":
-                            yield from map(
-                                AirbyteEntrypoint.airbyte_message_to_string,
-                                self.discover(source_spec, config),
-                            )
-                        elif cmd == "read":
-                            config_catalog = self.source.read_catalog(parsed_args.catalog)
-                            state = self.source.read_state(parsed_args.state)
-
-                            yield from map(
-                                AirbyteEntrypoint.airbyte_message_to_string,
-                                self.read(source_spec, config, config_catalog, state),
-                            )
-                        else:
-                            raise Exception("Unexpected command " + cmd)
+                        raise Exception("Unexpected command " + cmd)
         finally:
             yield from [
                 self.airbyte_message_to_string(queued_message)