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

solax_py_library/test/test_smart_scene/test_condition.py

@@ -0,0 +1,11 @@
+from unittest import TestCase
+
+from solax_py_library.smart_scene.core.condition import DateCondition, BaseCondition
+
+
+class TestCondition(TestCase):
+    def test_condition(self):
+        date_condition = DateCondition(
+            update_value_function=lambda: 1,
+        )
+        assert isinstance(date_condition, BaseCondition)

solax_py_library/test/test_utils/test_cloud_client.py

@@ -0,0 +1,14 @@
+from unittest import TestCase
+
+from solax_py_library.utils.cloud_client import CloudClient
+
+
+class TestCloudClient(TestCase):
+    def test_get_weather(self):
+        client = CloudClient()
+        ret = client.get_weather_data_from_cloud(
+            "https://aliyun-sit.solaxtech.net:5050",
+            "XMG11A011L",
+            "b080a22827484db6bd509d496f9df90b",
+        )
+        print(ret)

solax_py_library/upload/__init__.py

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

solax_py_library/upload/api/__init__.py

@@ -1,3 +1,3 @@
-from .service import upload, upload_service
-
-__all__ = ["upload", "upload_service"]
+from .service import upload, upload_service
+
+__all__ = ["upload", "upload_service"]

solax_py_library/upload/api/service.py

@@ -1,24 +1,24 @@
-from typing import Dict, Any
-
-from solax_py_library.upload.core.upload_service import upload_service_map
-from solax_py_library.exception import SolaxBaseError
-from solax_py_library.upload.types.client import UploadType, UploadData
-
-
-def upload_service(upload_type: UploadType, configuration: Dict[str, Any]):
-    """
-    upload_type: 上传类型。
-    configuration: 配置信息
-    """
-    upload_class = upload_service_map.get(upload_type)
-    if not upload_class:
-        raise SolaxBaseError
-    return upload_class(**configuration)
-
-
-async def upload(
-    upload_type: UploadType, configuration: Dict[str, Any], upload_data: UploadData
-):
-    service = upload_service(upload_type, configuration)
-    with service as s:
-        await s.upload(upload_data)
+from typing import Dict, Any
+
+from solax_py_library.upload.core.upload_service import upload_service_map
+from solax_py_library.exception import SolaxBaseError
+from solax_py_library.upload.types.client import UploadType, UploadData
+
+
+def upload_service(upload_type: UploadType, configuration: Dict[str, Any]):
+    """
+    upload_type: 上传类型。
+    configuration: 配置信息
+    """
+    upload_class = upload_service_map.get(upload_type)
+    if not upload_class:
+        raise SolaxBaseError
+    return upload_class(**configuration)
+
+
+async def upload(
+    upload_type: UploadType, configuration: Dict[str, Any], upload_data: UploadData
+):
+    service = upload_service(upload_type, configuration)
+    with service as s:
+        await s.upload(upload_data)

solax_py_library/upload/core/__init__.py

@@ -1,3 +1,3 @@
-from . import upload_service, data_adapter
-
-__all__ = ["upload_service", "data_adapter"]
+from . import upload_service, data_adapter
+
+__all__ = ["upload_service", "data_adapter"]

solax_py_library/upload/core/data_adapter/__init__.py

@@ -1,5 +1,5 @@
-from .base import BaseDataAdapter
-from .csv import CSVDataAdapter
-
-
-__all__ = ["BaseDataAdapter", "CSVDataAdapter"]
+from .base import BaseDataAdapter
+from .csv import CSVDataAdapter
+
+
+__all__ = ["BaseDataAdapter", "CSVDataAdapter"]

solax_py_library/upload/core/data_adapter/base.py

@@ -1,9 +1,9 @@
-from abc import ABCMeta, abstractmethod
-
-
-class BaseDataAdapter(metaclass=ABCMeta):
-    data_type = None
-
-    @abstractmethod
-    def parse_data(self, data):
-        ...
+from abc import ABCMeta, abstractmethod
+
+
+class BaseDataAdapter(metaclass=ABCMeta):
+    data_type = None
+
+    @abstractmethod
+    def parse_data(self, data):
+        ...

solax_py_library/upload/core/data_adapter/csv.py

@@ -1,26 +1,26 @@
-import csv
-import tempfile
-from typing import Any, Union, List, Dict
-
-from solax_py_library.upload.core.data_adapter.base import BaseDataAdapter
-
-
-class CSVDataAdapter(BaseDataAdapter):
-    @classmethod
-    def parse_data(cls, data: Union[List[List[Any]], List[Dict[str, Any]]]):
-        with tempfile.NamedTemporaryFile(
-            delete=False, mode="w", newline="", encoding="utf-8-sig"
-        ) as temp_file:
-            if isinstance(data[0], list):
-                writer = csv.writer(temp_file)
-                writer.writerows(data)
-            else:
-                headers = list(data[0].keys())
-                writer = csv.DictWriter(temp_file, fieldnames=headers)
-                writer.writeheader()
-                writer.writerows(data)
-
-            temp_file_path = temp_file.name
-            temp_file.seek(0)
-
-        return temp_file_path
+import csv
+import tempfile
+from typing import Any, Union, List, Dict
+
+from solax_py_library.upload.core.data_adapter.base import BaseDataAdapter
+
+
+class CSVDataAdapter(BaseDataAdapter):
+    @classmethod
+    def parse_data(cls, data: Union[List[List[Any]], List[Dict[str, Any]]]):
+        with tempfile.NamedTemporaryFile(
+            delete=False, mode="w", newline="", encoding="utf-8-sig"
+        ) as temp_file:
+            if isinstance(data[0], list):
+                writer = csv.writer(temp_file)
+                writer.writerows(data)
+            else:
+                headers = list(data[0].keys())
+                writer = csv.DictWriter(temp_file, fieldnames=headers)
+                writer.writeheader()
+                writer.writerows(data)
+
+            temp_file_path = temp_file.name
+            temp_file.seek(0)
+
+        return temp_file_path

solax_py_library/upload/core/upload_service/__init__.py

@@ -1,15 +1,15 @@
-from .base import BaseUploadService
-from .ftp import FTPUploadService
-
-
-upload_service_map = {}
-
-
-def _register(upload_obj):
-    upload_service_map[upload_obj.upload_type] = upload_obj
-
-
-_register(FTPUploadService)
-
-
-__all__ = ["BaseUploadService", "FTPUploadService"]
+from .base import BaseUploadService
+from .ftp import FTPUploadService
+
+
+upload_service_map = {}
+
+
+def _register(upload_obj):
+    upload_service_map[upload_obj.upload_type] = upload_obj
+
+
+_register(FTPUploadService)
+
+
+__all__ = ["BaseUploadService", "FTPUploadService"]

solax_py_library/upload/core/upload_service/base.py

@@ -1,43 +1,43 @@
-from abc import ABCMeta, abstractmethod
-
-from solax_py_library.upload.types.client import UploadData
-
-
-class BaseUploadService(metaclass=ABCMeta):
-    upload_type = None
-
-    def __init__(self, **kwargs) -> None:
-        self._client = None
-        self._is_connect = False
-        self.timeout = 5
-
-    @abstractmethod
-    def connect(self):
-        ...
-
-    async def upload(self, data: UploadData):
-        upload_data = self._parse(data.build_data())
-        return self._upload(upload_data)
-
-    @abstractmethod
-    def close(self):
-        ...
-
-    @property
-    def is_connect(self):
-        return self._is_connect
-
-    def __enter__(self):
-        self.connect()
-        return self
-
-    def __exit__(self, exc_type, exc_value, traceback):
-        self.close()
-
-    @abstractmethod
-    def _upload(self, data):
-        ...
-
-    @abstractmethod
-    def _parse(self, upload_data: UploadData):
-        ...
+from abc import ABCMeta, abstractmethod
+
+from solax_py_library.upload.types.client import UploadData
+
+
+class BaseUploadService(metaclass=ABCMeta):
+    upload_type = None
+
+    def __init__(self, **kwargs) -> None:
+        self._client = None
+        self._is_connect = False
+        self.timeout = 5
+
+    @abstractmethod
+    def connect(self):
+        ...
+
+    async def upload(self, data: UploadData):
+        upload_data = self._parse(data.build_data())
+        return self._upload(upload_data)
+
+    @abstractmethod
+    def close(self):
+        ...
+
+    @property
+    def is_connect(self):
+        return self._is_connect
+
+    def __enter__(self):
+        self.connect()
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.close()
+
+    @abstractmethod
+    def _upload(self, data):
+        ...
+
+    @abstractmethod
+    def _parse(self, upload_data: UploadData):
+        ...

solax_py_library/upload/exceptions/__init__.py

@@ -1,8 +1,8 @@
-from .upload_error import ConfigurationError, ConnectError, LoginError, SendDataError
-
-__all__ = [
-    "ConnectError",
-    "LoginError",
-    "SendDataError",
-    "ConfigurationError",
-]
+from .upload_error import ConfigurationError, ConnectError, LoginError, SendDataError
+
+__all__ = [
+    "ConnectError",
+    "LoginError",
+    "SendDataError",
+    "ConfigurationError",
+]

solax_py_library/upload/exceptions/upload_error.py

@@ -1,21 +1,21 @@
-from solax_py_library.exception import SolaxBaseError
-
-
-class ConnectError(SolaxBaseError):
-    code = 0x1001
-    message = "connect error"
-
-
-class LoginError(SolaxBaseError):
-    code = 0x1002
-    message = "authentication error"
-
-
-class SendDataError(SolaxBaseError):
-    code = 0x1003
-    message = "send data error"
-
-
-class ConfigurationError(SolaxBaseError):
-    code = 0x1004
-    message = "server configuration error"
+from solax_py_library.exception import SolaxBaseError
+
+
+class ConnectError(SolaxBaseError):
+    code = 0x1001
+    message = "connect error"
+
+
+class LoginError(SolaxBaseError):
+    code = 0x1002
+    message = "authentication error"
+
+
+class SendDataError(SolaxBaseError):
+    code = 0x1003
+    message = "send data error"
+
+
+class ConfigurationError(SolaxBaseError):
+    code = 0x1004
+    message = "server configuration error"

solax_py_library/upload/test/test_ftp.py

@@ -1,113 +1,113 @@
-import asyncio
-import unittest
-
-from solax_py_library.upload.api.service import upload
-from solax_py_library.upload.core.upload_service import FTPUploadService
-from solax_py_library.upload.exceptions import ConnectError, LoginError
-from solax_py_library.upload.types.client import UploadType, UploadData
-from solax_py_library.upload.types.ftp import FTPFileType
-
-
-class FTPTest(unittest.TestCase):
-    def test_connect(self):
-        ftp_config = {
-            "host": "10.1.31.181",  # 测试host
-            "port": 21,
-            "user": "solax",
-            "password": "123456",
-            "remote_path": "/xixi",
-        }
-        ftp = FTPUploadService(**ftp_config)
-        ftp.connect()
-
-    def test_connect_error_1(self):
-        ftp_config = {
-            "host": "10.1.31.182",  # 测试host
-            "port": 21,
-            "user": "solax",
-            "password": "123456",
-            "remote_path": "/xixi",
-        }
-        ftp = FTPUploadService(**ftp_config)
-        try:
-            ftp.connect()
-        except ConnectError:
-            ...
-
-    def test_connect_error_2(self):
-        ftp_config = {
-            "host": "10.1.31.181",  # 测试host
-            "port": 21,
-            "user": "solax123",
-            "password": "123456",
-            "remote_path": "/xixi",
-        }
-        ftp = FTPUploadService(**ftp_config)
-        try:
-            ftp.connect()
-        except LoginError:
-            ...
-
-    def test_ftp_upload_to_windows(self):
-        ftp_config = {
-            "host": "10.1.31.181",  # 测试host
-            "port": 21,
-            "user": "solax",
-            "password": "123456",
-            "remote_path": "嘻嘻",
-        }
-        asyncio.run(
-            upload(
-                upload_type=UploadType.FTP,
-                configuration=ftp_config,
-                upload_data=UploadData(
-                    upload_type=UploadType.FTP,
-                    data=dict(
-                        file_type=FTPFileType.CSV,
-                        file_name="中文",
-                        data=[
-                            {
-                                "EMS1000序列号": "XMG11A011L",
-                                "EMS1000本地时间": "2025-02-11 15:39:10",
-                                "EMS1000版本号": "V007.11.1",
-                                "电站所在国家和地区": None,
-                                "电站所在当前时区": None,
-                                "电站系统类型": None,
-                            }
-                        ],
-                    ),
-                ),
-            )
-        )
-
-    def test_ftp_upload_to_linux(self):
-        ftp_config = {
-            "host": "920729yofx76.vicp.fun",  # 测试host
-            "port": 59477,
-            "user": "test",
-            "password": "test123456",
-            "remote_path": "test",
-        }
-        asyncio.run(
-            upload(
-                upload_type=UploadType.FTP,
-                configuration=ftp_config,
-                upload_data=UploadData(
-                    upload_type=UploadType.FTP,
-                    data=dict(
-                        file_type=FTPFileType.CSV,
-                        file_name="中文",
-                        data=[
-                            {
-                                "EMS1000序列号": "XMG11A011L",
-                                "EMS1000本地时间": "2025-02-11 15:39:10",
-                                "EMS1000版本号": "V007.11.1",
-                                "电站所在国家和地区": None,
-                                "电站所在当前时区": None,
-                                "电站系统类型": None,
-                            }
-                        ],
-                    ),
-                ),
-            )
-        )
+import asyncio
+import unittest
+
+from solax_py_library.upload.api.service import upload
+from solax_py_library.upload.core.upload_service import FTPUploadService
+from solax_py_library.upload.exceptions import ConnectError, LoginError
+from solax_py_library.upload.types.client import UploadType, UploadData
+from solax_py_library.upload.types.ftp import FTPFileType
+
+
+class FTPTest(unittest.TestCase):
+    def test_connect(self):
+        ftp_config = {
+            "host": "10.1.31.181",  # 测试host
+            "port": 21,
+            "user": "solax",
+            "password": "123456",
+            "remote_path": "/xixi",
+        }
+        ftp = FTPUploadService(**ftp_config)
+        ftp.connect()
+
+    def test_connect_error_1(self):
+        ftp_config = {
+            "host": "10.1.31.182",  # 测试host
+            "port": 21,
+            "user": "solax",
+            "password": "123456",
+            "remote_path": "/xixi",
+        }
+        ftp = FTPUploadService(**ftp_config)
+        try:
+            ftp.connect()
+        except ConnectError:
+            ...
+
+    def test_connect_error_2(self):
+        ftp_config = {
+            "host": "10.1.31.181",  # 测试host
+            "port": 21,
+            "user": "solax123",
+            "password": "123456",
+            "remote_path": "/xixi",
+        }
+        ftp = FTPUploadService(**ftp_config)
+        try:
+            ftp.connect()
+        except LoginError:
+            ...
+
+    def test_ftp_upload_to_windows(self):
+        ftp_config = {
+            "host": "10.1.31.181",  # 测试host
+            "port": 21,
+            "user": "solax",
+            "password": "123456",
+            "remote_path": "嘻嘻",
+        }
+        asyncio.run(
+            upload(
+                upload_type=UploadType.FTP,
+                configuration=ftp_config,
+                upload_data=UploadData(
+                    upload_type=UploadType.FTP,
+                    data=dict(
+                        file_type=FTPFileType.CSV,
+                        file_name="中文",
+                        data=[
+                            {
+                                "EMS1000序列号": "XMG11A011L",
+                                "EMS1000本地时间": "2025-02-11 15:39:10",
+                                "EMS1000版本号": "V007.11.1",
+                                "电站所在国家和地区": None,
+                                "电站所在当前时区": None,
+                                "电站系统类型": None,
+                            }
+                        ],
+                    ),
+                ),
+            )
+        )
+
+    def test_ftp_upload_to_linux(self):
+        ftp_config = {
+            "host": "920729yofx76.vicp.fun",  # 测试host
+            "port": 59477,
+            "user": "test",
+            "password": "test123456",
+            "remote_path": "test",
+        }
+        asyncio.run(
+            upload(
+                upload_type=UploadType.FTP,
+                configuration=ftp_config,
+                upload_data=UploadData(
+                    upload_type=UploadType.FTP,
+                    data=dict(
+                        file_type=FTPFileType.CSV,
+                        file_name="中文",
+                        data=[
+                            {
+                                "EMS1000序列号": "XMG11A011L",
+                                "EMS1000本地时间": "2025-02-11 15:39:10",
+                                "EMS1000版本号": "V007.11.1",
+                                "电站所在国家和地区": None,
+                                "电站所在当前时区": None,
+                                "电站系统类型": None,
+                            }
+                        ],
+                    ),
+                ),
+            )
+        )

solax_py_library/upload/types/__init__.py

@@ -1,11 +1,11 @@
-from .client import UploadType, UploadData
-from .ftp import FTPData, FTPParsedData, FTPFileType, FTPServiceConfig
-
-__all__ = [
-    "FTPData",
-    "FTPParsedData",
-    "FTPServiceConfig",
-    "UploadType",
-    "UploadData",
-    "FTPFileType",
-]
+from .client import UploadType, UploadData
+from .ftp import FTPData, FTPParsedData, FTPFileType, FTPServiceConfig
+
+__all__ = [
+    "FTPData",
+    "FTPParsedData",
+    "FTPServiceConfig",
+    "UploadType",
+    "UploadData",
+    "FTPFileType",
+]

solax_py_library/upload/types/client.py

@@ -1,19 +1,19 @@
-from enum import IntEnum
-from pydantic import BaseModel
-from typing import Optional, Union
-
-from solax_py_library.upload.types.ftp import FTPData
-
-
-class UploadType(IntEnum):
-    FTP = 1
-
-
-class UploadData(BaseModel):
-    data: Optional[Union[dict, FTPData]]
-    upload_type: UploadType
-
-    def build_data(self):
-        dict_obj_data = self.data if isinstance(self.data, dict) else self.data.dict()
-        if self.upload_type == UploadType.FTP:
-            return FTPData(**dict_obj_data)
+from enum import IntEnum
+from pydantic import BaseModel
+from typing import Optional, Union
+
+from solax_py_library.upload.types.ftp import FTPData
+
+
+class UploadType(IntEnum):
+    FTP = 1
+
+
+class UploadData(BaseModel):
+    data: Optional[Union[dict, FTPData]]
+    upload_type: UploadType
+
+    def build_data(self):
+        dict_obj_data = self.data if isinstance(self.data, dict) else self.data.dict()
+        if self.upload_type == UploadType.FTP:
+            return FTPData(**dict_obj_data)