From 7422b4866ad16fbe95ab8292354432afd6c4538c Mon Sep 17 00:00:00 2001
From: sbabcock23 <steve.w.babcock@gmail.com>
Date: Thu, 25 Aug 2022 09:50:06 -0500
Subject: [PATCH 1/2] updated readme and increased version

---
 README.md                             | 2 ++
 custom_components/tryfi/manifest.json | 4 ++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 4ca85b8..6865821 100644
--- a/README.md
+++ b/README.md
@@ -139,6 +139,8 @@ Turns on the collar light after dark if the pet is not home.
 * Enable possibility of if pet not home and not with owner then trigger lost dog mode
 
 # Version History
+## 0.0.16
+* Enhancement - requested to add the attributes Activity Type, Current Place Name and Current Place Address.
 ## 0.0.15
 * Version bump to support latest pytryfi version 0.0.16 that includes multiple households
 ## 0.0.14
diff --git a/custom_components/tryfi/manifest.json b/custom_components/tryfi/manifest.json
index d69a83c..5fd24a7 100644
--- a/custom_components/tryfi/manifest.json
+++ b/custom_components/tryfi/manifest.json
@@ -1,12 +1,12 @@
 {
     "domain": "tryfi",
     "name": "TryFi",
-    "version": "0.0.15",
+    "version": "0.0.16",
     "config_flow": true,
     "documentation": "https://github.com/sbabcock23/hass-tryfi",
     "issue_tracker": "https://github.com/sbabcock23/hass-tryfi/issues",
     "requirements": [
-      "pytryfi>=0.0.16"
+      "pytryfi>=0.0.17"
     ],
     "dependencies": [],
     "iot_class": "cloud_polling",

From 3de8cb760c65ee80a34b9f8bf2174cb45e3c8fbf Mon Sep 17 00:00:00 2001
From: sbabcock23 <steve.w.babcock@gmail.com>
Date: Thu, 25 Aug 2022 09:51:20 -0500
Subject: [PATCH 2/2] added current place name and current place address
 attributes

---
 custom_components/tryfi/sensor.py | 84 +++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)

diff --git a/custom_components/tryfi/sensor.py b/custom_components/tryfi/sensor.py
index b92d4a4..cee207f 100644
--- a/custom_components/tryfi/sensor.py
+++ b/custom_components/tryfi/sensor.py
@@ -40,6 +40,11 @@ async def async_setup_entry(hass, config_entry, async_add_devices):
                 new_devices.append(
                     PetStatsSensor(hass, pet, coordinator, statType, statTime)
                 )
+        LOGGER.debug(f"Adding Pet Generic Sensor: {pet}")
+        new_devices.append(PetGenericSensor(hass, pet, coordinator, "Activity Type"))
+        new_devices.append(PetGenericSensor(hass, pet, coordinator, "Current Place Name"))
+        new_devices.append(PetGenericSensor(hass, pet, coordinator, "Current Place Address"))
+
     for base in tryfi.bases:
         LOGGER.debug(f"Adding Base: {base}")
         new_devices.append(TryFiBaseSensor(hass, base, coordinator))
@@ -105,6 +110,85 @@ def device_info(self):
             # "sw_version": self.pet.device.buildId,
         }
 
+class PetGenericSensor(CoordinatorEntity, Entity):
+    """Representation of a Sensor."""
+
+    def __init__(self, hass, pet, coordinator, statType):
+        self._hass = hass
+        self._petId = pet.petId
+        self._statType = statType
+        super().__init__(coordinator)
+    
+    @property
+    def statType(self):
+        return self._statType
+
+    @property
+    def statTime(self):
+        return self._statTime
+
+    @property
+    def name(self):
+        """Return the name of the sensor."""
+        return f"{self.pet.name} {self.statType.title()}"
+
+    @property
+    def unique_id(self):
+        """Return the ID of this sensor."""
+        formattedType = self.statType.lower().replace(" ", "-")
+        return f"{self.pet.petId}-{formattedType}"
+
+    @property
+    def petId(self):
+        return self._petId
+
+    @property
+    def pet(self):
+        return self.coordinator.data.getPet(self.petId)
+
+    @property
+    def device(self):
+        return self.pet.device
+
+    @property
+    def device_id(self):
+        return self.unique_id
+
+    @property
+    def device_class(self):
+        """Return the device class of the sensor."""
+        return None
+
+    @property
+    def icon(self):
+        if self.statType == "Activity Type":
+            return "mdi:run"
+        elif self.statType == "Current Place Name":
+            return "mdi:earth"
+        elif self.statType == "Current Place Address":
+            return "mdi:map-marker"
+
+    @property
+    def state(self):
+        if self.statType == "Activity Type":
+            return self.pet.getActivityType()
+        elif self.statType == "Current Place Name":
+            return self.pet.getCurrPlaceName()
+        elif self.statType == "Current Place Address":
+            return self.pet.getCurrPlaceAddress()
+    @property
+    def unit_of_measurement(self):
+        return None
+
+    @property
+    def device_info(self):
+        return {
+            "identifiers": {(DOMAIN, self.pet.petId)},
+            "name": self.pet.name,
+            "manufacturer": "TryFi",
+            "model": self.pet.breed,
+            "sw_version": self.pet.device.buildId,
+        }
 
 class PetStatsSensor(CoordinatorEntity, Entity):
     """Representation of a Sensor."""