Retry hdiutil create if it fails

heinrich5991 · Sep 30, 2021 · fdaf31a · fdaf31a
1 parent f35da54
commit fdaf31a
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/scripts/dmg.py b/scripts/dmg.py
@@ -3,6 +3,7 @@
 import shlex
 import subprocess
 import tempfile
+import time
 
 ConfigDmgtools = namedtuple('Config', 'dmg hfsplus newfs_hfs verbose')
 ConfigHdiutil = namedtuple('Config', 'hdiutil verbose')
@@ -71,9 +72,18 @@ def _hdiutil(self, *args):
 	def create(self, dmg, volume_name, directory, symlinks):
 		if symlinks:
 			raise NotImplementedError("symlinks are not yet implemented")
-		if os.path.exists(volume_name + '.dmg'):
-			os.remove(volume_name + '.dmg')
-		self._hdiutil('create', '-volname', volume_name, '-srcdir', directory, dmg)
+		for i in range(5):
+			if os.path.exists(volume_name + '.dmg'):
+				os.remove(volume_name + '.dmg')
+			try:
+				self._hdiutil('create', '-volname', volume_name, '-srcdir', directory, dmg)
+			except subprocess.CalledProcessError as e:
+				if i == 4:
+					raise e
+				print("Retrying hdiutil create")
+				time.sleep(5)
+			else:
+				break
 
 def main():
 	import argparse