From ca4d537529457133e407b988aa4d56c2d3e33f91 Mon Sep 17 00:00:00 2001 From: elgris <1905821+elgris@users.noreply.github.com> Date: Thu, 19 Feb 2026 00:32:23 +0100 Subject: [PATCH] Control datetime on SwitchBot Meter Pro CO2 (#161808) --- .../components/switchbot/__init__.py | 6 +- homeassistant/components/switchbot/button.py | 47 ++++++++ .../components/switchbot/strings.json | 3 + tests/components/switchbot/test_button.py | 109 +++++++++++++++++- 4 files changed, 163 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/switchbot/__init__.py b/homeassistant/components/switchbot/__init__.py index e24751c9a40..c002318d6da 100644 --- a/homeassistant/components/switchbot/__init__.py +++ b/homeassistant/components/switchbot/__init__.py @@ -53,7 +53,11 @@ PLATFORMS_BY_TYPE = { Platform.SENSOR, ], SupportedModels.HYGROMETER.value: [Platform.SENSOR], - SupportedModels.HYGROMETER_CO2.value: [Platform.SENSOR, Platform.SELECT], + SupportedModels.HYGROMETER_CO2.value: [ + Platform.BUTTON, + Platform.SENSOR, + Platform.SELECT, + ], SupportedModels.CONTACT.value: [Platform.BINARY_SENSOR, Platform.SENSOR], SupportedModels.MOTION.value: [Platform.BINARY_SENSOR, Platform.SENSOR], SupportedModels.PRESENCE_SENSOR.value: [Platform.BINARY_SENSOR, Platform.SENSOR], diff --git a/homeassistant/components/switchbot/button.py b/homeassistant/components/switchbot/button.py index a5a32f96f50..3d9db9074f2 100644 --- a/homeassistant/components/switchbot/button.py +++ b/homeassistant/components/switchbot/button.py @@ -5,8 +5,10 @@ import logging import switchbot from homeassistant.components.button import ButtonEntity +from homeassistant.const import EntityCategory from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback +from homeassistant.util import dt as dt_util from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator from .entity import SwitchbotEntity, exception_handler @@ -31,6 +33,9 @@ async def async_setup_entry( ] ) + if isinstance(coordinator.device, switchbot.SwitchbotMeterProCO2): + async_add_entities([SwitchBotMeterProCO2SyncDateTimeButton(coordinator)]) + class SwitchBotArtFrameButtonBase(SwitchbotEntity, ButtonEntity): """Base class for Art Frame buttons.""" @@ -64,3 +69,45 @@ class SwitchBotArtFramePrevButton(SwitchBotArtFrameButtonBase): """Handle the button press.""" _LOGGER.debug("Pressing previous image button %s", self._address) await self._device.prev_image() + + +class SwitchBotMeterProCO2SyncDateTimeButton(SwitchbotEntity, ButtonEntity): + """Button to sync date and time on Meter Pro CO2 to the current HA instance datetime.""" + + _device: switchbot.SwitchbotMeterProCO2 + _attr_entity_category = EntityCategory.CONFIG + _attr_translation_key = "sync_datetime" + + def __init__(self, coordinator: SwitchbotDataUpdateCoordinator) -> None: + """Initialize the sync time button.""" + super().__init__(coordinator) + self._attr_unique_id = f"{coordinator.base_unique_id}_sync_datetime" + + @exception_handler + async def async_press(self) -> None: + """Sync time with Home Assistant.""" + now = dt_util.now() + + # Get UTC offset components + utc_offset = now.utcoffset() + utc_offset_hours, utc_offset_minutes = 0, 0 + if utc_offset is not None: + total_seconds = int(utc_offset.total_seconds()) + utc_offset_hours = total_seconds // 3600 + utc_offset_minutes = abs(total_seconds % 3600) // 60 + + timestamp = int(now.timestamp()) + + _LOGGER.debug( + "Syncing time for %s: timestamp=%s, utc_offset_hours=%s, utc_offset_minutes=%s", + self._address, + timestamp, + utc_offset_hours, + utc_offset_minutes, + ) + + await self._device.set_datetime( + timestamp=timestamp, + utc_offset_hours=utc_offset_hours, + utc_offset_minutes=utc_offset_minutes, + ) diff --git a/homeassistant/components/switchbot/strings.json b/homeassistant/components/switchbot/strings.json index 9c9d36fd319..288cc5437e6 100644 --- a/homeassistant/components/switchbot/strings.json +++ b/homeassistant/components/switchbot/strings.json @@ -106,6 +106,9 @@ }, "previous_image": { "name": "Previous image" + }, + "sync_datetime": { + "name": "Sync date and time" } }, "climate": { diff --git a/tests/components/switchbot/test_button.py b/tests/components/switchbot/test_button.py index bce9c5f5d5a..e01353869b9 100644 --- a/tests/components/switchbot/test_button.py +++ b/tests/components/switchbot/test_button.py @@ -1,6 +1,7 @@ """Tests for the switchbot button platform.""" from collections.abc import Callable +from datetime import UTC, datetime, timedelta, timezone from unittest.mock import AsyncMock, patch import pytest @@ -8,8 +9,9 @@ import pytest from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN, SERVICE_PRESS from homeassistant.const import ATTR_ENTITY_ID from homeassistant.core import HomeAssistant +from homeassistant.setup import async_setup_component -from . import ART_FRAME_INFO +from . import ART_FRAME_INFO, DOMAIN, WOMETERTHPC_SERVICE_INFO from tests.common import MockConfigEntry from tests.components.bluetooth import inject_bluetooth_service_info @@ -60,3 +62,108 @@ async def test_art_frame_button_press( ) mocked_instance.assert_awaited_once() + + +async def test_meter_pro_co2_sync_datetime_button( + hass: HomeAssistant, + mock_entry_factory: Callable[[str], MockConfigEntry], +) -> None: + """Test pressing the sync datetime button on Meter Pro CO2.""" + await async_setup_component(hass, DOMAIN, {}) + inject_bluetooth_service_info(hass, WOMETERTHPC_SERVICE_INFO) + + entry = mock_entry_factory("hygrometer_co2") + entry.add_to_hass(hass) + + mock_set_datetime = AsyncMock(return_value=True) + + # Use a fixed datetime for testing + fixed_time = datetime(2025, 1, 9, 12, 30, 45, tzinfo=UTC) + + with ( + patch( + "switchbot.SwitchbotMeterProCO2.set_datetime", + mock_set_datetime, + ), + patch( + "homeassistant.components.switchbot.button.dt_util.now", + return_value=fixed_time, + ), + ): + assert await hass.config_entries.async_setup(entry.entry_id) + await hass.async_block_till_done() + + entity_ids = [ + entity.entity_id for entity in hass.states.async_all(BUTTON_DOMAIN) + ] + assert "button.test_name_sync_date_and_time" in entity_ids + + await hass.services.async_call( + BUTTON_DOMAIN, + SERVICE_PRESS, + {ATTR_ENTITY_ID: "button.test_name_sync_date_and_time"}, + blocking=True, + ) + + mock_set_datetime.assert_awaited_once_with( + timestamp=int(fixed_time.timestamp()), + utc_offset_hours=0, + utc_offset_minutes=0, + ) + + +@pytest.mark.parametrize( + ("tz", "expected_utc_offset_hours", "expected_utc_offset_minutes"), + [ + (timezone(timedelta(hours=0, minutes=0)), 0, 0), + (timezone(timedelta(hours=0, minutes=30)), 0, 30), + (timezone(timedelta(hours=8, minutes=0)), 8, 0), + (timezone(timedelta(hours=-5, minutes=30)), -5, 30), + (timezone(timedelta(hours=5, minutes=30)), 5, 30), + (timezone(timedelta(hours=-5, minutes=-30)), -6, 30), # -6h + 30m = -5:30 + (timezone(timedelta(hours=-5, minutes=-45)), -6, 15), # -6h + 15m = -5:45 + ], +) +async def test_meter_pro_co2_sync_datetime_button_with_timezone( + hass: HomeAssistant, + mock_entry_factory: Callable[[str], MockConfigEntry], + tz: timezone, + expected_utc_offset_hours: int, + expected_utc_offset_minutes: int, +) -> None: + """Test sync datetime button with non-UTC timezone.""" + await async_setup_component(hass, DOMAIN, {}) + inject_bluetooth_service_info(hass, WOMETERTHPC_SERVICE_INFO) + + entry = mock_entry_factory("hygrometer_co2") + entry.add_to_hass(hass) + + mock_set_datetime = AsyncMock(return_value=True) + + fixed_time = datetime(2025, 1, 9, 18, 0, 45, tzinfo=tz) + + with ( + patch( + "switchbot.SwitchbotMeterProCO2.set_datetime", + mock_set_datetime, + ), + patch( + "homeassistant.components.switchbot.button.dt_util.now", + return_value=fixed_time, + ), + ): + assert await hass.config_entries.async_setup(entry.entry_id) + await hass.async_block_till_done() + + await hass.services.async_call( + BUTTON_DOMAIN, + SERVICE_PRESS, + {ATTR_ENTITY_ID: "button.test_name_sync_date_and_time"}, + blocking=True, + ) + + mock_set_datetime.assert_awaited_once_with( + timestamp=int(fixed_time.timestamp()), + utc_offset_hours=expected_utc_offset_hours, + utc_offset_minutes=expected_utc_offset_minutes, + )