Eporner_API 1.9.6__tar.gz → 2.0__tar.gz

{eporner_api-1.9.6 → eporner_api-2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Eporner_API
-Version: 1.9.6
+Version: 2.0
 Summary: A Python API for the Porn Site Eporner.com
 Author: Johannes Habel
 Author-email: Johannes Habel <EchterAlsFake@proton.me>
@@ -10,8 +10,6 @@ Classifier: Programming Language :: Python
 Requires-Dist: bs4
 Requires-Dist: eaf-base-api
 Requires-Dist: lxml ; extra == 'full'
-Requires-Dist: httpx[http2] ; extra == 'full'
-Requires-Dist: httpx[socks] ; extra == 'full'
 Requires-Python: >=3.9
 Project-URL: Homepage, https://github.com/EchterAlsFake/EPorner_API
 Provides-Extra: full

{eporner_api-1.9.6 → eporner_api-2.0}/eporner_api/eporner_api.py

@@ -3,6 +3,7 @@ import html
 import json
 import os.path
 import logging
+import asyncio
 import argparse
 import traceback
 import threading
@@ -33,7 +34,7 @@ from bs4 import BeautifulSoup
 from urllib.parse import urljoin
 from functools import cached_property
 from base_api.modules.config import RuntimeConfig
-from typing import Generator, Union, Optional, List
+from typing import AsyncGenerator, Generator, Union, Optional, List
 from base_api.base import BaseCore, setup_logger, Helper
 
 """
@@ -102,21 +103,27 @@ def _choose_quality_from_list(available: List[str | int], target: Union[str, int
 
 
 class Video:
-    def __init__(self, url: str, enable_html_scraping: bool = True, core: Optional[BaseCore] = None):
+    def __init__(self, url: str, enable_html_scraping: bool = True, core: Optional[BaseCore] = None, html_content=None):
         self.core = core
         self.url = url
         self.enable_html = enable_html_scraping
-        self.html_content = None
+        self.html_content = html_content
         self.logger = setup_logger(name="EPorner API - [Video]", log_file=None, level=logging.CRITICAL)
-        self.json_data = self.raw_json_data()
+        self.json_data = {}
+        self.html_json_data = {}
+
+    async def init(self):
+        self.json_data = await self.get_raw_json_data()
         if self.enable_html:
-            self.request_html_content()
+            if not self.html_content:
+                await self.request_html_content()
             is_removed = REGEX_VIDEO_DISABLED.findall(self.html_content)
             for _ in is_removed:
                 if _ == "deletedfile":
                     raise VideoDisabled("Video has been removed because of a Copyright claim")
 
             self.html_json_data = self.extract_json_from_html()
+        return self
 
     def enable_logging(self, log_file: str, level, log_ip: str = None, log_port: int = None):
         self.logger = setup_logger(name="EPorner API - [Video]", log_file=log_file, level=level, http_ip=log_ip, http_port=log_port)
@@ -144,13 +151,13 @@ class Video:
         else:
             return self.url  # Assuming this is a video ID (hopefully)
 
-    def raw_json_data(self):
+    async def get_raw_json_data(self):
         """
         Uses the V2 API to retrieve information from a video
         :return:
         """
 
-        data = self.core.fetch(f"{ROOT_URL}{API_VIDEO_ID}?id={self.video_id}&thumbsize=medium&format=json")
+        data = await self.core.fetch(f"{ROOT_URL}{API_VIDEO_ID}?id={self.video_id}&thumbsize=medium&format=json")
         parsed_data = json.loads(data)
         return parsed_data
 
@@ -212,11 +219,11 @@ class Video:
     The following methods are using HTML scraping. This is against the ToS from EPorner.com!
     """
 
-    def request_html_content(self):
+    async def request_html_content(self):
         if not self.enable_html:
             raise HTML_IS_DISABLED("HTML content is disabled! See Documentation for more details")
 
-        self.html_content = html.unescape(self.core.fetch(self.url))
+        self.html_content = html.unescape(await self.core.fetch(self.url))
 
 
     def extract_json_from_html(self):
@@ -333,13 +340,7 @@ JSONDecodeError: I need your help to fix this error. Please report the URL you'v
                 self.logger.error("Couldn't find author. Please report this!")
                 return None
 
-    def direct_download_link(self, quality, mode) -> str:
-        """
-        Returns the direct download URL for a given quality (best/half/worst or a specific resolution).
-        :param quality: 'best', 'half', 'worst', or a specific resolution like '720', '720p', 1080, etc.
-        :param mode: The mode to filter links by (e.g., 'video')
-        :return: str
-        """
+    def _direct_download_quality_map(self, mode) -> dict[int, str]:
         if not self.enable_html:
             raise HTML_IS_DISABLED("HTML content is disabled! See Documentation for more details")
 
@@ -369,6 +370,25 @@ JSONDecodeError: I need your help to fix this error. Please report the URL you'v
             height = int(m.group(1))  # e.g., 1080
             quality_to_url[height] = href  # last one wins; order doesn't matter
 
+        return quality_to_url
+
+    def video_qualities(self, mode=Encoding.mp4_h264) -> list:
+        """
+        Returns the available direct download qualities for a given mode.
+        :param mode: The mode to filter links by (e.g., 'video')
+        :return: list
+        """
+        quality_to_url = self._direct_download_quality_map(mode)
+        return sorted(quality_to_url.keys())
+
+    def direct_download_link(self, quality, mode) -> str:
+        """
+        Returns the direct download URL for a given quality (best/half/worst or a specific resolution).
+        :param quality: 'best', 'half', 'worst', or a specific resolution like '720', '720p', 1080, etc.
+        :param mode: The mode to filter links by (e.g., 'video')
+        :return: str
+        """
+        quality_to_url = self._direct_download_quality_map(mode)
         if not quality_to_url:
             raise NotAvailable(f"No URLs available for mode '{mode}'")
 
@@ -384,7 +404,7 @@ JSONDecodeError: I need your help to fix this error. Please report the URL you'v
         self.logger.info(f"Using direct download link: {full_url} ({chosen_height}p)")
         return full_url
 
-    def download(self, quality, path, callback=None, mode=Encoding.mp4_h264, no_title=False, use_workaround=False,
+    async def download(self, quality, path, callback=None, mode=Encoding.mp4_h264, no_title=False, use_workaround=False,
                  stop_event: threading.Event = None):
         if not self.enable_html:
             raise HTML_IS_DISABLED("HTML content is disabled! See Documentation for more details")
@@ -395,13 +415,13 @@ JSONDecodeError: I need your help to fix this error. Please report the URL you'v
 
         url = self.direct_download_link(quality, mode)
         if use_workaround:
-            response_redirect_url = self.core.fetch(self.direct_download_link(quality, mode),
+            response_redirect_url = await self.core.fetch(self.direct_download_link(quality, mode),
                                             allow_redirects=True, get_response=True) # Sometimes the site trolls me
 
             url = response_redirect_url.url
 
         try:
-            self.core.legacy_download(url=url, callback=callback, path=path, stop_event=stop_event)
+            await self.core.legacy_download(url=url, callback=callback, path=path, stop_event=stop_event)
             return True
 
         except Exception:
@@ -412,18 +432,23 @@ JSONDecodeError: I need your help to fix this error. Please report the URL you'v
 
 
 class Pornstar(Helper):
-    def __init__(self, url: str, enable_html_scraping: bool = False, core: Optional[BaseCore] = None):
+    def __init__(self, url: str, enable_html_scraping: bool = False, core: Optional[BaseCore] = None, html_content=None):
         super().__init__(core=core, video=Video)
         self.core = core
         self.url = url
         self.enable_html_scraping = enable_html_scraping
         self.logger = setup_logger(name="EPorner API - [Pornstar]", log_file=None, level=logging.CRITICAL)
-        self.html_content = self.core.fetch(self.url)
+        self.html_content = html_content
+        
+    async def init(self):
+        if not self.html_content:
+            self.html_content = await self.core.fetch(self.url)
+        return self
 
     def enable_logging(self, log_file: str, level, log_ip: str = None, log_port: int = None):
         self.logger = setup_logger(name="EPorner API - [Pornstar]", log_file=log_file, level=level, http_ip=log_ip, http_port=log_port)
 
-    def videos(self, pages: int = 0, videos_concurrency: int = None, pages_concurrency: int = None) -> Generator[Video, None, None]:
+    async def videos(self, pages: int = 0, videos_concurrency: int = None, pages_concurrency: int = None) -> AsyncGenerator[Video, None]:
         if pages == 0:
             video_amount = str(self.video_amount).replace(",", "")
             pages = round(int(video_amount)) / 37 # One page contains 37 videos
@@ -433,8 +458,9 @@ class Pornstar(Helper):
 
         pages = round(pages) # Dont ask
         page_urls = [urljoin(f"{self.url}/", str(page)) for page in range(1, pages + 1)]
-        yield from self.iterator(page_urls=page_urls, extractor=extractor, pages_concurrency=pages_concurrency,
-                                 videos_concurrency=videos_concurrency)
+        async for video in self.iterator(page_urls=page_urls, extractor=extractor, pages_concurrency=pages_concurrency,
+                                 videos_concurrency=videos_concurrency):
+            yield await video.init()
 
     @cached_property
     def name(self) -> str:
@@ -551,41 +577,45 @@ class Client(Helper):
     def enable_logging(self, log_file: str, level, log_ip: str = None, log_port: int = None):
         self.logger = setup_logger(name="EPorner API - [Client]", log_file=log_file, level=level, http_ip=log_ip, http_port=log_port)
 
-    def get_video(self, url: str, enable_html_scraping: bool = True) -> Video:
+    async def get_video(self, url: str, enable_html_scraping: bool = True) -> Video:
         """Returns the Video object for a given URL"""
         self.logger.info(f"Returning video object for: {url} HTML Scraping -> {enable_html_scraping}")
-        return Video(url, enable_html_scraping=enable_html_scraping, core=self.core)
+        video = Video(url, enable_html_scraping=enable_html_scraping, core=self.core)
+        return await video.init()
 
-    def search_videos(self, query: str, sorting_gay: Union[str, Gay], sorting_order: Union[str, Order],
+    async def search_videos(self, query: str, sorting_gay: Union[str, Gay], sorting_order: Union[str, Order],
                       sorting_low_quality: Union[str, LowQuality],
-                      page: int, per_page: int, enable_html_scraping: bool = True) -> Generator[Video, None, None]:
+                      page: int, per_page: int, enable_html_scraping: bool = True) -> AsyncGenerator[Video, None]:
 
-        response = self.core.fetch(f"{ROOT_URL}{API_SEARCH}?query={query}&per_page={per_page}&%page={page}"
+        response = await self.core.fetch(f"{ROOT_URL}{API_SEARCH}?query={query}&per_page={per_page}&%page={page}"
                                 f"&thumbsize=medium&order={sorting_order}&gay={sorting_gay}&lq="
                                 f"{sorting_low_quality}&format=json")
 
         json_data = json.loads(response)
         for video_ in json_data.get("videos", []):  # Don't know why this works lmao
             id_ = video_["url"]
-            yield Video(id_, enable_html_scraping, core=self.core)
+            video = Video(id_, enable_html_scraping, core=self.core)
+            yield await video.init()
 
-    def get_videos_by_category(self, category: Union[str, Category], enable_html_scraping: bool = False,
-                               videos_concurrency: int = None, pages_concurrency: int = None) -> Generator[Video, None, None]:
+    async def get_videos_by_category(self, category: Union[str, Category], enable_html_scraping: bool = False,
+                               videos_concurrency: int = None, pages_concurrency: int = None) -> AsyncGenerator[Video, None]:
 
         page_urls = [f"{ROOT_URL}cat/{category}/{page}" for page in range(1, 100)]
 
         videos_concurrency = videos_concurrency or self.core.config.videos_concurrency
         pages_concurrency = pages_concurrency or self.core.config.pages_concurrency
-        yield from self.iterator(page_urls=page_urls, videos_concurrency=videos_concurrency,
-                                 pages_concurrency=pages_concurrency, extractor=extractor)
+        async for video in self.iterator(page_urls=page_urls, videos_concurrency=videos_concurrency,
+                                 pages_concurrency=pages_concurrency, extractor=extractor):
+            yield await video.init()
 
 
-    def get_pornstar(self, url: str, enable_html_scraping: bool = True) -> Pornstar:
+    async def get_pornstar(self, url: str, enable_html_scraping: bool = True) -> Pornstar:
         self.logger.info(f"Returning Pornstar object for: {url} HTML Scraping -> {enable_html_scraping}")
-        return Pornstar(url, enable_html_scraping, core=self.core)
+        pornstar = Pornstar(url, enable_html_scraping, core=self.core)
+        return await pornstar.init()
 
 
-def main():
+async def run_main():
     parser = argparse.ArgumentParser(description="API Command Line Interface")
     parser.add_argument("--download", metavar="URL (str)", type=str, help="URL to download from")
     parser.add_argument("--quality", metavar="best,half,worst", type=str, help="The video quality (best,half,worst)",
@@ -602,8 +632,8 @@ def main():
 
     if args.download:
         client = Client()
-        video = client.get_video(args.download, enable_html_scraping=True)
-        video.download(quality=args.quality, path=args.output, no_title=no_title)
+        video = await client.get_video(args.download, enable_html_scraping=True)
+        await video.download(quality=args.quality, path=args.output, no_title=no_title)
 
     if args.file:
         videos = []
@@ -613,11 +643,14 @@ def main():
             content = file.read().splitlines()
 
         for url in content:
-            videos.append(client.get_video(url, enable_html_scraping=True))
+            videos.append(await client.get_video(url, enable_html_scraping=True))
 
         for video in videos:
-            video.download(quality=args.quality, path=args.output, no_title=no_title)
+            await video.download(quality=args.quality, path=args.output, no_title=no_title)
 
 
+def main():
+    asyncio.run(run_main())
+
 if __name__ == "__main__":
     main()

{eporner_api-1.9.6 → eporner_api-2.0}/eporner_api/tests/test_category.py

@@ -1,31 +1,34 @@
-import time
-
+import pytest
 from ..eporner_api import Client, Category
 from base_api import BaseCore
-core = BaseCore()
-core.config.pages_concurrency = 1
-core.config.videos_concurrency = 1
-
 
-def test_category():
+@pytest.mark.asyncio
+async def test_category():
+    core = BaseCore()
+    core.config.pages_concurrency = 1
+    core.config.videos_concurrency = 1
+    
     videos_1 = Client(core).get_videos_by_category(category=Category.JAPANESE)
     videos_2 = Client(core).get_videos_by_category(category=Category.HD)
     videos_3 = Client(core).get_videos_by_category(category=Category.BLONDE)
 
-    for idx, video in enumerate(videos_1):
+    idx = 0
+    async for video in videos_1:
         if idx == 3:
             break
-
         assert isinstance(video.title, str) and len(video.title) > 0
+        idx += 1
 
-    for idx, video in enumerate(videos_2):
+    idx = 0
+    async for video in videos_2:
         if idx == 3:
             break
-
         assert isinstance(video.title, str) and len(video.title) > 0
+        idx += 1
 
-    for idx, video in enumerate(videos_3):
+    idx = 0
+    async for video in videos_3:
         if idx == 3:
             break
-
         assert isinstance(video.title, str) and len(video.title) > 0
+        idx += 1

{eporner_api-1.9.6 → eporner_api-2.0}/eporner_api/tests/test_pornstar.py

@@ -1,27 +1,28 @@
+import pytest
 from ..eporner_api import Client
-import time
-url = "https://www.eporner.com/pornstar/riley-reid/"
 from base_api import BaseCore
-core = BaseCore()
-core.config.pages_concurrency = 1
-core.config.videos_concurrency = 1
 
-pornstar = Client(core).get_pornstar(url, enable_html_scraping=True)
+@pytest.mark.asyncio
+async def test_pornstar():
+    url = "https://www.eporner.com/pornstar/riley-reid/"
+    core = BaseCore()
+    core.config.pages_concurrency = 1
+    core.config.videos_concurrency = 1
+    pornstar = await Client(core).get_pornstar(url, enable_html_scraping=True)
 
-
-def test_videos():
     videos = pornstar.videos(pages=1)
-
-    for idx, video in enumerate(videos):
+    
+    idx = 0
+    async for video in videos:
         assert isinstance(video.title, str) and len(video.title) > 3
         if idx == 5:
             break
+        idx += 1
 
-def test_information():
     assert isinstance(pornstar.pornstar_rank, str) and len(pornstar.pornstar_rank) >= 1
     assert isinstance(pornstar.aliases, list) and len(pornstar.aliases) > 1
     assert isinstance(pornstar.biography, str) and len(pornstar.biography) > 10
-    assert isinstance(pornstar.age, str) and len(pornstar.age) >= 2  # would be weird if this is 1-9 lmao (just kidding)
+    assert isinstance(pornstar.age, str) and len(pornstar.age) >= 2
     assert isinstance(pornstar.cup, str) and len(pornstar.cup) >= 1
     assert isinstance(pornstar.country, str) and len(pornstar.country) >= 2
     assert isinstance(pornstar.weight, str) and len(pornstar.weight) >= 2

{eporner_api-1.9.6 → eporner_api-2.0}/eporner_api/tests/test_search.py

@@ -1,48 +1,53 @@
+import pytest
 from ..eporner_api import Client, Gay, Order, LowQuality
-import time
 
-client = Client()
+
 query = "Mia Khalifa"
-pages = 2
-per_page = 10
+pages = 1
+per_page = 1
+
+@pytest.fixture
+def client():
+    return Client()
 
-def test_search_1():
+@pytest.mark.asyncio
+async def test_search_1(client):
     videos = client.search_videos(query, page=pages, per_page=per_page, sorting_gay=Gay.exclude_gay_content, sorting_order=Order.top_rated, sorting_low_quality=LowQuality.exclude_low_quality_content)
-    for video in videos:
+    async for video in videos:
         assert len(video.title) > 0
 
-
-def test_search_2():
+@pytest.mark.asyncio
+async def test_search_2(client):
     videos = client.search_videos(query, page=pages, per_page=per_page, sorting_gay=Gay.only_gay_content, sorting_order=Order.latest, sorting_low_quality=LowQuality.only_low_quality_content)
-    for video in videos:
+    async for video in videos:
         assert len(video.title) > 0
 
-
-def test_search_3():
+@pytest.mark.asyncio
+async def test_search_3(client):
     videos = client.search_videos(query, page=pages, per_page=per_page, sorting_gay=Gay.include_gay_content, sorting_order=Order.longest, sorting_low_quality=LowQuality.include_low_quality_content)
-    for video in videos:
+    async for video in videos:
         assert len(video.title) > 0
 
-
-def test_search_4():
+@pytest.mark.asyncio
+async def test_search_4(client):
     videos = client.search_videos(query, page=pages, per_page=pages, sorting_gay=Gay.exclude_gay_content, sorting_order=Order.shortest, sorting_low_quality=LowQuality.include_low_quality_content)
-    for video in videos:
+    async for video in videos:
         assert len(video.title) > 0
 
-
-def test_search_5():
+@pytest.mark.asyncio
+async def test_search_5(client):
     videos = client.search_videos(query, page=pages, per_page=per_page, sorting_order=Gay.include_gay_content, sorting_gay=Order.top_weekly, sorting_low_quality=LowQuality.include_low_quality_content)
-    for video in videos:
+    async for video in videos:
         assert len(video.title) > 0
 
-
-def test_search_6():
+@pytest.mark.asyncio
+async def test_search_6(client):
     videos = client.search_videos(query, page=pages, per_page=per_page, sorting_order=Order.most_popular, sorting_low_quality=LowQuality.include_low_quality_content, sorting_gay=Gay.only_gay_content)
-    for video in videos:
+    async for video in videos:
         assert len(video.title) > 0
 
-
-def test_search_7():
+@pytest.mark.asyncio
+async def test_search_7(client):
     videos = client.search_videos(query, page=pages, per_page=per_page, sorting_gay=Gay.include_gay_content, sorting_order=Order.top_monthly, sorting_low_quality=LowQuality.include_low_quality_content)
-    for video in videos:
+    async for video in videos:
         assert len(video.title) > 0

{eporner_api-1.9.6 → eporner_api-2.0}/eporner_api/tests/test_video.py

@@ -1,70 +1,26 @@
+import pytest
 from ..eporner_api import Client, Encoding, NotAvailable
-import time
-url = "https://www.eporner.com/video-bTwP6vsFj5U/human-anal-sex-toy/"
-video = Client().get_video(url, enable_html_scraping=True)
-time.sleep(5)  # Lmao
 
-def test_title():
+@pytest.mark.asyncio
+async def test_video():
+    url = "https://www.eporner.com/video-bTwP6vsFj5U/human-anal-sex-toy/"
+    video = await Client().get_video(url, enable_html_scraping=True)
     assert isinstance(video.title, str) and len(video.title) > 0
-
-
-def test_video_id():
     assert isinstance(video.video_id, str) and len(video.video_id) > 0
-
-
-def test_tags():
     assert isinstance(video.tags, list) and len(video.tags) > 0
-
-
-def test_views():
     assert isinstance(video.views, int) and video.views > 0
-
-
-def test_rate():
     assert isinstance(video.rate, str) and len(video.rate) > 0
-
-
-def test_publish_date():
     assert isinstance(video.publish_date, str) and len(video.publish_date) > 0
-
-
-def test_length_seconds():
-    assert isinstance(video.length, int) > 0
-
-
-def test_length_minutes():
+    assert isinstance(video.length, int) and video.length > 0
     assert isinstance(video.length_minutes, str) and len(video.length_minutes) > 0
-
-
-def test_embed_url():
     assert isinstance(video.embed_url, str) and len(video.embed_url) > 0
-
-
-def test_thumbnails():
     assert isinstance(video.thumbnail, str) and len(video.thumbnail) > 0
-
-
-def test_bitrate():
     assert isinstance(video.bitrate, str) and len(video.bitrate) > 0
-
-
-def test_source_video_url():
     assert isinstance(video.source_video_url, str) and len(video.source_video_url) > 0
-
-
-def test_rating():
     assert isinstance(video.rating, str) and len(video.rating) > 0
-
-
-def test_rating_count():
     assert isinstance(video.rating_count, str) and len(video.rating_count) > 0
-
-
-def test_author():
     assert isinstance(video.author, str) and len(video.author) > 0
 
-
-def test_direct_download_url():
     assert isinstance(video.direct_download_link(quality=2160, mode=Encoding.mp4_h264), str)
     assert isinstance(video.direct_download_link(quality="half", mode=Encoding.mp4_h264), str)
     assert isinstance(video.direct_download_link(quality="worst", mode=Encoding.mp4_h264), str)
@@ -72,6 +28,5 @@ def test_direct_download_url():
         assert isinstance(video.direct_download_link(quality="best", mode=Encoding.av1), str)
         assert isinstance(video.direct_download_link(quality="half", mode=Encoding.av1), str)
         assert isinstance(video.direct_download_link(quality="worst", mode=Encoding.av1), str)
-
     except NotAvailable:
         pass

{eporner_api-1.9.6 → eporner_api-2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "Eporner_API"
-version = "1.9.6"
+version = "2.0"
 description = "A Python API for the Porn Site Eporner.com"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.9" # 3.9 due to httpx requirements
@@ -15,14 +15,14 @@ authors = [
 ]
 dependencies = [
     "bs4",
-    "eaf_base_api",
+    "eaf-base-api",
 ]
 classifiers = [
     "Programming Language :: Python",
 ]
 
 [project.optional-dependencies]
-full = ["lxml", "httpx[http2]", "httpx[socks]"]
+full = ["lxml"]
 
 [project.urls]
 Homepage = "https://github.com/EchterAlsFake/EPorner_API"