Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -2,18 +2,18 @@

from dataclasses import dataclass
from datetime import datetime
from enum import Enum
from enum import StrEnum


class ComponentStatus(str, Enum):
class ComponentStatus(StrEnum):
"""Status of configuration component."""

MISSING = 'missing'
INCOMPLETE = 'incomplete'
COMPLETE = 'complete'


class ConfigurationStatus(str, Enum):
class ConfigurationStatus(StrEnum):
"""Overall configuration status."""

DRAFT = 'draft'
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from collections.abc import Sequence
from datetime import datetime
from datetime import timedelta
from enum import Enum
from enum import StrEnum
import logging
from typing import Any

Expand All @@ -23,7 +23,7 @@
logger = logging.getLogger(__name__)


class TrackSelectionStrategy(str, Enum):
class TrackSelectionStrategy(StrEnum):
"""Strategy for selecting tracks."""

ROUND_ROBIN = 'round_robin'
Expand All @@ -32,14 +32,14 @@ class TrackSelectionStrategy(str, Enum):
RANDOM = 'random'


class LocoDeliveryStrategy(str, Enum):
class LocoDeliveryStrategy(StrEnum):
"""Strategy for locomotive delivery."""

RETURN_TO_PARKING = 'return_to_parking'
DIRECT_DELIVERY = 'direct_delivery'


class LocoPriorityStrategy(str, Enum):
class LocoPriorityStrategy(StrEnum):
"""Strategy for locomotive task prioritization."""

WORKSHOP_PRIORITY = 'workshop_priority' # Park wagons immediately when loco available
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -296,7 +296,7 @@ def decouple_wagons(
wagon_count: int,
coupler_type: str | None = None,
wagon_ids: list[str] | None = None,
) -> Any: # ruff: C901
) -> Any:
"""Decouple wagons from locomotive."""

def decouple_gen() -> Generator:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -107,8 +107,8 @@ def test_select_best_retrofit_track_multiple_tracks(
track2.id = 'retrofit_2'

mock_railway_context.get_track_selection_service().get_tracks_by_type.return_value = [track1, track2]
mock_railway_context.get_available_capacity.side_effect = (
lambda track_id: 20.0 if track_id == 'retrofit_1' else 50.0
mock_railway_context.get_available_capacity.side_effect = lambda track_id: (
20.0 if track_id == 'retrofit_1' else 50.0
)

best_track = planner._select_best_retrofit_track()
Expand Down
Loading