python

custom_components/yandex_smart_home/schema/capability_mode.py

@@ -0,0 +1,133 @@
+"""Schema for mode capability.
+
+https://yandex.ru/dev/dialogs/smart-home/doc/concepts/mode.html
+"""
+
+from enum import StrEnum
+from typing import Literal, Self
+
+from .base import APIModel
+
+
+class ModeCapabilityInstance(StrEnum):
+    """Instance of a mode capability.
+
+    https://yandex.ru/dev/dialogs/smart-home/doc/concepts/mode-instance.html
+    """
+
+    CLEANUP_MODE = "cleanup_mode"
+    COFFEE_MODE = "coffee_mode"
+    DISHWASHING = "dishwashing"
+    FAN_SPEED = "fan_speed"
+    HEAT = "heat"
+    INPUT_SOURCE = "input_source"
+    PROGRAM = "program"
+    SWING = "swing"
+    TEA_MODE = "tea_mode"
+    THERMOSTAT = "thermostat"
+    VENTILATION_MODE = "ventilation_mode"
+    WORK_SPEED = "work_speed"
+
+
+class ModeCapabilityMode(StrEnum):
+    """Mode value of a mode capability.
+
+    https://yandex.ru/dev/dialogs/smart-home/doc/concepts/mode-instance-modes.html
+    """
+
+    WET_CLEANING = "wet_cleaning"
+    DRY_CLEANING = "dry_cleaning"
+    MIXED_CLEANING = "mixed_cleaning"
+    AUTO = "auto"
+    ECO = "eco"
+    SMART = "smart"
+    TURBO = "turbo"
+    COOL = "cool"
+    DRY = "dry"
+    FAN_ONLY = "fan_only"
+    HEAT = "heat"
+    PREHEAT = "preheat"
+    HIGH = "high"
+    LOW = "low"
+    MEDIUM = "medium"
+    MAX = "max"
+    MIN = "min"
+    FAST = "fast"
+    SLOW = "slow"
+    EXPRESS = "express"
+    NORMAL = "normal"
+    QUIET = "quiet"
+    HORIZONTAL = "horizontal"
+    STATIONARY = "stationary"
+    VERTICAL = "vertical"
+    SUPPLY_AIR = "supply_air"
+    EXTRACTION_AIR = "extraction_air"
+    ONE = "one"
+    TWO = "two"
+    THREE = "three"
+    FOUR = "four"
+    FIVE = "five"
+    SIX = "six"
+    SEVEN = "seven"
+    EIGHT = "eight"
+    NINE = "nine"
+    TEN = "ten"
+    AMERICANO = "americano"
+    CAPPUCCINO = "cappuccino"
+    DOUBLE = "double"
+    ESPRESSO = "espresso"
+    DOUBLE_ESPRESSO = "double_espresso"
+    LATTE = "latte"
+    BLACK_TEA = "black_tea"
+    FLOWER_TEA = "flower_tea"
+    GREEN_TEA = "green_tea"
+    HERBAL_TEA = "herbal_tea"
+    OOLONG_TEA = "oolong_tea"
+    PUERH_TEA = "puerh_tea"
+    RED_TEA = "red_tea"
+    WHITE_TEA = "white_tea"
+    GLASS = "glass"
+    INTENSIVE = "intensive"
+    PRE_RINSE = "pre_rinse"
+    ASPIC = "aspic"
+    BABY_FOOD = "baby_food"
+    BAKING = "baking"
+    BREAD = "bread"
+    BOILING = "boiling"
+    CEREALS = "cereals"
+    CHEESECAKE = "cheesecake"
+    DEEP_FRYER = "deep_fryer"
+    DESSERT = "dessert"
+    FOWL = "fowl"
+    FRYING = "frying"
+    MACARONI = "macaroni"
+    MILK_PORRIDGE = "milk_porridge"
+    MULTICOOKER = "multicooker"
+    PASTA = "pasta"
+    PILAF = "pilaf"
+    PIZZA = "pizza"
+    SAUCE = "sauce"
+    SLOW_COOK = "slow_cook"
+    SOUP = "soup"
+    STEAM = "steam"
+    STEWING = "stewing"
+    VACUUM = "vacuum"
+    YOGURT = "yogurt"
+
+
+class ModeCapabilityParameters(APIModel):
+    """Parameters of a mode capability."""
+
+    instance: ModeCapabilityInstance
+    modes: list[dict[Literal["value"], ModeCapabilityMode]]
+
+    @classmethod
+    def from_list(cls, instance: ModeCapabilityInstance, modes: list[ModeCapabilityMode]) -> Self:
+        return cls(instance=instance, modes=[{"value": m} for m in modes])
+
+
+class ModeCapabilityInstanceActionState(APIModel):
+    """New value for a mode capability."""
+
+    instance: ModeCapabilityInstance
+    value: ModeCapabilityMode