pypetkitapi 1.11.2__py3-none-any.whl → 1.11.3__py3-none-any.whl

pypetkitapi/__init__.py

@@ -51,7 +51,7 @@ from .media import DownloadDecryptMedia, MediaCloud, MediaFile, MediaManager
 from .purifier_container import Purifier
 from .water_fountain_container import WaterFountain
 
-__version__ = "1.11.2"
+__version__ = "1.11.3"
 
 __all__ = [
     "CTW3",

pypetkitapi/media.py

@@ -270,9 +270,7 @@ class MediaManager:
         device_type = (
             device_obj.device_nfo.device_type if device_obj.device_nfo else None
         )
-        cp_sub = (
-            device_obj.cloud_product.subscribe if device_obj.cloud_product else None
-        )
+        cp_sub = self.is_subscription_active(device_obj)
 
         if not feeder_id:
             _LOGGER.warning("Missing feeder_id for record")
@@ -299,7 +297,7 @@ class MediaManager:
                 _LOGGER.warning("Missing user_id for record item")
                 continue
             if not item.aes_key:
-                _LOGGER.warning("Missing aes_key for record item")
+                _LOGGER.debug("Missing aes_key for record item")
                 continue
 
             date_str = await self.get_date_from_ts(timestamp)
@@ -331,7 +329,7 @@ class MediaManager:
         litter_id = litter.device_nfo.device_id if litter.device_nfo else None
         device_type = litter.device_nfo.device_type if litter.device_nfo else None
         user_id = litter.user.id if litter.user else None
-        cp_sub = litter.cloud_product.subscribe if litter.cloud_product else None
+        cp_sub = self.is_subscription_active(litter)
 
         if not litter_id:
             _LOGGER.warning("Missing litter_id for record")
@@ -383,6 +381,19 @@ class MediaManager:
             )
         return media_files
 
+    @staticmethod
+    def is_subscription_active(device: Feeder | Litter) -> bool:
+        """Check if the subscription is active based on the work_indate timestamp.
+        :param device: Device object
+        :return: True if the subscription is active, False otherwise
+        """
+        if device.cloud_product and device.cloud_product.work_indate:
+            return (
+                datetime.fromtimestamp(device.cloud_product.work_indate)
+                > datetime.now()
+            )
+        return False
+
     @staticmethod
     async def get_date_from_ts(timestamp: int | None) -> str:
         """Get date from timestamp.
@@ -395,7 +406,10 @@ class MediaManager:
 
     @staticmethod
     async def construct_video_url(
-        device_type: str | None, media_url: str | None, user_id: int, cp_sub: int | None
+        device_type: str | None,
+        media_url: str | None,
+        user_id: int,
+        cp_sub: bool | None,
     ) -> str | None:
         """Construct the video URL.
         :param device_type: Device type
@@ -404,7 +418,7 @@ class MediaManager:
         :param cp_sub: Cpsub value
         :return: Constructed video URL
         """
-        if not media_url or not user_id or cp_sub != 1:
+        if not media_url or not user_id or not cp_sub:
             return None
         params = parse_qs(urlparse(media_url).query)
         param_dict = {k: v[0] for k, v in params.items()}
@@ -539,7 +553,7 @@ class DownloadDecryptMedia:
 
         media_api = video_data.get("mediaApi", None)
         if not media_api:
-            _LOGGER.warning("Missing mediaApi in video data")
+            _LOGGER.debug("Missing mediaApi in video data")
             raise ValueError("Missing mediaApi in video data")
         return await self.client.extract_segments_m3u8(str(media_api))
 

{pypetkitapi-1.11.2.dist-info → pypetkitapi-1.11.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pypetkitapi
-Version: 1.11.2
+Version: 1.11.3
 Summary: Python client for PetKit API
 License: MIT
 Author: Jezza34000

{pypetkitapi-1.11.2.dist-info → pypetkitapi-1.11.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pypetkitapi/__init__.py,sha256=qDhET_P65lnwbZgZnkYst3IgJxPlID7fLqYnfVSzJqM,2107
+pypetkitapi/__init__.py,sha256=Ta6BJBUyI7-C8kUXHJ1nym7-oXdmKWxXn1OfCssy9U8,2107
 pypetkitapi/bluetooth.py,sha256=eu6c2h6YHBafAhhSSy4As6tn29i5WbOH9tZzRlMm44U,7843
 pypetkitapi/client.py,sha256=wzZQUHg_ee6lmdAjli6zS7qw_sgPER9iLfcTZe4VTH4,27190
 pypetkitapi/command.py,sha256=cMCUutZCQo9Ddvjl_FYR5UjU_CqFz1iyetMznYwjpzM,7500
@@ -7,13 +7,13 @@ pypetkitapi/containers.py,sha256=F_uyDBD0a5QD4s_ArjYiKTAAg1XHYBvmV_lEnO9RQ-U,478
 pypetkitapi/exceptions.py,sha256=4BXUyYXLfZjNxdnOGJPjyE9ASIl7JmQphjws87jvHtE,1631
 pypetkitapi/feeder_container.py,sha256=PhidWd5WpsZqtdKZy60PzE67YXgQfApjm8CqvMCHK3U,14743
 pypetkitapi/litter_container.py,sha256=KWvHNAOJ6hDSeJ_55tqtzY9GxHtd9gAntPkbnVbdb-I,19275
-pypetkitapi/media.py,sha256=34q3pIfejO_z-Uk8ohQlOGaPOdaGtnI0zN1LtZ3k3F4,25904
+pypetkitapi/media.py,sha256=aGl4nw8q8UK5J2pDEH-lP-Jhy4RWFAqWfyZaLeX3wKM,26356
 pypetkitapi/purifier_container.py,sha256=ssyIxhNben5XJ4KlQTXTrtULg2ji6DqHqjzOq08d1-I,2491
 pypetkitapi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pypetkitapi/schedule_container.py,sha256=OjLAY6FY-g14JNJJnYMNFV5ZtdkjUzNBit1VUiiZKnQ,2053
 pypetkitapi/utils.py,sha256=z7325kcJQUburnF28HSXrJMvY_gY9007K73Zwxp-4DQ,743
 pypetkitapi/water_fountain_container.py,sha256=5J0b-fDZYcFLNX2El7fifv8H6JMhBCt-ttxSow1ozRQ,6787
-pypetkitapi-1.11.2.dist-info/LICENSE,sha256=u5jNkZEn6YMrtN4Kr5rU3TcBJ5-eAt0qMx4JDsbsnzM,1074
-pypetkitapi-1.11.2.dist-info/METADATA,sha256=8HbQJ6UeGion-oO5Y5U0X7THfeDRYeP-EK-XVwF_Lb8,6338
-pypetkitapi-1.11.2.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-pypetkitapi-1.11.2.dist-info/RECORD,,
+pypetkitapi-1.11.3.dist-info/LICENSE,sha256=u5jNkZEn6YMrtN4Kr5rU3TcBJ5-eAt0qMx4JDsbsnzM,1074
+pypetkitapi-1.11.3.dist-info/METADATA,sha256=Mqjd4OtD4X539bRYn15IxcRzz2nOMYNEj0jQRu_WnyU,6338
+pypetkitapi-1.11.3.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+pypetkitapi-1.11.3.dist-info/RECORD,,