feat: Add pumping sensor and service

README.md

@@ -95,6 +95,17 @@ This service adds a feeding entry for your child. Feeding start/end/child fields
 | amount | yes |   Specify amount of feeding as an integer
 | notes | yes |   Add notes text to entry
 
+### SERVICE ADD_PUMPING
+
+This service adds a pumping entry for your child.
+
+| Service data attribute | Optional | Description |
+|------|:----:|-------------|
+| entity_id | no |   entity_id for the child sensor
+| amount | no |   Specify amount of pumping as an integer
+| time | yes |   Specify pumping recording time (must be in the past, else now() will be used)
+| notes | yes |   Add notes text to entry
+
 ### SERVICE ADD_SLEEP
 
 This service adds a sleep entry for your child. Sleep start/end/child fields can be linked to an active timer.

custom_components/babybuddy/const.py

@@ -52,6 +52,7 @@
 ATTR_NOTE: Final = "note"
 ATTR_NOTES: Final = "notes"
 ATTR_PICTURE: Final = "picture"
+ATTR_PUMPING: Final = "pumping"
 ATTR_RESULTS: Final = "results"
 ATTR_SLEEP: Final = "sleep"
 ATTR_SLUG: Final = "slug"
@@ -126,6 +127,12 @@ class BabyBuddyEntityDescription(SensorEntityDescription, SwitchEntityDescriptio
         key=ATTR_NOTES,
         state_key=ATTR_TIME,
     ),
+    BabyBuddyEntityDescription(
+        icon="mdi:mother-nurse",
+        key=ATTR_PUMPING,
+        state_class=STATE_CLASS_MEASUREMENT,
+        state_key=ATTR_AMOUNT,
+    ),
     BabyBuddyEntityDescription(
         icon="mdi:sleep",
         key=ATTR_SLEEP,

custom_components/babybuddy/sensor.py

@@ -44,6 +44,7 @@
     ATTR_NOTE,
     ATTR_NOTES,
     ATTR_PICTURE,
+    ATTR_PUMPING,
     ATTR_SLUG,
     ATTR_SOLID,
     ATTR_TYPE,
@@ -130,6 +131,15 @@ def update_entities() -> None:
         },
         "async_add_note",
     )
+    platform.async_register_entity_service(
+        "add_pumping",
+        {
+            vol.Required(ATTR_AMOUNT): cv.positive_int,
+            vol.Optional(ATTR_TIME): vol.Any(cv.datetime, cv.time),
+            vol.Optional(ATTR_NOTES): cv.string,
+        },
+        "async_add_pumping",
+    )
     platform.async_register_entity_service(
         "add_temperature",
         {
@@ -316,6 +326,34 @@ async def async_add_note(
         await self.coordinator.client.async_post(ATTR_NOTES, data, date_time_now)
         await self.coordinator.async_request_refresh()
 
+    async def async_add_pumping(
+        self,
+        amount: float,
+        time: datetime | time | None = None,
+        notes: str | None = None,
+    ) -> None:
+        """Add a pumping entry."""
+        if not isinstance(self, BabyBuddyChildSensor):
+            _LOGGER.debug("Babybuddy child sensor should be selected. Ignoring.")
+            return
+        data = {
+            ATTR_CHILD: self.child[ATTR_ID],
+            ATTR_AMOUNT: amount,
+        }
+        if time:
+            try:
+                date_time = get_datetime_from_time(time)
+                data[ATTR_TIME] = date_time
+            except ValidationError as err:
+                _LOGGER.error(err)
+                return
+        if notes:
+            data[ATTR_NOTES] = notes
+
+        date_time_now = get_datetime_from_time(dt_util.now())
+        await self.coordinator.client.async_post(ATTR_PUMPING, data, date_time_now)
+        await self.coordinator.async_request_refresh()
+
     async def async_add_temperature(
         self,
         temperature: float,

custom_components/babybuddy/services.yaml

@@ -248,6 +248,35 @@ add_note:
       selector:
         time:
 
+add_pumping:
+  name: Add pumping
+  description: Adds a pumping entry
+  target:
+    entity:
+      integration: babybuddy
+      domain: sensor
+      device_class: babybuddy_child
+  fields:
+    amount:
+      name: Pumping amount
+      description: Pumping amount
+      selector:
+        number:
+          min: 1
+          max: 10
+          mode: box
+    time:
+      name: Pumping time
+      description: Pumping time
+      selector:
+        time:
+    notes:
+      name: Pumping notes
+      description: Pumping notes
+      selector:
+        text:
+          multiline: true
+
 add_sleep:
   name: Add sleep
   description: Adds sleep entry