eaf_base_api 3.2.1__tar.gz → 3.2.3__tar.gz

{eaf_base_api-3.2.1 → eaf_base_api-3.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eaf_base_api
-Version: 3.2.1
+Version: 3.2.3
 Summary: A base API for EchterAlsFake's Porn APIs
 Author: Johannes Habel
 Author-email: Johannes Habel <EchterAlsFake@proton.me>

{eaf_base_api-3.2.1 → eaf_base_api-3.2.3}/base_api/base.py

@@ -259,6 +259,12 @@ class Helper:
             if asyncio.iscoroutine(video_instance):
                 video_instance = await video_instance
 
+            # Automatically call and await the init method if available (async initialization)
+            if hasattr(video_instance, "init") and callable(video_instance.init):
+                init_result = video_instance.init()
+                if asyncio.iscoroutine(init_result):
+                    await init_result
+
             elapsed_ms = (time.perf_counter() - start_timestamp) * 1000
             logger.debug("video_init ok url=%s (%.2f ms)", video_url, elapsed_ms)
             return video_instance
@@ -1196,6 +1202,10 @@ a new Python file, import only m3u8 and see what error you get.
             self.logger.debug("download: no callback provided, using default text progress bar")
 
         m3u8_url = getattr(video, "m3u8_base_url", None)
+
+        if asyncio.iscoroutine(m3u8_url):
+            m3u8_url = await m3u8_url # For youporn api
+
         self.logger.info(
     """
     Download requested: quality=%s path=%s remux=%s max_workers=%s
@@ -1437,7 +1447,7 @@ a new Python file, import only m3u8 and see what error you get.
 
                                     try:
                                         _, segment_data, is_success = await self.download_segment(url, timeout, stop_event)
-                                        if is_success and data:
+                                        if is_success and segment_data:
                                             return idx, True, segment_data
                                     except Exception as exception:
                                         self.logger.error(f"Worker exception for segment {idx}: {exception}")

{eaf_base_api-3.2.1 → eaf_base_api-3.2.3}/base_api/modules/type_hints.py

@@ -18,6 +18,9 @@ class DownloadState:
     missing: list[int]
     segments: list[str]
 
+    def __getitem__(self, key: str) -> Any:
+        return getattr(self, key)
+
 
 # Download state is used for the literal file that tracks it
 @dataclass
@@ -32,3 +35,7 @@ class DownloadReport:
     start_segment: int
     quality: str | int
 
+    def __getitem__(self, key: str) -> Any:
+        return getattr(self, key)
+
+

{eaf_base_api-3.2.1 → eaf_base_api-3.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "eaf_base_api"
-version = "3.2.1"
+version = "3.2.3"
 description = "A base API for EchterAlsFake's Porn APIs"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
@@ -46,5 +46,7 @@ module-name = "base_api" # Because I named it differently we need to change here
 dev = [
     "mypy",
     "types-m3u8",
-    "pylint"
+    "pylint",
+    "pytest",
+    "pytest-asyncio"
 ]