fix(dmg-builder): the "import" unbound issue for python 2/3 (#6672)

electron-userland · Feb 25, 2022 · 3a4b64a · 3a4b64a
1 parent e1a439b
commit 3a4b64a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/.changeset/cold-crabs-destroy.md b/.changeset/cold-crabs-destroy.md
@@ -0,0 +1,5 @@
+---
+"dmg-builder": patch
+---
+
+fix(dmg-builder): the "import" unbound issue
diff --git a/packages/dmg-builder/vendor/dmgbuild/core.py b/packages/dmg-builder/vendor/dmgbuild/core.py
@@ -4,13 +4,16 @@
 import os
 import re
 import sys
+
 if sys.version_info.major == 3:
-    try:
-        from importlib import reload
-    except ImportError:
-        from imp import reload
-reload(sys)  # Reload is a hack
-sys.setdefaultencoding('UTF8')
+  try:
+      from importlib import reload
+  except ImportError:
+      from imp import reload
+  reload(sys)  # To workaround the unbound issue
+else:
+  reload(sys)  # Reload is a hack
+  sys.setdefaultencoding('UTF8')
 
 sys.path.append(os.path.normpath(os.path.join(os.path.dirname(__file__), "..")))
 
@@ -284,4 +287,4 @@ def build_dmg():
   except:
     raise
 
-build_dmg()
+build_dmg()