Address review: broaden exception handling, reuse Docker client

Catch docker.errors.DockerException and requests.exceptions.RequestException
in pull_or_build_image to handle connection-level failures during pull.
Pass outer ContainerConnection client to MySQLMgmt instead of creating
a fresh docker.from_env() instance.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: szachovy

src/container.py

@@ -104,7 +104,7 @@ def __init__(self, container: str | None) -> None:
     def pull_or_build_image(client: docker.client.DockerClient, image: str, build_context: str) -> None:
         try:
             client.images.pull(image)
-        except docker.errors.APIError:
+        except (docker.errors.DockerException, requests.exceptions.RequestException):
             client.images.build(path=build_context, tag=image)
 
     def run_command_on_the_container(
@@ -288,6 +288,7 @@ class MySQLMgmt(ContainerInstance):
             # pylint: disable=too-many-instance-attributes
             def __init__(
                         self,
+                        client: docker.client.DockerClient,
                         virtual_ip_address: str,
                         virtual_network_mask: str,
                         virtual_network_interface: str,
@@ -297,6 +298,7 @@ def __init__(
                         state: str,
                         priority: str
                     ) -> None:
+                self.client = client
                 self.virtual_ip_address = virtual_ip_address
                 self.virtual_network_mask = virtual_network_mask
                 self.virtual_network_interface = virtual_network_interface
@@ -329,7 +331,7 @@ def setup_env(self) -> None:
             def run(self) -> None:
                 self.setup_env()
                 ContainerConnection.pull_or_build_image(
-                    docker.from_env(),
+                    self.client,
                     "ghcr.io/szachovy/superset-cluster-mysql-mgmt:latest",
                     "/opt/superset-cluster/mysql-mgmt"
                 )
@@ -359,6 +361,7 @@ def run(self) -> None:
         return print(
             self.wait_until_healthy(
                 MySQLMgmt(  # type: ignore[arg-type]
+                    client=self.client,
                     virtual_ip_address=virtual_ip_address,
                     virtual_network_mask=virtual_network_mask,
                     virtual_network_interface=virtual_network_interface,