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

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

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

@@ -1,6 +1,7 @@
 import requests
 import logging
 import time
+import string
 
 from base_api.modules.quality import Quality
 from base_api.modules.progress_bars import Callback
@@ -23,11 +24,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

eaf_base_api-1.5/base_api/modules/consts.py

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

{eaf_base_api-1.2 → eaf_base_api-1.5}/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.2 → eaf_base_api-1.5}/eaf_base_api.egg-info/PKG-INFO

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

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

eaf_base_api-1.2/base_api/modules/consts.py

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