solax-py-library 1.0.0.24__py3-none-any.whl → 1.0.0.26__py3-none-any.whl

solax_py_library/smart_scene/types/condition.py

@@ -0,0 +1,299 @@
+import operator
+from enum import IntEnum, Enum
+from typing import Optional, List, Union, Any
+
+from pydantic import BaseModel, validator, root_validator
+
+from solax_py_library.device.types.alarm import AlarmLevel
+from solax_py_library.smart_scene.constant.message_entry import MESSAGE_ENTRY
+
+
+class LogicFunc(IntEnum):
+    OR = 0
+    AND = 1
+
+
+class ConditionFunc(IntEnum):
+    GT = 100
+    LT = 101
+    EQ = 102
+
+    def function(self):
+        return {
+            ConditionFunc.GT: operator.gt,
+            ConditionFunc.LT: operator.lt,
+            ConditionFunc.EQ: operator.eq,
+        }.get(self)
+
+
+class RepeatFunc(IntEnum):
+    ONCE = 103
+    EVERYDAY = 104
+    WEEKDAY = 105
+    WEEKEND = 106
+    CUSTOM = 107
+
+
+class ConditionType(str, Enum):
+    date = "date"
+    weather = "weather"
+    buyingPrice = "buyingPrice"
+    sellingPrice = "sellingPrice"
+    systemCondition = "systemCondition"
+    cabinet = "cabinet"
+
+
+class WeatherConditionType(str, Enum):
+    irradiance = "irradiance"
+    temperature = "temperature"
+
+
+class PriceConditionType(str, Enum):
+    price = "price"
+    lowerPrice = "lowerPrice"
+    higherPrice = "higherPrice"
+    expensiveHours = "expensiveHours"
+    cheapestHours = "cheapestHours"
+
+
+class DateConditionType(str, Enum):
+    time = "time"
+    duration = "duration"
+
+
+class SystemConditionType(str, Enum):
+    systemSoc = "systemSoc"
+    systemImportPower = "systemImportPower"  # 买电功率
+    systemExportPower = "systemExportPower"  # 馈电功率
+
+
+class CabinetConditionType(str, Enum):
+    cabinetAlarm = "cabinetAlarm"
+    cabinetSoc = "cabinetSoc"
+
+
+class SmartSceneUnit(IntEnum):
+    PERCENT = 1
+    NUM = 2
+
+
+class ConditionItemChildData(BaseModel):
+    function: Optional[ConditionFunc]
+    data: List[Any]
+
+
+class PriceConditionItemData(BaseModel):
+    childType: PriceConditionType
+    childData: ConditionItemChildData
+
+    @validator("childData", always=True)
+    def _check_child_data(cls, value, values):
+        child_type = values.get("childType")
+        if child_type in {
+            PriceConditionType.lowerPrice,
+            PriceConditionType.higherPrice,
+        }:
+            assert value.data[0] > 0, ValueError
+        elif child_type in {
+            PriceConditionType.expensiveHours,
+            PriceConditionType.cheapestHours,
+        }:
+            assert 1 <= value.data[2] <= 24, ValueError
+        return value
+
+    def to_text(self, lang, unit):
+        data = self.childData.data
+        func = self.childData.function
+        if self.childType == PriceConditionType.price:
+            return MESSAGE_ENTRY[self.childType][lang].format(
+                MESSAGE_ENTRY[str(func.value)][lang], data[0], unit
+            )
+        elif self.childType in {
+            PriceConditionType.lowerPrice,
+            PriceConditionType.higherPrice,
+        }:
+            return MESSAGE_ENTRY[self.childType][lang].format(
+                data[0], "%" if data[1] == 1 else unit
+            )
+        elif self.childType in {
+            PriceConditionType.expensiveHours,
+            PriceConditionType.cheapestHours,
+        }:
+            return MESSAGE_ENTRY[self.childType][lang].format(data[0], data[1], data[2])
+
+
+class SystemConditionItemData(BaseModel):
+    childType: SystemConditionType
+    childData: ConditionItemChildData
+
+    @validator("childData", always=True)
+    def _check_child_data(cls, value, values):
+        child_type = values.get("childType")
+        if child_type in {
+            SystemConditionType.systemExportPower,
+            SystemConditionType.systemImportPower,
+        }:
+            assert 0 <= value.data[0] <= 100000, ValueError
+            value.data[0] = round(value.data[0], 2)  # 功率保留两位小数
+        elif child_type == SystemConditionType.systemSoc:
+            assert 5 <= value.data[0] <= 100, ValueError
+        return value
+
+    def to_text(self, lang, unit):
+        data = self.childData.data
+        func = self.childData.function
+        if self.childType == SystemConditionType.systemSoc:
+            return MESSAGE_ENTRY[self.childType][lang].format(
+                MESSAGE_ENTRY[str(func.value)][lang], data[0]
+            )
+        elif self.childType in {
+            SystemConditionType.systemImportPower,
+            SystemConditionType.systemExportPower,
+        }:
+            return MESSAGE_ENTRY[self.childType][lang].format(
+                MESSAGE_ENTRY[str(func.value)][lang], data[0]
+            )
+
+
+class CabinetConditionItemData(BaseModel):
+    childType: CabinetConditionType
+    childData: ConditionItemChildData
+
+    @validator("childData", always=True)
+    def _check_child_data(cls, value, values):
+        child_type = values.get("childType")
+        if child_type == CabinetConditionType.cabinetAlarm:
+            assert value.data[0] in {
+                AlarmLevel.TIPS,
+                AlarmLevel.NORMAL,
+                AlarmLevel.EMERGENCY,
+            }, ValueError
+        if child_type == CabinetConditionType.cabinetSoc:
+            assert 0 <= value.data[0] <= 100, ValueError
+        return value
+
+    def to_text(self, lang, unit):
+        data = self.childData.data
+        func = self.childData.function
+        if self.childType == CabinetConditionType.cabinetSoc:
+            return MESSAGE_ENTRY[self.childType][lang].format(
+                MESSAGE_ENTRY[str(func.value)][lang], data[0]
+            )
+        elif self.childType == CabinetConditionType.cabinetAlarm:
+            return MESSAGE_ENTRY[self.childType][lang].format(
+                MESSAGE_ENTRY[str(AlarmLevel(data[0]))][lang], lang
+            )
+
+
+class DateConditionItemData(BaseModel):
+    childType: DateConditionType
+    childData: ConditionItemChildData
+
+    @validator("childData", always=True)
+    def check_param(cls, value, values):
+        child_type = values.get("childType")
+        data = value.data
+        if child_type == DateConditionType.time:
+            assert isinstance(data[0], str), ValueError
+        elif child_type == DateConditionType.duration:
+            assert isinstance(data[0], int), ValueError
+        return value
+
+    def to_text(self, lang, unit):
+        if self.childType == DateConditionType.duration:
+            return MESSAGE_ENTRY[self.childType][lang].format(self.childData.data[0])
+        elif self.childType == DateConditionType.time:
+            return MESSAGE_ENTRY[self.childType][lang] + "/" + self.childData.data[0]
+
+
+class WeatherConditionItemData(BaseModel):
+    childType: WeatherConditionType
+    childData: ConditionItemChildData
+
+    @validator("childData", always=True)
+    def _check_child_data(cls, value, values):
+        child_type = values.get("childType")
+        if child_type == WeatherConditionType.irradiance:
+            assert value.data[0] > 0, ValueError
+            assert 0 <= value.data[1] <= 24, ValueError
+        return value
+
+    def to_text(self, lang, unit):
+        func = self.childData.function
+        data = self.childData.data
+        if self.childType == WeatherConditionType.irradiance:
+            return MESSAGE_ENTRY[self.childType][lang].format(
+                MESSAGE_ENTRY[str(func.value)][lang], data[0], data[1]
+            )
+        else:
+            return MESSAGE_ENTRY[self.childType][lang].format(
+                MESSAGE_ENTRY[str(func.value)][lang], data[0]
+            )
+
+
+class ConditionItem(BaseModel):
+    type: ConditionType
+    cabinet: Optional[List[str]]
+    data: List[
+        Union[
+            DateConditionItemData,
+            WeatherConditionItemData,
+            PriceConditionItemData,
+            SystemConditionItemData,
+            CabinetConditionItemData,
+        ]
+    ]
+
+    @validator("cabinet")
+    def _check_cabinet(cls, value, values):
+        condition_type = values.get("type")
+        if condition_type == ConditionType.cabinet:
+            assert value, "cabinet is None"
+        return value
+
+    def to_text(self, lang, unit):
+        if self.type != ConditionType.cabinet:
+            return {self.type: [d.to_text(lang, unit) for d in self.data]}
+        elif self.type == ConditionType.cabinet:
+            cabinet_sns = ",".join(self.cabinet)
+            return {
+                self.type: [d.to_text(lang, unit) for d in self.data] + [cabinet_sns]
+            }
+
+
+class SmartSceneCondition(BaseModel):
+    operation: LogicFunc
+    value: List[ConditionItem]
+
+    @root_validator
+    def _root_check(cls, values):
+        if values.get("operation") == LogicFunc.OR:
+            new_value = []
+            for item in values.get("value"):
+                if item.type != ConditionType.date:
+                    new_value.append(item)
+                    continue
+                new_data = []
+                for date_item in item.data:
+                    if date_item.childType != DateConditionType.duration:
+                        new_data.append(date_item)
+                item.data = new_data
+                if item.data:
+                    new_value.append(item)
+            values["value"] = new_value
+        return values
+
+    def to_text(self, lang, unit):
+        ret = {"operation": [MESSAGE_ENTRY[self.operation.name][lang]]}
+        for v in self.value:
+            ret.update(v.to_text(lang, unit))
+        return ret
+
+    def get_duration_info(self):
+        for item in self.value:
+            if item.type != ConditionType.date:
+                continue
+            for date_item in item.data:
+                if date_item.childType == DateConditionType.duration:
+                    return (date_item.childData.data[0] // 10) + 1
+        return 1

solax_py_library/smart_scene/types/smart_scene_content.py

@@ -0,0 +1,173 @@
+import json
+import uuid
+from typing import Optional, List
+
+from pydantic import Field
+from pydantic.main import BaseModel
+
+from solax_py_library.device.constant.cabinet import TRENE_CABINET_ENUM
+from solax_py_library.smart_scene.constant.message_entry import MESSAGE_ENTRY
+from solax_py_library.smart_scene.types.action import (
+    SmartSceneAction,
+    ActionType,
+    SystemActionType,
+)
+from solax_py_library.smart_scene.types.condition import (
+    RepeatFunc,
+    SmartSceneCondition,
+    LogicFunc,
+    ConditionType,
+    PriceConditionType,
+    ConditionFunc,
+)
+
+
+class SmartSceneOtherInfo(BaseModel):
+    duration_times: Optional[int] = Field(
+        description="如果需要判断持续时间，记录当前已持续的时间", default=0
+    )
+    version: Optional[int] = Field(
+        description="当前版本，现用于是否做数据迁移，V009版本开始", default=9
+    )
+    once_flag: Optional[bool] = Field(
+        description="是否是单次执行，默认False", default=False
+    )
+
+    @classmethod
+    def new_other_info(cls):
+        return SmartSceneOtherInfo(
+            duration_times=0,
+            version=9,
+            once_flag=False,
+        ).dict()
+
+
+class SmartSceneContent(BaseModel):
+    name: str = Field(description="Scene name", max_length=100)
+    repeatFunction: Optional[RepeatFunc] = Field(description="重复规则")
+    weekList: Optional[List[int]]
+    conditions: SmartSceneCondition = Field(alias="if")
+    thenActions: List[SmartSceneAction] = Field(alias="then")
+    elseThenActions: Optional[List[SmartSceneAction]] = Field(alias="elseThen")
+
+    def build_smart_scene(self, copied=False):
+        return {
+            "scene_id": str(uuid.uuid4()),
+            "switch": False,
+            "content": json.dumps(
+                {
+                    "name": self.name if copied is False else self.name + "-copy",
+                    "repeatFunction": self.repeatFunction,
+                    "weekList": self.weekList,
+                    "if": self.conditions.dict(),
+                    "then": [item.dict() for item in self.thenActions],
+                    "elseThen": [item.dict() for item in self.elseThenActions]
+                    if self.elseThenActions
+                    else None,
+                }
+            ),
+            "other": json.dumps(SmartSceneOtherInfo.new_other_info()),
+        }
+
+    @classmethod
+    def rec_scene(cls, lang, cabinet_type):
+        scene_rec_1 = {
+            "name": MESSAGE_ENTRY["rec_name_01"][lang],
+            "instruction": MESSAGE_ENTRY["instruction_01"][lang],
+            "repeatFunction": RepeatFunc.EVERYDAY,
+            "weekList": [1, 2, 3, 4, 5, 6, 7],
+            "if": {
+                "operation": LogicFunc.AND,
+                "value": [
+                    {
+                        "type": ConditionType.buyingPrice,
+                        "data": [
+                            {
+                                "childType": PriceConditionType.price,
+                                "childData": {
+                                    "function": ConditionFunc.GT,
+                                    "data": [1],
+                                },
+                            }
+                        ],
+                    }
+                ],
+            },
+            "then": [
+                {
+                    "type": ActionType.system,
+                    "data": [
+                        {
+                            "childType": SystemActionType.workMode,
+                            "childData": {
+                                "data": [16, 9, 0, -60, 100],
+                            },
+                        }
+                    ],
+                }
+            ],
+            "elseThen": [],
+        }
+        scene_rec_2 = {
+            "name": MESSAGE_ENTRY["rec_name_02"][lang],
+            "instruction": MESSAGE_ENTRY["instruction_02"][lang],
+            "repeatFunction": RepeatFunc.EVERYDAY,
+            "weekList": [1, 2, 3, 4, 5, 6, 7],
+            "if": {
+                "operation": LogicFunc.AND,
+                "value": [
+                    {
+                        "type": ConditionType.sellingPrice,
+                        "data": [
+                            {
+                                "childType": PriceConditionType.price,
+                                "childData": {
+                                    "function": ConditionFunc.LT,
+                                    "data": [1],
+                                },
+                            }
+                        ],
+                    }
+                ],
+            },
+            "then": [
+                {
+                    "type": ActionType.system,
+                    "data": [
+                        {
+                            "childType": SystemActionType.exportControl,
+                            "childData": {
+                                "data": [1, 2, 0, 1],
+                            },
+                        }
+                    ],
+                }
+            ],
+            "elseThen": [],
+        }
+        if cabinet_type in TRENE_CABINET_ENUM:
+            return [scene_rec_2]
+        else:
+            return [scene_rec_1, scene_rec_2]
+
+    def get_weekday_index(self):
+        if self.repeatFunction == RepeatFunc.EVERYDAY:
+            return [1, 2, 3, 4, 5, 6, 7]
+        elif self.repeatFunction == RepeatFunc.WEEKDAY:
+            return [1, 2, 3, 4, 5]
+        elif self.repeatFunction == RepeatFunc.WEEKEND:
+            return [6, 7]
+        elif self.repeatFunction == RepeatFunc.CUSTOM:
+            return self.weekList
+
+    def get_condition_types(self):
+        condition_types = set()
+        for v in self.conditions.value:
+            condition_types.add(v.type)
+        return condition_types
+
+    def repeat_text_info(self, lang):
+        if self.repeatFunction != RepeatFunc.CUSTOM:
+            return MESSAGE_ENTRY[str(self.repeatFunction.value)][lang]
+        else:
+            return ",".join([MESSAGE_ENTRY[str(d)][lang] for d in self.weekList])

solax_py_library/snap_shot/__init__.py

@@ -1,3 +1,3 @@
-from . import types, core, exceptions
-
-__all__ = ["core", "exceptions", "types"]
+from . import types, core, exceptions
+
+__all__ = ["core", "exceptions", "types"]

solax_py_library/snap_shot/constant/__init__.py

@@ -1,5 +1,5 @@
-from .crc_table import CRC_Table
-
-__all__ = [
-    "CRC_Table",
-]
+from .crc_table import CRC_Table
+
+__all__ = [
+    "CRC_Table",
+]