eaf-base-api 1.0__tar.gz → 1.2__tar.gz

{eaf_base_api-1.0 → eaf_base_api-1.2}/PKG-INFO

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

{eaf_base_api-1.0 → eaf_base_api-1.2}/base_api/base.py

@@ -5,6 +5,8 @@ import time
 from base_api.modules.quality import Quality
 from base_api.modules.progress_bars import Callback
 from base_api.modules.download import default, threaded, FFMPEG
+from base_api.modules.consts import MAX_RETRIES
+from pathlib import Path
 
 
 def setup_api(do_logging=False):
@@ -21,8 +23,8 @@ base_qualities = ["250p", "360p", "480p", "720p", "1080p", "1440p", "2160p"]
 
 class Core:
     @classmethod
-    def get_content(cls, url, headers=None, cookies=None, retries=4):
-        for i in range(retries):
+    def get_content(cls, url, headers=None, cookies=None):
+        for i in range(MAX_RETRIES):
             try:
                 logging.debug(f"Trying to fetch {url} / Attempt: [{i+1}]")
                 response = requests.get(url, headers=headers, cookies=cookies)
@@ -33,9 +35,9 @@ class Core:
                     logging.warning(f"Failed to fetch {url} with status code {response.status_code} on attempt [{i+1}]")
             except requests.exceptions.RequestException as e:
                 logging.error(f"Exception occurred when trying to fetch {url} on attempt [{i+1}]: {e}")
-            if i < retries - 1:  # Implement exponential backoff
+            if i < MAX_RETRIES - 1:  # Implement exponential backoff
                 time.sleep(2 ** i)
-        logging.error(f"Failed to fetch {url} after {retries} attempts.")
+        logging.error(f"Failed to fetch {url} after {MAX_RETRIES} attempts.")
         logging.error("Returning None")
         return None
 
@@ -86,12 +88,12 @@ class Core:
 
         return self.quality_url_map.get(selected_quality)
 
-    def download(self, video, quality, downloader, output_path, callback=None):
+    def download(self, video, quality, downloader, path, callback=None):
         """
         :param callback:
         :param downloader:
         :param quality:
-        :param output_path:
+        :param path:
         :return:
         """
         quality = self.fix_quality(quality)
@@ -100,11 +102,52 @@ class Core:
             callback = Callback.text_progress_bar
 
         if downloader == default or str(downloader) == "default":
-            default(video=video, quality=quality, path=output_path, callback=callback)
+            default(video=video, quality=quality, path=path, callback=callback)
 
         elif downloader == threaded or str(downloader) == "threaded":
             threaded_download = threaded(max_workers=20, timeout=10)
-            threaded_download(video=video, quality=quality, path=output_path, callback=callback)
+            threaded_download(video=video, quality=quality, path=path, callback=callback)
 
         elif downloader == FFMPEG or str(downloader) == "FFMPEG":
-            FFMPEG(video=video, quality=quality, path=output_path, callback=callback)
+            FFMPEG(video=video, quality=quality, path=path, callback=callback)
+
+    @classmethod
+    def correct_path(cls, path):
+        return Path(path)
+
+    @classmethod
+    def get_segments(cls, quality, m3u8_base_url):
+        quality = Core().fix_quality(quality)
+        base_url = m3u8_base_url
+        new_segment = Core().get_m3u8_by_quality(quality, m3u8_base_url=base_url)
+        # Split the base URL into components
+        url_components = base_url.split('/')
+
+        # Replace the last component with the new segment
+        url_components[-1] = new_segment
+
+        # Rejoin the components into the new full URL
+        new_url = '/'.join(url_components)
+        master_src = Core().get_content(url=new_url).decode("utf-8")
+
+        urls = [l for l in master_src.splitlines()
+                if l and not l.startswith('#')]
+
+        segments = []
+
+        for url in urls:
+            url_components[-1] = url
+            new_url = '/'.join(url_components)
+            segments.append(new_url)
+
+        return segments
+
+    @classmethod
+    def strip_title(cls, title: str) -> str:
+        """
+        :param title:
+        :return: str: strips out non UTF-8 chars of the title
+        """
+        illegal_chars = '<>:"/\\|?*'
+        cleaned_title = ''.join([char for char in title if char in string.printable and char not in illegal_chars])
+        return cleaned_title

eaf_base_api-1.2/base_api/modules/consts.py

@@ -0,0 +1 @@
+MAX_RETRIES = 4

{eaf_base_api-1.0 → eaf_base_api-1.2}/eaf_base_api.egg-info/PKG-INFO

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

{eaf_base_api-1.0 → eaf_base_api-1.2}/eaf_base_api.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ setup.py
 base_api/__init__.py
 base_api/base.py
 base_api/modules/__init__.py
+base_api/modules/consts.py
 base_api/modules/download.py
 base_api/modules/progress_bars.py
 base_api/modules/quality.py

{eaf_base_api-1.0 → eaf_base_api-1.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="eaf_base_api",
-    version="1.0",
+    version="1.2",
     packages=find_packages(),
     install_requires=[
         "requests", "ffmpeg-progress-yield"