🎨 Package name as attribute

This commit is contained in:
Nikhil Badyal
2023-08-24 17:27:45 +05:30
parent 2ace3aa05a
commit 2129d7ea43
3 changed files with 7 additions and 6 deletions
+5
View File
@@ -17,6 +17,8 @@ class APP(object):
"""Patched APK.""" """Patched APK."""
def __init__(self, app_name: str, config: RevancedConfig): def __init__(self, app_name: str, config: RevancedConfig):
from src.patches import Patches
self.app_name = app_name self.app_name = app_name
self.app_version = config.env.str(f"{app_name}_VERSION".upper(), None) self.app_version = config.env.str(f"{app_name}_VERSION".upper(), None)
self.experiment = False self.experiment = False
@@ -48,6 +50,9 @@ class APP(object):
self.download_dl = config.env.str(f"{app_name}_DL".upper(), "") self.download_dl = config.env.str(f"{app_name}_DL".upper(), "")
self.download_patch_resources(config) self.download_patch_resources(config)
self.download_source = config.env.str(f"{app_name}_DL_SOURCE".upper(), "") self.download_source = config.env.str(f"{app_name}_DL_SOURCE".upper(), "")
self.package_name = config.env.str(
f"{app_name}_PACKAGE_NAME".upper(), Patches.get_package_name(app_name)
)
def download_apk_for_patching(self, config: RevancedConfig) -> None: def download_apk_for_patching(self, config: RevancedConfig) -> None:
"""Download apk to be patched.""" """Download apk to be patched."""
+1 -3
View File
@@ -3,7 +3,6 @@ from typing import Any, Tuple
from src.app import APP from src.app import APP
from src.downloader.download import Downloader from src.downloader.download import Downloader
from src.patches import Patches
class ApkPure(Downloader): class ApkPure(Downloader):
@@ -16,8 +15,7 @@ class ApkPure(Downloader):
:param app: Name of the application :param app: Name of the application
:return: Version of downloaded apk :return: Version of downloaded apk
""" """
package_name = Patches.get_package_name(app.app_name) download_url = app.download_source.format(app.package_name)
download_url = app.download_source.format(package_name)
file_name = f"{app.app_name}.apk" file_name = f"{app.app_name}.apk"
self._download(download_url, file_name) self._download(download_url, file_name)
return file_name, download_url return file_name, download_url
+1 -3
View File
@@ -7,7 +7,6 @@ from bs4 import BeautifulSoup
from src.app import APP from src.app import APP
from src.downloader.download import Downloader from src.downloader.download import Downloader
from src.exceptions import APKSosAPKDownloadFailure from src.exceptions import APKSosAPKDownloadFailure
from src.patches import Patches
from src.utils import bs4_parser, request_header from src.utils import bs4_parser, request_header
@@ -38,6 +37,5 @@ class ApkSos(Downloader):
:param app: Name of the application :param app: Name of the application
:return: Version of downloaded apk :return: Version of downloaded apk
""" """
package_name = Patches.get_package_name(app.app_name) download_url = app.download_source.format(app.package_name)
download_url = app.download_source.format(package_name)
return self.extract_download_link(download_url, app.app_name) return self.extract_download_link(download_url, app.app_name)