eaf-base-api 1.3__tar.gz → 1.6__tar.gz

{eaf_base_api-1.3 → eaf_base_api-1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: eaf_base_api
-Version: 1.3
+Version: 1.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-1.3 → eaf_base_api-1.6}/base_api/base.py

@@ -2,6 +2,7 @@ import requests
 import logging
 import time
 import string
+import os
 
 from base_api.modules.quality import Quality
 from base_api.modules.progress_bars import Callback
@@ -24,11 +25,11 @@ base_qualities = ["250p", "360p", "480p", "720p", "1080p", "1440p", "2160p"]
 
 class Core:
     @classmethod
-    def get_content(cls, url, headers=None, cookies=None):
+    def get_content(cls, url, headers=None, cookies=None, stream=False):
         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)
+                response = requests.get(url, headers=headers, cookies=cookies, stream=stream)
                 if response.status_code == 200:
                     logging.info(f"Successfully fetched {url} on attempt [{i+1}]")
                     return response.content
@@ -151,4 +152,18 @@ class Core:
         """
         illegal_chars = '<>:"/\\|?*'
         cleaned_title = ''.join([char for char in title if char in string.printable and char not in illegal_chars])
-        return cleaned_title
+        return cleaned_title
+
+    @classmethod
+    def return_path(cls, video, args):
+        path = args.output
+        if args.use_title:
+            if not str(path).endswith(os.sep):
+                path += os.sep
+
+            path += video.title + ".mp4"
+
+        else:
+            path = args.output
+
+        return path

eaf_base_api-1.6/base_api/modules/consts.py

@@ -0,0 +1,2 @@
+MAX_RETRIES = 4
+FFMPEG_PATH = "ffmpeg"

{eaf_base_api-1.3 → eaf_base_api-1.6}/base_api/modules/download.py

@@ -7,6 +7,13 @@ import os
 from ffmpeg_progress_yield import FfmpegProgress
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from typing import Callable, List
+from base_api.modules.progress_bars import Callback
+
+try:
+    from base_api.modules.consts import FFMPEG_PATH
+
+except (ModuleNotFoundError, ImportError):
+    from .consts import FFMPEG_PATH
 
 CallbackType = Callable[[int, int], None]
 
@@ -109,7 +116,7 @@ def FFMPEG(video, quality, callback, path, start=0) -> bool:
 
     # Build the command for FFMPEG as a list directly
     command = [
-        "ffmpeg",
+        FFMPEG_PATH,
         "-i", new_url,  # Input URL
         "-bsf:a", "aac_adtstoasc",
         "-y",  # Overwrite output files without asking
@@ -126,4 +133,34 @@ def FFMPEG(video, quality, callback, path, start=0) -> bool:
         if progress == 100:
             return True
 
-    return False
+    return False
+
+
+def legacy_download(stream, path, url, callback=None):
+    response = requests.get(url, stream=stream)
+    file_size = int(response.headers.get('content-length', 0))
+
+    if callback is None:
+        progress_bar = Callback()
+
+    downloaded_so_far = 0
+
+    if not os.path.exists(path):
+        with open(path, 'wb') as file:
+            for chunk in response.iter_content(chunk_size=1024):
+                file.write(chunk)
+                downloaded_so_far += len(chunk)
+
+                if callback:
+                    callback(downloaded_so_far, file_size)
+
+                else:
+                    progress_bar.text_progress_bar(downloaded=downloaded_so_far, total=file_size)
+
+        if not callback:
+            del progress_bar
+
+        return True
+
+    else:
+        raise FileExistsError("The file already exists.")

{eaf_base_api-1.3 → eaf_base_api-1.6}/eaf_base_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: eaf_base_api
-Version: 1.3
+Version: 1.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-1.3 → eaf_base_api-1.6}/setup.py

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

eaf_base_api-1.3/base_api/modules/consts.py

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