clean up once finished with restore

namlee135 · Aug 25, 2020 · 85d0eb2 · 85d0eb2
1 parent 71f3f3b
commit 85d0eb2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/restituere.py b/restituere.py
@@ -125,5 +125,8 @@
         restore.restore(args.ipsw[0], restore.is_cellular(args.device[0]), 'yes')
     else:
         restore.restore(args.ipsw[0], restore.is_cellular(args.device[0]))
+    print('Restore finished! Cleaning up...')
+    shutil.rmtree('work')
+    print('Done.')
 else:
     exit(parser.print_help(sys.stderr))