Merge pull request #11 from MGlants/check_outfolder

Add check for output folder existence

payload_dumper.py

@@ -6,7 +6,7 @@
 import argparse
 import bsdiff4
 import io
-
+import os
 try:
     import lzma
 except ImportError:
@@ -114,7 +114,7 @@ def dump_part(part):
 
 
 parser = argparse.ArgumentParser(description='OTA payload dumper')
-parser.add_argument('payloadfile', type=argparse.FileType('rb'), 
+parser.add_argument('payloadfile', type=argparse.FileType('rb'),
                     help='payload file name')
 parser.add_argument('--out', default='output',
                     help='output directory (defaul: output)')
@@ -124,6 +124,10 @@ def dump_part(part):
                     help='directory with original images for differential OTA (defaul: old)')
 args = parser.parse_args()
 
+#Check for --out directory exists
+if not os.path.exists(args.out):
+    os.makedirs(args.out)
+
 magic = args.payloadfile.read(4)
 assert magic == b'CrAU'
 
@@ -154,4 +158,4 @@ def dump_part(part):
     # extents = flatten([op.dst_extents for op in part.operations])
     # assert verify_contiguous(extents), 'operations do not span full image'
 
-    dump_part(part)
+    dump_part(part)