update config path

Author: pnilan

airbyte_cdk/entrypoint.py

@@ -187,7 +187,7 @@ def run(self, parsed_args: argparse.Namespace) -> Iterable[str]:
                     raw_config = self.source.read_config(parsed_args.config)
                     config = self.source.configure(raw_config, temp_dir)
                     mutable_config = dict(config)
-                    config_path = self.extract_config(sys.argv[1:])
+                    config_path = parsed_args.config if parsed_args.config else None
                     if config_path:
                         self.source.migrate_config(config_path, mutable_config)
                     self.source.transform_config(mutable_config)

airbyte_cdk/sources/declarative/manifest_declarative_source.py

@@ -8,7 +8,7 @@
 from copy import deepcopy
 from importlib import metadata
 from types import ModuleType
-from typing import Any, Dict, Iterator, List, Mapping, MutableMapping, Optional, Set, cast
+from typing import Any, Dict, Iterator, List, Mapping, MutableMapping, Optional, Set
 
 import yaml
 from jsonschema.exceptions import ValidationError
@@ -296,7 +296,7 @@ def streams(self, config: Mapping[str, Any]) -> List[Stream]:
 
         return source_streams
 
-    def migrate_config(self, config_path: Optional[Any], config: MutableMapping[str, Any]) -> None:
+    def migrate_config(self, config_path: Optional[str], config: MutableMapping[str, Any]) -> None:
         self._spec_component.migrate_config(config_path, config) if self._spec_component else None
 
     def transform_config(self, config: MutableMapping[str, Any]) -> None:

airbyte_cdk/sources/declarative/spec/spec.py

@@ -69,7 +69,7 @@ def generate_spec(self) -> ConnectorSpecification:
         # We remap these keys to camel case because that's the existing format expected by the rest of the platform
         return ConnectorSpecificationSerializer.load(obj)
 
-    def migrate_config(self, config_path: Optional[Any], config: Mapping[str, Any]) -> None:
+    def migrate_config(self, config_path: Optional[str], config: Mapping[str, Any]) -> None:
         """
         Apply all specified config transformations to the provided config and save the modified config to the given path and emit a control message.
 

airbyte_cdk/sources/source.py

@@ -94,8 +94,14 @@ def name(self) -> str:
         """Source name"""
         return self.__class__.__name__
 
-    def migrate_config(self, config_path: str, config: MutableMapping[str, Any]) -> None:
+    def migrate_config(self, config_path: Optional[str], config: MutableMapping[str, Any]) -> None:
+        """
+        Optional method to migrate config.
+        """
         pass
 
     def transform_config(self, config: MutableMapping[str, Any]) -> None:
+        """
+        Optional method to transform config.
+        """
         pass

unit_tests/sources/declarative/spec/test_spec.py

@@ -222,7 +222,7 @@ def test_given_unmigrated_config_when_migrating_then_config_is_migrated(migratio
     )
     spec.message_repository = migration_mocks["message_repository"]
 
-    spec.migrate_config(["spec"], migration_mocks["source"], input_config)
+    spec.migrate_config("/fake/config/path", input_config)
 
     migration_mocks["message_repository"].emit_message.assert_called_once()
     migration_mocks["open"].assert_called_once_with("/fake/config/path", "w")
@@ -252,7 +252,7 @@ def test_given_already_migrated_config_no_control_message_is_emitted(migration_m
     )
     spec.message_repository = migration_mocks["message_repository"]
 
-    spec.migrate_config(["spec"], migration_mocks["source"], input_config)
+    spec.migrate_config("/fake/config/path", input_config)
 
     migration_mocks["message_repository"].emit_message.assert_not_called()
     migration_mocks["open"].assert_not_called()