add option for output dir

secretmango · Aug 20, 2019 · 5170d8a · 5170d8a
1 parent ccbb8c9
commit 5170d8a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/payload_dumper.py b/payload_dumper.py
@@ -58,7 +58,7 @@ def data_for_op(op,out_file):
 def dump_part(part):
     print(part.partition_name)
 
-    out_file = open('output/%s.img' % part.partition_name, 'wb')
+    out_file = open('%s/%s.img' % (args.out, part.partition_name), 'wb')
     h = hashlib.sha256()
 
     for op in part.operations:
@@ -68,6 +68,8 @@ def dump_part(part):
 parser = argparse.ArgumentParser(description='OTA payload dumper')
 parser.add_argument('payloadfile', type=argparse.FileType('rb'), 
                     help='payload file name')
+parser.add_argument('--out', default='output',
+                    help='output directory')
 args = parser.parse_args()
 
 magic = args.payloadfile.read(4)