extract_utils: add shortcircuit to extract all files from zip

Change-Id: I7296aca27ba06d809c5044e003b773e69c3c98f4

Author: Demon000

extract.py

@@ -36,6 +36,11 @@
     type=str,
     help='Extra partitions to extract',
 )
+parser.add_argument(
+    '--all',
+    action='store_true',
+    help='Extract all files from archive',
+)
 
 parser.add_argument(
     'source',
@@ -54,6 +59,7 @@
     ctx = ExtractCtx(
         keep_dump=True,
         extract_partitions=extract_partitions,
+        extract_all=args.all,
     )
 
     with get_dump_dir(args.source, ctx) as dump_dir:

extract_utils/args.py

@@ -13,6 +13,11 @@
 parser = argparse.ArgumentParser(description='Extract utils')
 
 group = parser.add_mutually_exclusive_group()
+group.add_argument(
+    '--extract-all',
+    action='store_true',
+    help='Extract all files from archive',
+)
 group.add_argument(
     '--only-common',
     action='store_true',
@@ -87,6 +92,7 @@ class ArgsSource(str, Enum):
 class Args:
     def __init__(self, args: argparse.Namespace):
         # Wrap to provide type hints
+        self.extract_all: bool = args.extract_all
         self.only_common: bool = args.only_common
         self.only_target: bool = args.only_target
         self.extract_factory: bool = args.extract_factory

extract_utils/extract.py

@@ -64,6 +64,7 @@ def __init__(
         firmware_partitions: Optional[List[str]] = None,
         firmware_files: Optional[List[str]] = None,
         factory_files: Optional[List[str]] = None,
+        extract_all=False,
     ):
         if extract_fns is None:
             extract_fns = {}
@@ -90,6 +91,8 @@ def __init__(
         self.factory_files = factory_files
         self.extra_files: List[str] = []
 
+        self.extract_all = extract_all
+
 
 def file_name_to_partition(file_name: str):
     return file_name.split('.', 1)[0]
@@ -238,6 +241,9 @@ def filter_extract_file_paths(
     ctx: ExtractCtx,
     file_paths: List[str],
 ):
+    if ctx.extract_all:
+        return file_paths
+
     return filter_files(
         [
             ctx.extract_partitions,

extract_utils/main.py

@@ -164,6 +164,7 @@ def run(self):
                 list(firmware_partitions),
                 list(firmware_files),
                 list(factory_files),
+                self.__args.extract_all,
             )
 
             with create_source(self.__args.source, extract_ctx) as source: