eaf-base-api 2.3.4__tar.gz → 2.3.6__tar.gz

{eaf_base_api-2.3.4/eaf_base_api.egg-info → eaf_base_api-2.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eaf_base_api
-Version: 2.3.4
+Version: 2.3.6
 Summary: A base API for EchterAlsFake's Porn APIs
 Home-page: https://github.com/EchterAlsFake/eaf_base_api
 Author: Johannes Habel

{eaf_base_api-2.3.4 → eaf_base_api-2.3.6}/base_api/base.py

@@ -326,7 +326,26 @@ class BaseCore:
 
                 # Process and return content
                 # Collecting all chunks before processing because some sites cause issues with real-time decoding
-                raw_content = response.content
+                if self.config.max_bandwidth_mb is not None and self.config.max_bandwidth_mb >= 0.2:
+                    raw_content = bytearray()
+                    chunk_size = 64 * 1024 # 64 KB
+                    speed_limit = self.config.max_bandwidth_mb * 1024 * 1024
+                    min_time_per_chunk = chunk_size / speed_limit
+                    start_time = time.time()
+
+                    for chunk in response.iter_bytes(chunk_size=chunk_size):
+                        raw_content.extend(chunk)
+                        elapsed = time.time() - start_time
+                        sleep_time = min_time_per_chunk - elapsed
+                        if sleep_time > 0:
+                            time.sleep(sleep_time)
+
+                        start_time = time.time()
+
+                    raw_content = bytes(raw_content)
+
+                else:
+                    raw_content = response.content
 
                 if get_bytes:
                     content = raw_content

{eaf_base_api-2.3.4 → eaf_base_api-2.3.6}/base_api/modules/config.py

@@ -6,6 +6,7 @@ class RuntimeConfig:
         self.max_retries = 4
         self.request_delay = 0
         self.timeout = 20
+        self.max_bandwidth_mb = None # Set speed limit in megabytes per second e.g, 2.0, 3.5 etc...
         self.ffmpeg_path = "ffmpeg"
         self.proxy = None
         self.verify_ssl = True

{eaf_base_api-2.3.4 → eaf_base_api-2.3.6/eaf_base_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eaf_base_api
-Version: 2.3.4
+Version: 2.3.6
 Summary: A base API for EchterAlsFake's Porn APIs
 Home-page: https://github.com/EchterAlsFake/eaf_base_api
 Author: Johannes Habel

{eaf_base_api-2.3.4 → eaf_base_api-2.3.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="eaf_base_api",
-    version="2.3.4",
+    version="2.3.6",
     packages=find_packages(),
     install_requires=["m3u8", "httpx"],
     extras_require={