isolate 0.12.13__py3-none-any.whl → 0.12.15__py3-none-any.whl

isolate/_isolate_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.12.13'
-__version_tuple__ = version_tuple = (0, 12, 13)
+__version__ = version = '0.12.15'
+__version_tuple__ = version_tuple = (0, 12, 15)

isolate/backends/_base.py

@@ -12,7 +12,6 @@ from typing import (
 )
 
 from isolate.backends.settings import DEFAULT_SETTINGS, IsolateSettings
-from isolate.exceptions import IsolateException
 from isolate.logs import Log, LogLevel, LogSource
 
 __all__ = [
@@ -28,7 +27,7 @@ CallResultType = TypeVar("CallResultType")
 BasicCallable = Callable[[], CallResultType]
 
 
-class EnvironmentCreationError(IsolateException):
+class EnvironmentCreationError(Exception):
     """Raised when the environment cannot be created."""
 
 

isolate/backends/container.py

@@ -0,0 +1,49 @@
+from __future__ import annotations
+
+import sys
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Any, ClassVar
+
+from isolate.backends import BaseEnvironment
+from isolate.backends.common import sha256_digest_of
+from isolate.backends.settings import DEFAULT_SETTINGS, IsolateSettings
+from isolate.connections import PythonIPC
+
+
+@dataclass
+class ContainerizedPythonEnvironment(BaseEnvironment[Path]):
+    BACKEND_NAME: ClassVar[str] = "container"
+
+    image: dict[str, Any] = field(default_factory=dict)
+    python_version: str | None = None
+    tags: list[str] = field(default_factory=list)
+
+    @classmethod
+    def from_config(
+        cls,
+        config: dict[str, Any],
+        settings: IsolateSettings = DEFAULT_SETTINGS,
+    ) -> BaseEnvironment:
+        environment = cls(**config)
+        environment.apply_settings(settings)
+        return environment
+
+    @property
+    def key(self) -> str:
+        # dockerfile_str is always there, but the validation is handled by the
+        # controller.
+        dockerfile_str = self.image.get("dockerfile_str", "")
+        return sha256_digest_of(dockerfile_str, *sorted(self.tags))
+
+    def create(self, *, force: bool = False) -> Path:
+        return Path(sys.exec_prefix)
+
+    def destroy(self, connection_key: Path) -> None:
+        raise NotImplementedError("ContainerizedPythonEnvironment cannot be destroyed")
+
+    def exists(self) -> bool:
+        return True
+
+    def open_connection(self, connection_key: Path) -> PythonIPC:
+        return PythonIPC(self, connection_key)

isolate/connections/common.py

@@ -2,13 +2,12 @@ from __future__ import annotations
 
 import importlib
 import os
+from contextlib import contextmanager
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any, cast
+from typing import TYPE_CHECKING, Any, Iterator, cast
 
 from tblib import Traceback, TracebackParseError
 
-from isolate.exceptions import IsolateException
-
 if TYPE_CHECKING:
     from typing import Protocol
 
@@ -22,18 +21,23 @@ AGENT_SIGNATURE = "IS_ISOLATE_AGENT"
 
 
 @dataclass
-class BaseSerializationError(IsolateException):
+class SerializationError(Exception):
     """An error that happened during the serialization process."""
 
     message: str
 
 
-class SerializationError(BaseSerializationError):
-    pass
-
+@contextmanager
+def _step(message: str) -> Iterator[None]:
+    """A context manager to capture every expression
+    underneath it and if any of them fails for any reason
+    then it will raise a SerializationError with the
+    given message."""
 
-class DeserializationError(BaseSerializationError):
-    pass
+    try:
+        yield
+    except BaseException as exception:
+        raise SerializationError("Error while " + message) from exception
 
 
 def as_serialization_method(backend: Any) -> SerializationBackend:
@@ -62,22 +66,13 @@ def load_serialized_object(
     flag is set to true, then the given object will be raised as an exception (instead
     of being returned)."""
 
-    try:
+    with _step(f"preparing the serialization backend ({serialization_method})"):
         serialization_backend = as_serialization_method(
             importlib.import_module(serialization_method)
         )
-    except BaseException as exc:
-        raise DeserializationError(
-            "Error while preparing the serialization backend "
-            f"({serialization_method})"
-        ) from exc
 
-    try:
+    with _step("deserializing the given object"):
         result = serialization_backend.loads(raw_object)
-    except BaseException as exc:
-        raise DeserializationError(
-            "Error while deserializing the given object"
-        ) from exc
 
     if was_it_raised:
         raise prepare_exc(result, stringized_traceback=stringized_traceback)
@@ -89,19 +84,13 @@ def serialize_object(serialization_method: str, object: Any) -> bytes:
     """Serialize the given object using the given serialization method. If
     anything fails, then a SerializationError will be raised."""
 
-    try:
+    with _step(f"preparing the serialization backend ({serialization_method})"):
         serialization_backend = as_serialization_method(
             importlib.import_module(serialization_method)
         )
-    except BaseException as exc:
-        raise SerializationError(
-            f"Error while preparing the serialization backend ({serialization_method})"
-        ) from exc
 
-    try:
+    with _step("serializing the given object"):
         return serialization_backend.dumps(object)
-    except BaseException as exc:
-        raise SerializationError("Error while serializing the given object") from exc
 
 
 def is_agent() -> bool:

isolate/connections/grpc/_base.py

@@ -16,11 +16,10 @@ from isolate.connections.common import serialize_object
 from isolate.connections.grpc import agent, definitions
 from isolate.connections.grpc.configuration import get_default_options
 from isolate.connections.grpc.interface import from_grpc
-from isolate.exceptions import IsolateException
 from isolate.logs import LogLevel, LogSource
 
 
-class AgentError(IsolateException):
+class AgentError(Exception):
     """An internal problem caused by (most probably) the agent."""
 
 

isolate/connections/grpc/agent.py

@@ -22,12 +22,11 @@ from isolate.connections.common import SerializationError, serialize_object
 from isolate.connections.grpc import definitions
 from isolate.connections.grpc.configuration import get_default_options
 from isolate.connections.grpc.interface import from_grpc, to_grpc
-from isolate.exceptions import IsolateException
 from isolate.logs import Log, LogLevel, LogSource
 
 
 @dataclass
-class AbortException(IsolateException):
+class AbortException(Exception):
     message: str
 
 
@@ -107,11 +106,14 @@ class AgentServicer(definitions.AgentServicer):
             )
 
         try:
-            # NOTE: technically any sort of exception could be raised here, since
+            # TODO: technically any sort of exception could be raised here, since
             # depickling is basically involves code execution from the *user*.
             function = from_grpc(function)
-        except BaseException as exc:
-            return exc, True, traceback.format_exc()
+        except SerializationError:
+            yield from self.log(traceback.format_exc(), level=LogLevel.ERROR)
+            raise AbortException(
+                f"The {function_kind} function could not be deserialized."
+            )
 
         if not callable(function):
             raise AbortException(

{isolate-0.12.13.dist-info → isolate-0.12.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: isolate
-Version: 0.12.13
+Version: 0.12.15
 Summary: Managed isolated environments for Python
 Author-email: Features & Labels <hello@fal.ai>
 Project-URL: Issues, https://github.com/fal-ai/isolate/issues

{isolate-0.12.13.dist-info → isolate-0.12.15.dist-info}/RECORD

@@ -1,14 +1,14 @@
 isolate/__init__.py,sha256=uXOKnONs7sXgARNgElwr4_A1sKoA6ACHVEvs3IDiX1M,127
-isolate/_isolate_version.py,sha256=rYB4wxX5D44-YwCyT1hUwPhja5ktD6OYXizpLgC8gMI,415
+isolate/_isolate_version.py,sha256=VMxsT9HmXnFACz1Qv_ryjpdx94Iy2Tkd2O8BMdpck7w,415
 isolate/_version.py,sha256=05pXvy-yr5t3I1m9JMn42Ilzpg7fa8IB2J8a3G7t1cU,274
-isolate/exceptions.py,sha256=ki-f7khY-Yo1J8nPdy5IP-aZhQyLUP9pp3dE0moEVKw,44
 isolate/logs.py,sha256=R_AHUVYD18z_PhtK_mDWi9Gch79CxmwHY09hUDShtwg,2079
 isolate/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 isolate/registry.py,sha256=hpzv4HI7iihG5I7i5r8Pb257ibhEKY18xQcG-w1-BgI,1590
 isolate/backends/__init__.py,sha256=LLrSM7UnDFW8tIT5oYlE1wVJrxKcaj_v7cFwvTjQTlc,119
-isolate/backends/_base.py,sha256=senNj-k2Dp2DkyxmZy7WxzsexmPfU6_DBo7bbZDqTLQ,4167
+isolate/backends/_base.py,sha256=Kt5pkhDzXZblq4vxaM3DQTo-Bj-7pIRZFlqJR7OfejY,4112
 isolate/backends/common.py,sha256=ZiU0Vkz78qaPH_3ThV28OcHi8QgXie4aBrkPXCJuNBM,8321
 isolate/backends/conda.py,sha256=OIQrpt_VffW2PjPOIzp-JvonW4e7rDQ1ASHOEjyzD8E,7646
+isolate/backends/container.py,sha256=Ek06tMwrCzEGFUlIQ8lIWURiwchTiuiS4HHDk-9bN9k,1567
 isolate/backends/local.py,sha256=woxe4dmXuEHxWKsGNndoRA1_sP6yG-dg6tlFZni0mZc,1360
 isolate/backends/pyenv.py,sha256=G-OIwESUOU5TqqumwsKVUhRiFQzxB1xrPn-bGm4LQoI,5428
 isolate/backends/remote.py,sha256=qUm54mpqk0kaEfbPZl962Td3_P3qcpyVcfGdKfmkJHs,4234
@@ -17,13 +17,13 @@ isolate/backends/virtualenv.py,sha256=DiBAiybOLWMxJobIRaH4AgDn9agY5VrVPaSQObQa1L
 isolate/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 isolate/common/timestamp.py,sha256=seh7FrMRH4i1SCQavA8d-7z8qi0pP8lYYhd29gTPMwE,367
 isolate/connections/__init__.py,sha256=oa0PNo7ZQ0StPIDvKnJ02_CNVMyfOhxJ3M1C0VMvj9c,627
-isolate/connections/common.py,sha256=bmQ7qDtBIbN7wj55OUwcRtRr9fWDSuxYajY1BRnb8Xc,3655
+isolate/connections/common.py,sha256=PAfBGKZNUdtFlZQlw3_nQaUCKQXTnEkxzNNRV_i4R2A,3498
 isolate/connections/_local/__init__.py,sha256=6FtCKRSFBvTvjm5LNlNA-mieKEq3J7DZZRPcXVedERo,146
 isolate/connections/_local/_base.py,sha256=qCx2M8kbxuPTruj9kH5z005LU2FaC0BkLxsgY-sXRlg,6214
 isolate/connections/_local/agent_startup.py,sha256=swCs6Q0yVkDw7w-RftizHSMyJDM7DQwuP3TB0qI1ucg,1552
 isolate/connections/grpc/__init__.py,sha256=tcesLxlC36P6wSg2lBcO2egsJWMbSKwc8zFXhWac3YU,85
-isolate/connections/grpc/_base.py,sha256=hfGy_kFjifpK3zHD0_OTqpTF6CTOZH6abNnET4XLwE8,5613
-isolate/connections/grpc/agent.py,sha256=xRBfbmsWlOU4U6pBl0g_fhMdRZgVzXoKmi4gKTM3dYQ,7747
+isolate/connections/grpc/_base.py,sha256=8C1qnEZghPk4GYBlh0rjbtlx8FYB7gNvioG9re1OMD8,5558
+isolate/connections/grpc/agent.py,sha256=Vne6wsu1mQQm_jZwqLbWjrtfi7DjHce0TDPzjDrN1lc,7838
 isolate/connections/grpc/configuration.py,sha256=50YvGGHA9uyKg74xU_gc73j7bsFk973uIpMhmw2HhxY,788
 isolate/connections/grpc/interface.py,sha256=yt63kytgXRXrTnjePGJVdXz4LJJVSSrNkJCF1yz6FIE,2270
 isolate/connections/grpc/definitions/__init__.py,sha256=Z0453Bbjoq-Oxm2Wfi9fae-BFf8YsZwmuh88strmvxo,459
@@ -52,9 +52,9 @@ isolate/server/health/health.proto,sha256=wE2_QD0OQAblKkEBG7sALLXEOj1mOLKG-FbC4t
 isolate/server/health/health_pb2.py,sha256=mCnDq0-frAddHopN_g_LueHddbW-sN5kOfntJDlAvUY,1783
 isolate/server/health/health_pb2.pyi,sha256=boMRHMlX770EuccQCFTeRgf_KA_VMgW7l9GZIwxvMok,2546
 isolate/server/health/health_pb2_grpc.py,sha256=JRluct2W4af83OYxwmcCn0vRc78zf04Num0vBApuPEo,4005
-isolate-0.12.13.dist-info/LICENSE,sha256=427vuyirL5scgBLqA9UWcdnxKrtSGc0u_JfUupk6lAA,11359
-isolate-0.12.13.dist-info/METADATA,sha256=OKOL9Y8v5rY-Fs9zx5HwIDSdpZjRoyyseprthtrEdQs,3154
-isolate-0.12.13.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-isolate-0.12.13.dist-info/entry_points.txt,sha256=XQ_nl-8MR94UnekxbBJRNGlY-lZ_Qh50N4mzwFDdwV8,290
-isolate-0.12.13.dist-info/top_level.txt,sha256=W9QJBHcq5WXRkbOXf25bvftzFsOZZN4n1DAatdroZrs,8
-isolate-0.12.13.dist-info/RECORD,,
+isolate-0.12.15.dist-info/LICENSE,sha256=427vuyirL5scgBLqA9UWcdnxKrtSGc0u_JfUupk6lAA,11359
+isolate-0.12.15.dist-info/METADATA,sha256=9wG2gpvyozTg6s7Z_km_4SQMMscmcX6Dn3jZqVA3Oe4,3154
+isolate-0.12.15.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+isolate-0.12.15.dist-info/entry_points.txt,sha256=s3prh2EERaVCbL8R45tfY5WFPZ1TsYOsz305YR7s-Pc,360
+isolate-0.12.15.dist-info/top_level.txt,sha256=W9QJBHcq5WXRkbOXf25bvftzFsOZZN4n1DAatdroZrs,8
+isolate-0.12.15.dist-info/RECORD,,

{isolate-0.12.13.dist-info → isolate-0.12.15.dist-info}/entry_points.txt

@@ -1,5 +1,6 @@
 [isolate.backends]
 conda = isolate.backends.conda:CondaEnvironment
+container = isolate.backends.container:ContainerizedPythonEnvironment
 isolate-server = isolate.backends.remote:IsolateServer
 local = isolate.backends.local:LocalPythonEnvironment
 pyenv = isolate.backends.pyenv:PyenvEnvironment

isolate/exceptions.py

@@ -1,2 +0,0 @@
-class IsolateException(Exception):
-    pass