Skip to content

Commit

Permalink
refactor: adjust imports/types according to prev decisions
Browse files Browse the repository at this point in the history
  • Loading branch information
pedohorse committed Oct 28, 2024
1 parent b7cd8fb commit 97a3486
Show file tree
Hide file tree
Showing 33 changed files with 70 additions and 1,548 deletions.
2 changes: 1 addition & 1 deletion src/lifeblood/basenode_serializer_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from .basenode_serialization import NodeSerializerBase, IncompatibleDeserializationMethod, FailedToApplyNodeState, FailedToApplyParameters
from .basenode import BaseNode
from .enums import NodeParameterType
from .uidata import ParameterFullValue
from .node_parameters import ParameterFullValue

from typing import Optional, Tuple, Union

Expand Down
2 changes: 0 additions & 2 deletions src/lifeblood/core_nodes/environment_resolver_setter.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,6 @@
from lifeblood.processingcontext import ProcessingContext
from lifeblood.enums import NodeParameterType
from lifeblood.environment_resolver import EnvironmentResolverArguments
from lifeblood.uidata import NodeUi, MultiGroupLayout, Parameter
from lifeblood.node_visualization_classes import NodeColorScheme

from typing import Iterable

Expand Down
4 changes: 0 additions & 4 deletions src/lifeblood/core_nodes/mod_attrib.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,6 @@
from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult
from lifeblood.taskspawn import TaskSpawn
from lifeblood.exceptions import NodeNotReadyToProcess
from lifeblood.enums import NodeParameterType
from lifeblood.uidata import NodeUi, MultiGroupLayout, Parameter
from lifeblood.node_visualization_classes import NodeColorScheme

from typing import Iterable

Expand Down
11 changes: 2 additions & 9 deletions src/lifeblood/core_nodes/parent_children_waiter.py
Original file line number Diff line number Diff line change
@@ -1,20 +1,13 @@
import dataclasses
from dataclasses import dataclass
from lifeblood.attribute_serialization import deserialize_attributes_core
from lifeblood.node_plugin_base import BaseNode, ProcessingError
from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult
from lifeblood.taskspawn import TaskSpawn
from lifeblood.exceptions import NodeNotReadyToProcess
from lifeblood.enums import NodeParameterType
from lifeblood.uidata import NodeUi
from lifeblood.processingcontext import ProcessingContext

from threading import Lock

from typing import Any, Dict, Iterable, List, Optional, Set, TypedDict , TYPE_CHECKING

if TYPE_CHECKING:
from lifeblood.scheduler import Scheduler
from typing import Dict, Iterable, List, Set


def node_class():
Expand Down
11 changes: 4 additions & 7 deletions src/lifeblood/core_nodes/python.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,13 @@
import re
import time

from lifeblood.node_plugin_base import BaseNodeWithTaskRequirements
from lifeblood.invocationjob import InvocationJob, InvocationEnvironment
from lifeblood.invocationjob import InvocationJob
from lifeblood.processingcontext import ProcessingContext
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.uidata import NodeParameterType
from lifeblood.enums import NodeParameterType

from typing import Iterable

from types import MappingProxyType
from typing import TYPE_CHECKING, Iterable
if TYPE_CHECKING:
from lifeblood.scheduler import Scheduler


def node_class():
Expand Down
3 changes: 0 additions & 3 deletions src/lifeblood/core_nodes/rename_attrib.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,6 @@
from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.taskspawn import TaskSpawn
from lifeblood.exceptions import NodeNotReadyToProcess
from lifeblood.enums import NodeParameterType
from lifeblood.uidata import NodeUi

from typing import Iterable

Expand Down
8 changes: 1 addition & 7 deletions src/lifeblood/core_nodes/split_waiter.py
Original file line number Diff line number Diff line change
@@ -1,18 +1,12 @@
from dataclasses import dataclass
from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult
from lifeblood.taskspawn import TaskSpawn
from lifeblood.exceptions import NodeNotReadyToProcess
from lifeblood.enums import NodeParameterType
from lifeblood.uidata import NodeUi
from lifeblood.processingcontext import ProcessingContext

from threading import Lock

from typing import Dict, TypedDict, Set, Iterable, Optional, Any, TYPE_CHECKING

if TYPE_CHECKING:
from lifeblood.scheduler import Scheduler
from typing import Dict, Set, Iterable, Optional


@dataclass
Expand Down
4 changes: 1 addition & 3 deletions src/lifeblood/core_nodes/switch.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.nodethings import ProcessingResult
from lifeblood.processingcontext import ProcessingContext
from lifeblood.enums import NodeParameterType
from lifeblood.uidata import NodeUi, Parameter, VerticalParametersLayout, ParameterHierarchyItem, ParametersLayoutBase
from lifeblood.node_visualization_classes import NodeColorScheme

from typing import Iterable

Expand Down
6 changes: 2 additions & 4 deletions src/lifeblood/core_nodes/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,9 @@
from lifeblood.node_plugin_base import BaseNode
from lifeblood.invocationjob import InvocationJob, InvocationEnvironment
from lifeblood.nodethings import ProcessingResult
from lifeblood.uidata import NodeParameterType
from lifeblood.enums import NodeParameterType

from typing import TYPE_CHECKING, Iterable
if TYPE_CHECKING:
from lifeblood.scheduler import Scheduler
from typing import Iterable


def node_class():
Expand Down
3 changes: 0 additions & 3 deletions src/lifeblood/core_nodes/wait_for_task.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,8 @@
import shlex
from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult, ProcessingContext
from lifeblood.taskspawn import TaskSpawn
from lifeblood.exceptions import NodeNotReadyToProcess
from lifeblood.enums import NodeParameterType
from lifeblood.uidata import NodeUi, MultiGroupLayout, Parameter
from lifeblood.node_visualization_classes import NodeColorScheme

from typing import Dict, Iterable, List, Optional, Set

Expand Down
1 change: 0 additions & 1 deletion src/lifeblood/core_nodes/wedge.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.enums import NodeParameterType
from lifeblood.uidata import NodeUi, MultiGroupLayout, Parameter

from typing import Iterable

Expand Down
4 changes: 2 additions & 2 deletions src/lifeblood/node_plugin_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from .invocationjob import ResourceRequirement, ResourceRequirements
from .nodethings import ProcessingResult, ProcessingError # unused import - for easy reexport to plugins
from .worker_resource_definition import WorkerResourceDefinition, WorkerResourceDataType, WorkerDeviceTypeDefinition
from .uidata import NodeUi
from .node_ui import NodeUi
from .scheduler.scheduler import Scheduler

from typing import Dict, Optional, Tuple, Union
Expand Down Expand Up @@ -142,7 +142,7 @@ def set_parent(self, graph_holder: NodeGraphHolderBase, node_id_in_graph: int):

def __apply_requirements(self, task_dict: dict, node_config: dict, result: ProcessingResult):
if result.invocation_job is not None:
context = ProcessingContext(self, task_dict, node_config)
context = ProcessingContext(self.name(), self.label(), self.get_ui(), task_dict, node_config)
raw_groups = context.param_value('__requirements__.worker_groups').strip()
reqs = result.invocation_job.requirements()
if raw_groups != '':
Expand Down
2 changes: 1 addition & 1 deletion src/lifeblood/node_type_metadata.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from .node_dataprovider_base import NodeDataProvider
from .plugin_info import PluginInfo

from typing import Optional, TYPE_CHECKING, Tuple, Set
from typing import Optional, Tuple, Set


class NodeTypePluginMetadata:
Expand Down
3 changes: 1 addition & 2 deletions src/lifeblood/pulse_checker.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import asyncio
from .logging import get_logger
#from .scheduler_task_protocol import SchedulerTaskClient
from .scheduler_message_processor import SchedulerWorkerControlClient
from .scheduler_message_processor_client import SchedulerWorkerControlClient
from .net_messages.address import AddressChain
from .net_messages.message_processor import MessageProcessorBase
from .net_messages.exceptions import MessageTransferError
Expand Down
3 changes: 1 addition & 2 deletions src/lifeblood/stock_nodes/ffmpeg.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,8 @@

from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.uidata import NodeParameterType
from lifeblood.enums import NodeParameterType
from lifeblood.invocationjob import InvocationJob
from lifeblood.invocationjob import InvocationJob, InvocationEnvironment

from typing import Iterable

Expand Down
3 changes: 1 addition & 2 deletions src/lifeblood/stock_nodes/file_watcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,9 @@

from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.uidata import NodeParameterType
from lifeblood.processingcontext import ProcessingContext
from lifeblood.invocationjob import InvocationJob, InvocationRequirements
from lifeblood.enums import WorkerType
from lifeblood.enums import WorkerType, NodeParameterType

from typing import Iterable

Expand Down
2 changes: 1 addition & 1 deletion src/lifeblood/stock_nodes/fileop.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.invocationjob import InvocationJob
from lifeblood.processingcontext import ProcessingContext
from lifeblood.uidata import NodeParameterType
from lifeblood.enums import NodeParameterType

from typing import Iterable

Expand Down
2 changes: 1 addition & 1 deletion src/lifeblood/stock_nodes/filepattern.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.invocationjob import InvocationJob
from lifeblood.processingcontext import ProcessingContext
from lifeblood.uidata import NodeParameterType
from lifeblood.enums import NodeParameterType
from lifeblood.enums import WorkerType
from lifeblood.text import match_pattern

Expand Down
4 changes: 1 addition & 3 deletions src/lifeblood/stock_nodes/imagemagik.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,11 @@
import os
import shlex
import re
from math import sqrt, floor, ceil

from lifeblood.node_plugin_base import BaseNode
from lifeblood.nodethings import ProcessingResult, ProcessingError
from lifeblood.uidata import NodeParameterType
from lifeblood.enums import NodeParameterType
from lifeblood.invocationjob import InvocationJob
from lifeblood.invocationjob import InvocationJob, InvocationEnvironment

from typing import Iterable

Expand Down
Loading

0 comments on commit 97a3486

Please sign in to comment.