pypetkitapi 1.7.4__py3-none-any.whl → 1.7.5__py3-none-any.whl

pypetkitapi/client.py

@@ -206,6 +206,7 @@ class PetKitClient:
             if account.device_list:
                 _LOGGER.debug("Devices in account: %s", account.device_list)
                 device_list.extend(account.device_list)
+                _LOGGER.debug("Found %s devices", len(account.device_list))
 
         for device in device_list:
             device_type = device.device_type.lower()
@@ -259,6 +260,8 @@ class PetKitClient:
             | FeederRecord
             | LitterRecord
             | WaterFountainRecord
+            | PetOutGraph
+            | LitterStats
         ],
     ) -> None:
         """Fetch the device data from the PetKit servers."""
@@ -345,16 +348,14 @@ class PetKitClient:
             return 0
         return end - start
 
-    async def populate_pet_stats(self, stats_data: Litter | None) -> None:
+    async def populate_pet_stats(self, stats_data: Litter) -> None:
         """Collect data from litter data to populate pet stats."""
-        if stats_data is None:
-            return
 
         pets_list = await self.get_pets_list()
         for pet in pets_list:
-            if stats_data.device_records:
+            if stats_data.device_type == T4 and stats_data.device_records:
                 await self._process_t4(pet, stats_data.device_records)
-            elif stats_data.device_pet_graph_out:
+            elif stats_data.device_type == T6 and stats_data.device_pet_graph_out:
                 await self._process_t6(pet, stats_data.device_pet_graph_out)
 
     async def _process_t4(self, pet, device_records):

pypetkitapi/const.py

@@ -31,7 +31,7 @@ CTW3 = "ctw3"
 K2 = "k2"
 
 DEVICES_LITTER_BOX = [T3, T4, T5, T6]
-DEVICES_FEEDER = [FEEDER, FEEDER_MINI, D4, D4S, D4H, D4SH]
+DEVICES_FEEDER = [FEEDER, FEEDER_MINI, D3, D4, D4S, D4H, D4SH]
 DEVICES_WATER_FOUNTAIN = [W5, CTW3]
 ALL_DEVICES = [*DEVICES_LITTER_BOX, *DEVICES_FEEDER, *DEVICES_WATER_FOUNTAIN]
 
@@ -131,3 +131,4 @@ class PetkitEndpoint(StrEnum):
     MANUAL_FEED_MINI = "feedermini/save_dailyfeed"
     MANUAL_FEED_FRESH_ELEMENT = "feeder/save_dailyfeed"
     MANUAL_FEED_DUAL = "saveDailyFeed"
+    DAILY_FEED_AND_EAT = "dailyFeedAndEat"

pypetkitapi/containers.py

@@ -70,7 +70,7 @@ class Pet(BaseModel):
     pet_id: int = Field(alias="petId")
     pet_name: str | None = Field(None, alias="petName")
     id: int | None = None  # Fictive field (for HA compatibility) copied from id
-    sn: int | None = None  # Fictive field (for HA compatibility) copied from id
+    sn: str  # Fictive field (for HA compatibility) copied from id
     name: str | None = None  # Fictive field (for HA compatibility) copied from pet_name
     device_type: str = "pet"  # Fictive field (for HA compatibility) fixed
     firmware: str | None = None  # Fictive field (for HA compatibility) fixed

pypetkitapi/feeder_container.py

@@ -5,7 +5,7 @@ from typing import Any, ClassVar
 
 from pydantic import BaseModel, Field
 
-from pypetkitapi.const import DEVICE_DATA, DEVICE_RECORDS, PetkitEndpoint
+from pypetkitapi.const import D3, DEVICE_DATA, DEVICE_RECORDS, PetkitEndpoint
 from pypetkitapi.containers import CloudProduct, Device, FirmwareDetail, Wifi
 
 
@@ -89,6 +89,7 @@ class SettingsFeeder(BaseModel):
     selected_sound: int | None = Field(None, alias="selectedSound")
     smart_frame: int | None = Field(None, alias="smartFrame")
     sound_enable: int | None = Field(None, alias="soundEnable")
+    surplus: int | None = None  # D3
     surplus_control: int | None = Field(None, alias="surplusControl")
     surplus_standard: int | None = Field(None, alias="surplusStandard")
     system_sound_enable: int | None = Field(None, alias="systemSoundEnable")
@@ -113,7 +114,7 @@ class FeedState(BaseModel):
     eat_avg: int | None = Field(None, alias="eatAvg")
     eat_count: int | None = Field(None, alias="eatCount")
     eat_times: list[int] | None = Field(None, alias="eatTimes")
-    feed_times: dict | None = Field(None, alias="feedTimes")
+    feed_times: dict | list | None = Field(None, alias="feedTimes")
     times: int | None = None
     add_amount_total: int | None = Field(None, alias="addAmountTotal")
     plan_amount_total: int | None = Field(None, alias="planAmountTotal")
@@ -135,17 +136,24 @@ class StateFeeder(BaseModel):
     battery_power: int | None = Field(None, alias="batteryPower")
     battery_status: int | None = Field(None, alias="batteryStatus")
     bowl: int | None = None
+    block: int | None = None
+    broadcast: dict | None = None
     camera_status: int | None = Field(None, alias="cameraStatus")
+    charge: int | None = None
     desiccant_left_days: int | None = Field(None, alias="desiccantLeftDays")
     desiccant_time: int | None = Field(None, alias="desiccantTime")
     door: int | None = None
     feed_state: FeedState | None = Field(None, alias="feedState")
     feeding: int | None = None
+    error_code: str | None = Field(None, alias="errorCode")
+    error_detail: str | None = Field(None, alias="errorDetail")
+    error_level: int | None = Field(None, alias="errorLevel")
     error_msg: str | None = Field(None, alias="errorMsg")
     ota: int | None = None
     overall: int | None = None
     pim: int | None = None
     runtime: int | None = None
+    weight: int | None = None
     wifi: Wifi | None = None
     eating: int | None = None
     food: int | None = None
@@ -198,6 +206,7 @@ class RecordsItems(BaseModel):
     eat_end_time: int | None = Field(None, alias="eatEndTime")
     eat_start_time: int | None = Field(None, alias="eatStartTime")
     eat_video: int | None = Field(None, alias="eatVideo")
+    eat_weight: int | None = Field(None, alias="eatWeight")  # D3
     empty: int | None = None
     end_time: int | None = Field(None, alias="endTime")
     enum_event_type: str | None = Field(None, alias="enumEventType")
@@ -210,10 +219,12 @@ class RecordsItems(BaseModel):
     id: str | None = None
     is_executed: int | None = Field(None, alias="isExecuted")
     is_need_upload_video: int | None = Field(None, alias="isNeedUploadVideo")
+    left_weight: int | None = Field(None, alias="leftWeight")  # D3
     mark: int | None = None
     media_api: str | None = Field(None, alias="mediaApi")
     media_list: list[Any] | None = Field(None, alias="mediaList")
     name: str | None = None
+    pet_id: str | None = Field(None, alias="petId")
     preview: str | None = None
     preview1: str | None = Field(None, alias="preview1")
     preview2: str | None = Field(None, alias="preview2")
@@ -235,11 +246,13 @@ class RecordsType(BaseModel):
     day: int | None = None
     device_id: int | None = Field(None, alias="deviceId")
     eat_count: int | None = Field(None, alias="eatCount")
+    eat_amount: int | None = Field(None, alias="eatAmount")  # D3
     items: list[RecordsItems] | None = None
     plan_amount: int | None = Field(None, alias="planAmount")
     real_amount: int | None = Field(None, alias="realAmount")
     amount: int | None = None
     times: int | None = None
+    user_id: str | None = Field(None, alias="userId")  # D3
 
 
 class FeederRecord(BaseModel):
@@ -254,8 +267,10 @@ class FeederRecord(BaseModel):
     device_type: str | None = Field(None, alias="deviceType")
 
     @classmethod
-    def get_endpoint(cls, device_type: str) -> str:
+    def get_endpoint(cls, device_type: str) -> str | None:
         """Get the endpoint URL for the given device type."""
+        if device_type == D3:
+            return PetkitEndpoint.DAILY_FEED_AND_EAT
         return PetkitEndpoint.GET_DEVICE_RECORD
 
     @classmethod

pypetkitapi/litter_container.py

@@ -109,6 +109,9 @@ class StateLitter(BaseModel):
     box_full: bool | None = Field(None, alias="boxFull")
     box_state: int | None = Field(None, alias="boxState")
     deodorant_left_days: int | None = Field(None, alias="deodorantLeftDays")
+    error_code: str | None = Field(None, alias="errorCode")
+    error_detail: str | None = Field(None, alias="errorDetail")
+    error_level: int | None = Field(None, alias="errorLevel")
     error_msg: str | None = Field(None, alias="errorMsg")
     frequent_restroom: int | None = Field(None, alias="frequentRestroom")
     liquid_empty: bool | None = Field(None, alias="liquidEmpty")

pypetkitapi/medias.py

@@ -41,28 +41,30 @@ async def extract_filename_from_url(url: str) -> str:
 class MediaHandler:
     """Class to find media files from PetKit devices."""
 
-    def __init__(self, device: Feeder, file_path: str):
+    def __init__(self, file_path: Path):
         """Initialize the class."""
-        self.device = device
         self.media_download_decode = MediaDownloadDecode(file_path)
         self.media_files: list[MediasFiles] = []
 
-    async def get_last_image(self) -> list[MediasFiles]:
+    async def get_last_image(self, device: Feeder) -> list[MediasFiles]:
         """Process device records and extract media info."""
         record_types = ["eat", "feed", "move", "pet"]
         self.media_files = []
 
-        if not self.device.device_records:
+        if not isinstance(device, Feeder):
+            _LOGGER.error("Device is not a Feeder")
+            return []
+
+        if not device.device_records:
             _LOGGER.error("No device records found for feeder")
             return []
 
         for record_type in record_types:
-            records = getattr(self.device.device_records, record_type, None)
+            records = getattr(device.device_records, record_type, None)
             if records:
                 self.media_files.extend(
                     await self._process_records(records, record_type)
                 )
-
         return self.media_files
 
     async def _process_records(
@@ -111,45 +113,39 @@ class MediaHandler:
 class MediaDownloadDecode:
     """Class to download"""
 
-    def __init__(self, download_path: str):
+    def __init__(self, download_path: Path):
         """Initialize the class."""
         self.download_path = download_path
 
     async def get_file(self, url: str, aes_key: str) -> bool:
         """Download a file from a URL and decrypt it."""
-        try:
-            # Check if the file already exists
-            filename = await extract_filename_from_url(url)
-            full_file_path = Path(self.download_path) / filename
-            if full_file_path.exists():
-                _LOGGER.debug(
-                    "File already exist : %s don't need to download it", filename
+        # Check if the file already exists
+        filename = await extract_filename_from_url(url)
+        full_file_path = Path(self.download_path) / filename
+        if full_file_path.exists():
+            _LOGGER.debug("File already exist : %s don't need to download it", filename)
+            return True
+
+        # Download the file
+        async with aiohttp.ClientSession() as session, session.get(url) as response:
+            if response.status != 200:
+                _LOGGER.error(
+                    "Failed to download %s, status code: %s", url, response.status
                 )
-                return True
+                return False
 
-            # Download the file
-            async with aiohttp.ClientSession() as session, session.get(url) as response:
-                if response.status != 200:
-                    _LOGGER.error(
-                        "Failed to download %s, status code: %s", url, response.status
-                    )
-                    return False
+            content = await response.read()
 
-                content = await response.read()
-                encrypted_file_path = await self._save_file(content, f"{filename}.enc")
-                # Decrypt the image
-                decrypted_data = await self._decrypt_image_from_file(
-                    encrypted_file_path, aes_key
-                )
+        encrypted_file_path = await self._save_file(content, f"{filename}.enc")
+        # Decrypt the image
+        decrypted_data = await self._decrypt_image_from_file(
+            encrypted_file_path, aes_key
+        )
 
-                if decrypted_data:
-                    _LOGGER.debug("Decrypt was successful")
-                    await self._save_file(decrypted_data, filename)
-                    Path(encrypted_file_path).unlink()
-                    return True
-                _LOGGER.error("Failed to decrypt %s", encrypted_file_path)
-        except Exception as e:  # noqa: BLE001
-            _LOGGER.error("Error get media file from %s: %s", url, e)
+        if decrypted_data:
+            _LOGGER.debug("Decrypt was successful")
+            await self._save_file(decrypted_data, filename)
+            return True
         return False
 
     async def _save_file(self, content: bytes, filename: str) -> Path:
@@ -198,4 +194,5 @@ class MediaDownloadDecode:
         except Exception as e:  # noqa: BLE001
             logging.error("Error decrypting image from file %s: %s", file_path, e)
             return None
+        Path(file_path).unlink()
         return decrypted_data

{pypetkitapi-1.7.4.dist-info → pypetkitapi-1.7.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pypetkitapi
-Version: 1.7.4
+Version: 1.7.5
 Summary: Python client for PetKit API
 Home-page: https://github.com/Jezza34000/pypetkit
 License: MIT
@@ -15,7 +15,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: aiofiles (>=24.1.0,<25.0.0)
 Requires-Dist: aiohttp (>=3.10.10,<4.0.0)
 Requires-Dist: pycryptodome (>=3.19.1,<4.0.0)
-Requires-Dist: pydantic (>=1.10.19,<2.0.0)
+Requires-Dist: pydantic (>=1.10.18,<2.0.0)
 Description-Content-Type: text/markdown
 
 # Petkit API Client

pypetkitapi-1.7.5.dist-info/RECORD

@@ -0,0 +1,15 @@
+pypetkitapi/__init__.py,sha256=eVpyGMD3tkYtiHUkdKEeNSZhQlZ4woI2Y5oVoV7CwXM,61
+pypetkitapi/client.py,sha256=aBfwh6O-egpaX7vhW1I2l3flnXL_XrwGraH1uOzmH-w,19888
+pypetkitapi/command.py,sha256=gw3_J_oZHuuGLk66P8uRSqSrySjYa8ArpKaPHi2ybCw,7155
+pypetkitapi/const.py,sha256=ZpFWBgzb3nvy0Z4oyM550cCunlIceWgXqqtwtJc3mFo,3479
+pypetkitapi/containers.py,sha256=nhp50QwyoQRveTnEgWL7JFEY3Tl5m5wp9EqZvklW87Y,4338
+pypetkitapi/exceptions.py,sha256=NWmpsI2ewC4HaIeu_uFwCeuPIHIJxZBzjoCP7aNwvhs,1139
+pypetkitapi/feeder_container.py,sha256=y1A5WhObXCdbcWwtKgSPzTWokYchoUDlholAD-AMgGQ,14069
+pypetkitapi/litter_container.py,sha256=wsM-v7ibx17lsD-o17RGz2wvcHCUu1iZ2RqAh3CN1Qc,18254
+pypetkitapi/medias.py,sha256=8hrMdzFR9d0L0PQOVYdy-MReSF9GMp8Ft0IGGHtL1Ag,6904
+pypetkitapi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pypetkitapi/water_fountain_container.py,sha256=3F4GP5pXJqq6kxLMSK__GgFMuZ-rz1VDIIhaVd19Kl8,6781
+pypetkitapi-1.7.5.dist-info/LICENSE,sha256=4FWnKolNLc1e3w6cVlT61YxfPh0DQNeQLN1CepKKSBg,1067
+pypetkitapi-1.7.5.dist-info/METADATA,sha256=bvYlZOdBLw9_nu53rnNrPSXvgYdSvQMjqAVeSFMm7VE,4852
+pypetkitapi-1.7.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pypetkitapi-1.7.5.dist-info/RECORD,,

pypetkitapi-1.7.4.dist-info/RECORD

@@ -1,15 +0,0 @@
-pypetkitapi/__init__.py,sha256=eVpyGMD3tkYtiHUkdKEeNSZhQlZ4woI2Y5oVoV7CwXM,61
-pypetkitapi/client.py,sha256=h2HMfUSlfa5ppGn7ev3VVxlv0-T7FCqHGGDyP6DoNBw,19751
-pypetkitapi/command.py,sha256=gw3_J_oZHuuGLk66P8uRSqSrySjYa8ArpKaPHi2ybCw,7155
-pypetkitapi/const.py,sha256=9XNLhM9k0GwNmWPgGef5roULpsYVZ7hzxptGgNhjs74,3432
-pypetkitapi/containers.py,sha256=8HA2LEzHga062LcC08c9JCowyDRmjjf8en4ImZ6tg_4,4352
-pypetkitapi/exceptions.py,sha256=NWmpsI2ewC4HaIeu_uFwCeuPIHIJxZBzjoCP7aNwvhs,1139
-pypetkitapi/feeder_container.py,sha256=28GXZ8Nbs08PnFZZI4ENBe3UJ63gsXT3rFa151KrPxo,13310
-pypetkitapi/litter_container.py,sha256=fTPHa5VQvM6Ycm-XIeu3N1E6dTI4g1zEqC-o23sxdhE,18068
-pypetkitapi/medias.py,sha256=1Nso1YZK_Bz4MrSuQ_zjawoyg5KqO_JoBSQaMmvh6Jo,7216
-pypetkitapi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pypetkitapi/water_fountain_container.py,sha256=3F4GP5pXJqq6kxLMSK__GgFMuZ-rz1VDIIhaVd19Kl8,6781
-pypetkitapi-1.7.4.dist-info/LICENSE,sha256=4FWnKolNLc1e3w6cVlT61YxfPh0DQNeQLN1CepKKSBg,1067
-pypetkitapi-1.7.4.dist-info/METADATA,sha256=zOLLrY09rRjinO6hen7UAiBhi1diVDMz3ISu9ywTB04,4852
-pypetkitapi-1.7.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pypetkitapi-1.7.4.dist-info/RECORD,,