Add --post flag and pass to mobilesync.rebuild(...)

iospy/__main__.py

@@ -78,14 +78,15 @@ def files(ctx: click.Context, domain: str):
 
 @cli.command()
 @click.option("--domain", type=str, help="Limit to specific domain.")
+@click.option("--post/--raw", help="Run postprocessing on written files.")
 @click.option(
     "--output",
     type=click.Path(exists=True, file_okay=False),
     help="Directory to write file structure into. [defaults to current directory]",
     default=".",
 )
 @click.pass_context
-def rebuild(ctx: click.Context, domain: str, output: str):
+def rebuild(ctx: click.Context, domain: str, post: bool, output: str):
     """
     Copy and rename files from backup into `output`.
 
@@ -97,7 +98,7 @@ def rebuild(ctx: click.Context, domain: str, output: str):
     """
     manifest = ctx.obj["manifest"]
     logger.info("Rebuilding file structure from manifest at %s", manifest)
-    mobilesync.rebuild(manifest, domain=domain, target=output)
+    mobilesync.rebuild(manifest, domain=domain, target=output, postprocess_files=post)
 
 
 main = cli.main

iospy/mobilesync.py

@@ -6,7 +6,7 @@
 
 import appdirs
 
-from .util import query
+from .util import query, postprocess
 
 logger = logging.getLogger(__name__)
 
@@ -67,6 +67,7 @@ def rebuild(
     manifest: Union[bytes, str, os.PathLike],
     domain: str = None,
     target: Union[bytes, str, os.PathLike] = ".",
+    postprocess_files: bool = False,
 ):
     """
     Rebuild the deep structure (creating directories as needed) specified in the
@@ -86,5 +87,7 @@ def rebuild(
             dst.parent.mkdir(parents=True, exist_ok=True)
             shutil.copy2(src, dst)
             logger.info("Copied %s -> %s", fileID, dst)
+            if postprocess_files:
+                postprocess(dst)
         else:
             logger.debug("Skipping missing file %s -> %s", fileID, relativePath)