airbyte-lib: Add uninstall

airbyte-lib/airbyte_lib/executor.py

@@ -38,6 +38,10 @@ def ensure_installation(self):
     def install(self):
         pass
 
+    @abstractmethod
+    def uninstall(self):
+        pass
+
 
 @contextmanager
 def _stream_from_subprocess(args: List[str]) -> Generator[Iterable[str], None, None]:
@@ -109,6 +113,11 @@ def _run_subprocess_and_raise_on_failure(self, args: List[str]):
         if result.returncode != 0:
             raise Exception(f"Install process exited with code {result.returncode}")
 
+    def uninstall(self):
+        venv_name = self._get_venv_name()
+        if os.path.exists(venv_name):
+            self._run_subprocess_and_raise_on_failure(["rm", "-rf", venv_name])
+
     def install(self):
         venv_name = self._get_venv_name()
         self._run_subprocess_and_raise_on_failure([sys.executable, "-m", "venv", venv_name])
@@ -187,6 +196,9 @@ def ensure_installation(self):
     def install(self):
         raise Exception(f"Connector {self.metadata.name} is not available - cannot install it")
 
+    def uninstall(self):
+        raise Exception(f"Connector {self.metadata.name} is installed manually and not managed by airbyte-lib - please remove it manually")
+
     def execute(self, args: List[str]) -> Iterable[str]:
         with _stream_from_subprocess([self.metadata.name] + args) as stream:
             yield from stream

airbyte-lib/airbyte_lib/factories.py

@@ -19,7 +19,7 @@ def get_connector(
     pip_url: str | None = None,
     config: dict[str, Any] | None = None,
     use_local_install: bool = False,
-    install_if_missing: bool = False,
+    install_if_missing: bool = True,
 ):
     """
     Get a connector by name and version.

airbyte-lib/airbyte_lib/source.py

@@ -172,8 +172,17 @@ def check(self):
             raise Exception(f"Connector did not return check status. Last logs: {self._last_log_messages}")
 
     def install(self):
+        """
+        Install the connector if it is not yet installed.
+        """
         self.executor.install()
 
+    def uninstall(self):
+        """
+        Uninstall the connector if it is installed. This only works if the use_local_install flag wasn't used and installation is managed by airbyte-lib.
+        """
+        self.executor.uninstall()
+
     def _read(self) -> Iterable[AirbyteRecordMessage]:
         """
         Call read on the connector.