daybetter-services-python 1.0.5__tar.gz → 1.0.6__tar.gz

{daybetter_services_python-1.0.5/daybetter_services_python.egg-info → daybetter_services_python-1.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: daybetter-services-python
-Version: 1.0.5
+Version: 1.0.6
 Summary: Python client for DayBetter devices and services
 Home-page: https://github.com/THDayBetter/daybetter-python
 Author: THDayBetter

{daybetter_services_python-1.0.5 → daybetter_services_python-1.0.6}/daybetter_python/__init__.py

@@ -3,5 +3,5 @@
 from .client import DayBetterClient
 from .exceptions import DayBetterError, AuthenticationError, APIError
 
-__version__ = "1.0.5"
+__version__ = "1.0.6"
 __all__ = ["DayBetterClient", "DayBetterError", "AuthenticationError", "APIError"]

{daybetter_services_python-1.0.5 → daybetter_services_python-1.0.6}/daybetter_python/client.py

@@ -12,19 +12,40 @@ _LOGGER = logging.getLogger(__name__)
 class DayBetterClient:
     """DayBetter API client."""
     
+    # 测试环境URL
+    TEST_BASE_URL = "https://cloud.v2.dbiot.link/daybetter/hass/api/v1.0/"
+    # 正式环境URL
+    PROD_BASE_URL = "https://a.dbiot.org/daybetter/hass/api/v1.0/"
+    
     def __init__(
         self, 
         token: str, 
-        base_url: str = "https://cloud.v2.dbiot.link/daybetter/hass/api/v1.0/"
+        base_url: Optional[str] = None,
+        hass_code: Optional[str] = None
     ):
         """Initialize the client.
         
         Args:
             token: Authentication token
-            base_url: Base URL for the API
+            base_url: Base URL for the API (optional, will be determined by hass_code if not provided)
+            hass_code: Home Assistant integration code (optional, if provided and starts with "db-", 
+                      will use production environment)
         """
         self.token = token
-        self.base_url = base_url
+        
+        # 根据 hass_code 或 base_url 确定使用的环境
+        if base_url is not None:
+            # 如果明确指定了 base_url，使用指定的 URL
+            self.base_url = base_url
+        elif hass_code is not None and hass_code.startswith("db-"):
+            # 如果 hass_code 以 "db-" 开头，使用正式环境
+            self.base_url = self.PROD_BASE_URL
+            _LOGGER.debug("Using production environment based on hass_code")
+        else:
+            # 默认使用测试环境
+            self.base_url = self.TEST_BASE_URL
+            _LOGGER.debug("Using test environment")
+        
         self._session: Optional[aiohttp.ClientSession] = None
         self._auth_valid = True
         self._devices: List[Dict[str, Any]] = []
@@ -306,6 +327,24 @@ class DayBetterClient:
         Raises:
             APIError: If API request fails
         """
+        # 根据 hass_code 动态更新 base_url（如果之前没有明确指定）
+        # 如果 hass_code 以 "db-" 开头，切换到正式环境
+        if hass_code.startswith("db-") and self.base_url != self.PROD_BASE_URL:
+            old_url = self.base_url
+            self.base_url = self.PROD_BASE_URL
+            _LOGGER.info(
+                "Switching to production environment based on hass_code. "
+                "URL changed from %s to %s", old_url, self.base_url
+            )
+        elif not hass_code.startswith("db-") and self.base_url != self.TEST_BASE_URL:
+            # 如果 hass_code 不以 "db-" 开头，且当前不是测试环境，切换到测试环境
+            old_url = self.base_url
+            self.base_url = self.TEST_BASE_URL
+            _LOGGER.info(
+                "Switching to test environment based on hass_code. "
+                "URL changed from %s to %s", old_url, self.base_url
+            )
+        
         try:
             session = self._get_session()
             url = f"{self.base_url}hass/integrate"

{daybetter_services_python-1.0.5 → daybetter_services_python-1.0.6/daybetter_services_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: daybetter-services-python
-Version: 1.0.5
+Version: 1.0.6
 Summary: Python client for DayBetter devices and services
 Home-page: https://github.com/THDayBetter/daybetter-python
 Author: THDayBetter

{daybetter_services_python-1.0.5 → daybetter_services_python-1.0.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "daybetter-services-python"
-version = "1.0.5"
+version = "1.0.6"
 description = "Python client for DayBetter devices and services"
 readme = "README.md"
 license = "MIT"

{daybetter_services_python-1.0.5 → daybetter_services_python-1.0.6}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="daybetter-services-python",
-    version="1.0.5",
+    version="1.0.6",
     author="THDayBetter",
     author_email="chenp2368@163.com",
     description="Python client for DayBetter devices and services",