Deabstract savable interface

docs/source/nitpick-exceptions

@@ -18,7 +18,7 @@ py:class kiwipy.communications.Communicator
 
 # unavailable forward references
 py:class plumpy.process_states.Command
-py:class plumpy.process_states.State
+py:class plumpy.state_machine.State
 py:class plumpy.base.state_machine.State
 py:class State
 py:class Process

pyproject.toml

@@ -32,6 +32,7 @@ dependencies = [
     'kiwipy[rmq]~=0.8.5',
     'nest_asyncio~=1.5,>=1.5.1',
     'pyyaml~=6.0',
+    'typing-extensions~=4.12'
 ]
 
 [project.urls]

src/plumpy/__init__.py

@@ -1,42 +1,144 @@
 # -*- coding: utf-8 -*-
-# mypy: disable-error-code=name-defined
 __version__ = '0.24.0'
 
 import logging
 
+from .base.state_machine import TransitionFailed
+
 # interfaces
 from .controller import ProcessController
 from .coordinator import Coordinator
-from .events import *
-from .exceptions import *
-from .futures import *
-from .loaders import *
-from .message import *
-from .mixins import *
-from .persistence import *
-from .ports import *
-from .process_listener import *
-from .process_states import *
-from .processes import *
-from .rmq import *
-from .utils import *
-from .workchains import *
+from .events import (
+    PlumpyEventLoopPolicy,
+    get_event_loop,
+    new_event_loop,
+    reset_event_loop_policy,
+    run_until_complete,
+    set_event_loop,
+    set_event_loop_policy,
+)
+from .exceptions import (
+    ClosedError,
+    CoordinatorConnectionError,
+    CoordinatorTimeoutError,
+    InvalidStateError,
+    KilledError,
+    PersistenceError,
+    UnsuccessfulResult,
+)
+from .futures import CancellableAction, Future, capture_exceptions
+from .loaders import DefaultObjectLoader, ObjectLoader, get_object_loader, set_object_loader
+from .message import MsgContinue, MsgCreate, MsgKill, MsgLaunch, MsgPause, MsgPlay, MsgStatus, ProcessLauncher
+from .persistence import (
+    Bundle,
+    InMemoryPersister,
+    LoadSaveContext,
+    PersistedCheckpoint,
+    Persister,
+    PicklePersister,
+    Savable,
+    SavableFuture,
+    auto_persist,
+)
+from .ports import UNSPECIFIED, InputPort, OutputPort, Port, PortNamespace, PortValidationError
+from .process_listener import ProcessListener
+from .process_spec import ProcessSpec
+from .process_states import (
+    Continue,
+    Created,
+    Excepted,
+    Finished,
+    Interruption,
+    Kill,
+    Killed,
+    KillInterruption,
+    PauseInterruption,
+    ProcessState,
+    Running,
+    Stop,
+    Wait,
+    Waiting,
+)
+from .processes import BundleKeys, Process
+from .utils import AttributesDict
+from .workchains import ToContext, WorkChain, WorkChainSpec, if_, return_, while_
 
 __all__ = (
-    events.__all__
-    + exceptions.__all__
-    + processes.__all__
-    + utils.__all__
-    + futures.__all__
-    + mixins.__all__
-    + persistence.__all__
-    + message.__all__
-    + process_listener.__all__
-    + workchains.__all__
-    + loaders.__all__
-    + ports.__all__
-    + process_states.__all__
-) + ['ProcessController', 'Coordinator']
+    'UNSPECIFIED',
+    'AttributesDict',
+    'Bundle',
+    'BundleKeys',
+    'CancellableAction',
+    'ClosedError',
+    'Continue',
+    'Coordinator',
+    'CoordinatorConnectionError',
+    'CoordinatorTimeoutError',
+    'Created',
+    'DefaultObjectLoader',
+    'Excepted',
+    'Finished',
+    'Future',
+    'InMemoryPersister',
+    'InputPort',
+    'Interruption',
+    'InvalidStateError',
+    'Kill',
+    'KillInterruption',
+    'Killed',
+    'KilledError',
+    'LoadSaveContext',
+    'MsgContinue',
+    'MsgCreate',
+    'MsgKill',
+    'MsgLaunch',
+    'MsgPause',
+    'MsgPlay',
+    'MsgStatus',
+    'ObjectLoader',
+    'OutputPort',
+    'PauseInterruption',
+    'PersistedCheckpoint',
+    'PersistenceError',
+    'Persister',
+    'PicklePersister',
+    'PlumpyEventLoopPolicy',
+    'Port',
+    'PortNamespace',
+    'PortValidationError',
+    'Process',
+    'ProcessController',
+    'ProcessLauncher',
+    'ProcessListener',
+    'ProcessSpec',
+    'ProcessState',
+    'Running',
+    'Savable',
+    'SavableFuture',
+    'Stop',
+    'ToContext',
+    'TransitionFailed',
+    'UnsuccessfulResult',
+    'Wait',
+    'Waiting',
+    'WorkChain',
+    'WorkChainSpec',
+    'auto_persist',
+    'capture_exceptions',
+    'create_continue_body',
+    'create_launch_body',
+    'get_event_loop',
+    'get_object_loader',
+    'if_',
+    'new_event_loop',
+    'reset_event_loop_policy',
+    'return_',
+    'run_until_complete',
+    'set_event_loop',
+    'set_event_loop_policy',
+    'set_object_loader',
+    'while_',
+)
 
 
 # Do this se we don't get the "No handlers could be found..." warnings that will be produced