mirror of
https://github.com/sotam0316/docker-py-revanced.git
synced 2026-04-25 03:48:37 +09:00
🚨 Updated lints (#308)
This commit is contained in:
+25
-40
@@ -1,5 +1,5 @@
|
||||
"""Downloader Class."""
|
||||
from typing import Any, Dict, Tuple
|
||||
from typing import Any, Dict, Self, Tuple
|
||||
|
||||
import requests
|
||||
from bs4 import BeautifulSoup, Tag
|
||||
@@ -8,31 +8,29 @@ from loguru import logger
|
||||
from src.app import APP
|
||||
from src.downloader.download import Downloader
|
||||
from src.downloader.sources import APK_MIRROR_BASE_URL
|
||||
from src.exceptions import APKMirrorAPKDownloadFailure
|
||||
from src.downloader.utils import status_code_200
|
||||
from src.exceptions import APKMirrorAPKDownloadError
|
||||
from src.utils import bs4_parser, contains_any_word, request_header
|
||||
|
||||
|
||||
class ApkMirror(Downloader):
|
||||
"""Files downloader."""
|
||||
|
||||
def _extract_force_download_link(self, link: str, app: str) -> Tuple[str, str]:
|
||||
def _extract_force_download_link(self: Self, link: str, app: str) -> Tuple[str, str]:
|
||||
"""Extract force download link."""
|
||||
notes_divs = self._extracted_search_div(link, "tab-pane")
|
||||
apk_type = self._extracted_search_div(link, "apkm-badge").get_text()
|
||||
extension = "zip" if apk_type == "BUNDLE" else "apk"
|
||||
possible_links = notes_divs.find_all("a")
|
||||
for possible_link in possible_links:
|
||||
if possible_link.get("href") and "download.php?id=" in possible_link.get(
|
||||
"href"
|
||||
):
|
||||
if possible_link.get("href") and "download.php?id=" in possible_link.get("href"):
|
||||
file_name = f"{app}.{extension}"
|
||||
self._download(APK_MIRROR_BASE_URL + possible_link["href"], file_name)
|
||||
return file_name, APK_MIRROR_BASE_URL + possible_link["href"]
|
||||
raise APKMirrorAPKDownloadFailure(
|
||||
f"Unable to extract force download for {app}", url=link
|
||||
)
|
||||
msg = f"Unable to extract force download for {app}"
|
||||
raise APKMirrorAPKDownloadError(msg, url=link)
|
||||
|
||||
def extract_download_link(self, page: str, app: str) -> Tuple[str, str]:
|
||||
def extract_download_link(self: Self, page: str, app: str) -> Tuple[str, str]:
|
||||
"""Function to extract the download link from apkmirror html page.
|
||||
|
||||
:param page: Url of the page
|
||||
@@ -45,19 +43,15 @@ class ApkMirror(Downloader):
|
||||
(
|
||||
download_link["href"]
|
||||
for download_link in download_links
|
||||
if download_link.get("href")
|
||||
and "download/?key=" in download_link.get("href")
|
||||
if download_link.get("href") and "download/?key=" in download_link.get("href")
|
||||
),
|
||||
None,
|
||||
):
|
||||
return self._extract_force_download_link(
|
||||
APK_MIRROR_BASE_URL + final_download_link, app
|
||||
)
|
||||
raise APKMirrorAPKDownloadFailure(
|
||||
f"Unable to extract link from {app} version list", url=page
|
||||
)
|
||||
return self._extract_force_download_link(APK_MIRROR_BASE_URL + final_download_link, app)
|
||||
msg = f"Unable to extract link from {app} version list"
|
||||
raise APKMirrorAPKDownloadError(msg, url=page)
|
||||
|
||||
def get_download_page(self, main_page: str) -> str:
|
||||
def get_download_page(self: Self, main_page: str) -> str:
|
||||
"""Function to get the download page in apk_mirror.
|
||||
|
||||
:param main_page: Main Download Page in APK mirror(Index)
|
||||
@@ -77,26 +71,23 @@ class ApkMirror(Downloader):
|
||||
links[apk_type] = f"{APK_MIRROR_BASE_URL}{sub_url}"
|
||||
if preferred_link := links.get("APK", links.get("BUNDLE")):
|
||||
return preferred_link
|
||||
raise APKMirrorAPKDownloadFailure(
|
||||
"Unable to extract download page", url=main_page
|
||||
)
|
||||
msg = "Unable to extract download page"
|
||||
raise APKMirrorAPKDownloadError(msg, url=main_page)
|
||||
|
||||
@staticmethod
|
||||
def _extracted_search_div(url: str, search_class: str) -> Tag:
|
||||
"""Extract search div."""
|
||||
r = requests.get(url, headers=request_header, timeout=60)
|
||||
if r.status_code != 200:
|
||||
raise APKMirrorAPKDownloadFailure(
|
||||
f"Unable to connect with {url} on ApkMirror. Are you blocked by APKMirror or abused apkmirror "
|
||||
f"?.Reason - {r.text}",
|
||||
if r.status_code != status_code_200:
|
||||
msg = f"Unable to connect with {url}. Are you blocked by APKMirror or abused apkmirror ?.Reason - {r.text}"
|
||||
raise APKMirrorAPKDownloadError(
|
||||
msg,
|
||||
url=url,
|
||||
)
|
||||
soup = BeautifulSoup(r.text, bs4_parser)
|
||||
return soup.find(class_=search_class)
|
||||
return soup.find(class_=search_class) # type: ignore[return-value]
|
||||
|
||||
def specific_version(
|
||||
self, app: APP, version: str, main_page: str = ""
|
||||
) -> Tuple[str, str]:
|
||||
def specific_version(self: Self, app: APP, version: str, main_page: str = "") -> Tuple[str, str]:
|
||||
"""Function to download the specified version of app from apkmirror.
|
||||
|
||||
:param app: Name of the application
|
||||
@@ -112,18 +103,14 @@ class ApkMirror(Downloader):
|
||||
download_page = self.get_download_page(main_page)
|
||||
return self.extract_download_link(download_page, app.app_name)
|
||||
|
||||
def latest_version(self, app: APP, **kwargs: Any) -> Tuple[str, str]:
|
||||
"""Function to download whatever the latest version of app from
|
||||
apkmirror.
|
||||
def latest_version(self: Self, app: APP, **kwargs: Any) -> Tuple[str, str]:
|
||||
"""Function to download whatever the latest version of app from apkmirror.
|
||||
|
||||
:param app: Name of the application
|
||||
:return: Version of downloaded apk
|
||||
"""
|
||||
|
||||
app_main_page = app.download_source
|
||||
versions_div = self._extracted_search_div(
|
||||
app_main_page, "listWidget p-relative"
|
||||
)
|
||||
versions_div = self._extracted_search_div(app_main_page, "listWidget p-relative")
|
||||
app_rows = versions_div.find_all(class_="appRow")
|
||||
version_urls = [
|
||||
app_row.find(class_="downloadLink")["href"]
|
||||
@@ -131,6 +118,4 @@ class ApkMirror(Downloader):
|
||||
if "beta" not in app_row.find(class_="appRowTitle").get_text().lower()
|
||||
and "alpha" not in app_row.find(class_="appRowTitle").get_text().lower()
|
||||
]
|
||||
return self.specific_version(
|
||||
app, "latest", APK_MIRROR_BASE_URL + max(version_urls)
|
||||
)
|
||||
return self.specific_version(app, "latest", APK_MIRROR_BASE_URL + max(version_urls))
|
||||
|
||||
Reference in New Issue
Block a user