diff --git a/mvt/android/download_apks.py b/mvt/android/download_apks.py index 8cfe6b8..5c79729 100644 --- a/mvt/android/download_apks.py +++ b/mvt/android/download_apks.py @@ -40,11 +40,15 @@ class DownloadAPKs(AndroidExtraction): """DownloadAPKs is the main class operating the download of APKs from the device.""" - def __init__(self, file_path=None, base_folder=None, output_folder=None, - serial=None, fast_mode=False, log=None, results=[]): - super().__init__(file_path=file_path, base_folder=base_folder, - output_folder=output_folder, fast_mode=fast_mode, - log=log, results=results) + def __init__(self, output_folder=None, all_apks=False, packages=None): + """Initialize module. + :param output_folder: Path to the folder where data should be stored + :param all_apks: Boolean indicating whether to download all packages + or filter known-goods + :param packages: Provided list of packages, typically for JSON checks + """ + super().__init__(file_path=None, base_folder=None, + output_folder=output_folder) self.output_folder_apk = None self.packages = packages or []