Use relative imports in core package.

Author: tillahoffmann

core/testcontainers/core/container.py

@@ -1,10 +1,10 @@
 import os
 from docker.models.containers import Container
 
-from testcontainers.core.waiting_utils import wait_container_is_ready
-from testcontainers.core.docker_client import DockerClient
-from testcontainers.core.exceptions import ContainerStartException
-from testcontainers.core.utils import setup_logger, inside_container, is_arm
+from .waiting_utils import wait_container_is_ready
+from .docker_client import DockerClient
+from .exceptions import ContainerStartException
+from .utils import setup_logger, inside_container, is_arm
 
 logger = setup_logger(__name__)
 

core/testcontainers/core/docker_client.py

@@ -17,9 +17,7 @@
 from docker.errors import NotFound
 from docker.models.containers import Container
 
-from testcontainers.core.utils import inside_container
-from testcontainers.core.utils import default_gateway_ip
-from testcontainers.core.utils import setup_logger
+from .utils import default_gateway_ip, inside_container, setup_logger
 
 
 LOGGER = setup_logger(__name__)

core/testcontainers/core/exceptions.py

@@ -12,15 +12,7 @@
 #    under the License.
 
 
-class ContainerStartException(Exception):
-    pass
-
-
-class TimeoutException(Exception):
-    pass
-
-
-class NoSuchBrowserException(Exception):
+class ContainerStartException(RuntimeError):
     pass
 
 

core/testcontainers/core/generic.py

@@ -11,10 +11,10 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
-from testcontainers.core.container import DockerContainer
-from testcontainers.core.waiting_utils import wait_container_is_ready
-ADDITIONAL_TRANSIENT_ERRORS = []
+from .container import DockerContainer
+from .waiting_utils import wait_container_is_ready
 
+ADDITIONAL_TRANSIENT_ERRORS = []
 try:
     from sqlalchemy.exc import DBAPIError
     ADDITIONAL_TRANSIENT_ERRORS.append(DBAPIError)

core/testcontainers/core/waiting_utils.py

@@ -18,9 +18,8 @@
 
 import wrapt
 
-from testcontainers.core import config
-from testcontainers.core.exceptions import TimeoutException
-from testcontainers.core.utils import setup_logger
+from . import config
+from .utils import setup_logger
 
 logger = setup_logger(__name__)
 
@@ -59,7 +58,7 @@ def wrapper(wrapped, instance, args, kwargs):
                              f"failed: {traceback.format_exc()}")
                 time.sleep(config.SLEEP_TIME)
                 exception = e
-        raise TimeoutException(
+        raise TimeoutError(
             f'Wait time ({config.MAX_TRIES * config.SLEEP_TIME}s) exceeded for {wrapped.__name__}'
             f'(args: {args}, kwargs {kwargs}). Exception: {exception}'
         )