pypetkitapi 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

pypetkitapi/client.py

@@ -19,6 +19,7 @@ from pypetkitapi.const import (
     DEVICES_WATER_FOUNTAIN,
     ERR_KEY,
     LOGIN_DATA,
+    PET_DATA,
     RES_KEY,
     SUCCESS_KEY,
     Header,
@@ -48,10 +49,11 @@ class PetKitClient:
     _session: SessionInfo | None = None
     _servers_list: list[RegionInfo] = []
     account_data: list[AccountData] = []
-    petkit_entities: dict[int, Feeder | Litter | WaterFountain | Pet] = {}
-    petkit_entities_records: dict[
-        int, FeederRecord | LitterRecord | WaterFountainRecord
-    ] = {}
+    petkit_entities: dict[
+        str,
+        dict[int, Feeder | Litter | WaterFountain | Pet]
+        | dict[int, FeederRecord | LitterRecord | WaterFountainRecord],
+    ]
 
     def __init__(
         self,
@@ -67,6 +69,7 @@ class PetKitClient:
         self.region = region.lower()
         self.timezone = timezone
         self._session = None
+        self.petkit_entities = {DEVICE_RECORDS: {}, DEVICE_DATA: {}, PET_DATA: {}}
         self.aiohttp_session = session or aiohttp.ClientSession()
         self.req = PrepReq(
             base_url=PetkitDomain.PASSPORT_PETKIT, session=self.aiohttp_session
@@ -169,7 +172,6 @@ class PetKitClient:
         elif half_max_age < token_age <= max_age:
             _LOGGER.debug("Token still OK, but refreshing session")
             await self.refresh_session()
-        return
 
     async def get_session_id(self) -> dict:
         """Return the session ID."""
@@ -192,7 +194,7 @@ class PetKitClient:
         for account in self.account_data:
             if account.pet_list:
                 for pet in account.pet_list:
-                    self.petkit_entities[pet.pet_id] = pet
+                    self.petkit_entities[PET_DATA][pet.pet_id] = pet
 
     async def get_devices_data(self) -> None:
         """Get the devices data from the PetKit servers."""
@@ -300,12 +302,7 @@ class PetKitClient:
 
         _LOGGER.debug("Reading device type : %s (id=%s)", device_type, device_id)
 
-        if data_class.data_type == DEVICE_DATA:
-            self.petkit_entities[device_id] = device_data
-        elif data_class.data_type == DEVICE_RECORDS:
-            self.petkit_entities_records[device_id] = device_data
-        else:
-            _LOGGER.error("Unknown data type: %s", data_class.data_type)
+        self.petkit_entities[data_class.data_type][device_id] = device_data
 
     async def send_api_request(
         self,
@@ -314,7 +311,8 @@ class PetKitClient:
         setting: dict | None = None,
     ) -> None:
         """Control the device using the PetKit API."""
-        device = self.petkit_entities.get(device_id)
+        device_dict = self.petkit_entities.get(DEVICE_DATA, {})
+        device = device_dict.get(device_id)
         if not device:
             raise PypetkitError(f"Device with ID {device_id} not found.")
 

pypetkitapi/const.py

@@ -11,6 +11,7 @@ SUCCESS_KEY = "success"
 
 DEVICE_RECORDS = "deviceRecords"
 DEVICE_DATA = "deviceData"
+PET_DATA = "petData"
 
 # PetKit Models
 FEEDER = "feeder"

{pypetkitapi-1.0.0.dist-info → pypetkitapi-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pypetkitapi
-Version: 1.0.0
+Version: 1.1.0
 Summary: Python client for PetKit API
 Home-page: https://github.com/Jezza34000/pypetkit
 License: MIT

{pypetkitapi-1.0.0.dist-info → pypetkitapi-1.1.0.dist-info}/RECORD

@@ -1,14 +1,14 @@
 pypetkitapi/__init__.py,sha256=eVpyGMD3tkYtiHUkdKEeNSZhQlZ4woI2Y5oVoV7CwXM,61
-pypetkitapi/client.py,sha256=RCt3vxNJ14NKzwUpvL_M6xSYC4DfxEyDu7B8sJ7HFz0,16591
+pypetkitapi/client.py,sha256=lC5kWLDcZcB5WHZeoyg0RagEKzK6F0Vk_H8Be_1MyDM,16493
 pypetkitapi/command.py,sha256=gw3_J_oZHuuGLk66P8uRSqSrySjYa8ArpKaPHi2ybCw,7155
-pypetkitapi/const.py,sha256=EE84cCOEE6AMv3QuDJa8-FreCffhfq8nxPxqBpvB5XY,3270
+pypetkitapi/const.py,sha256=Mxgezd9sy-LtpgjYuVKb9ii9HSUnz8FVyKx1GXbXBAU,3291
 pypetkitapi/containers.py,sha256=m7vzWcJG0U1EPftuBF6OB8eTVRhCoA2DFqekxI6LozI,3428
 pypetkitapi/exceptions.py,sha256=NWmpsI2ewC4HaIeu_uFwCeuPIHIJxZBzjoCP7aNwvhs,1139
 pypetkitapi/feeder_container.py,sha256=XpZqeQ8E4_ud3juxDBKGTMfmCfvc9QYcJInacmbWK-o,12785
 pypetkitapi/litter_container.py,sha256=TK2-_yg3EmT0RQzMLurud2PHwa2P-o2KGyV9m_SQ7qw,11651
 pypetkitapi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pypetkitapi/water_fountain_container.py,sha256=xmqYuxyppRLhqi7cF4JNRprhgmz3KKrbi4iv942wNaU,6566
-pypetkitapi-1.0.0.dist-info/LICENSE,sha256=4FWnKolNLc1e3w6cVlT61YxfPh0DQNeQLN1CepKKSBg,1067
-pypetkitapi-1.0.0.dist-info/METADATA,sha256=t1zmfg9-IZqpzBWybjhVTExCgkVJ50Vu8W-Twx8njiE,4590
-pypetkitapi-1.0.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pypetkitapi-1.0.0.dist-info/RECORD,,
+pypetkitapi-1.1.0.dist-info/LICENSE,sha256=4FWnKolNLc1e3w6cVlT61YxfPh0DQNeQLN1CepKKSBg,1067
+pypetkitapi-1.1.0.dist-info/METADATA,sha256=9OvaTvc__catDl-YG3HWqk9ycquCVymNfAS6TrEQtBk,4590
+pypetkitapi-1.1.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pypetkitapi-1.1.0.dist-info/RECORD,,