ci: install pefile

2 files changed, 8 insertions, 1 deletions

diff --git a/src/ukify/test/test_ukify.py b/src/ukify/test/test_ukify.py
index a4afa56ad9..48ffc6d495 100755
--- a/src/ukify/test/test_ukify.py
+++ b/src/ukify/test/test_ukify.py
@@ -20,6 +20,12 @@ try:
 except ImportError:
     sys.exit(77)
 
+try:
+    # pyflakes: noqa
+    import pefile  # noqa
+except ImportError:
+    sys.exit(77)
+
 # We import ukify.py, which is a template file. But only __version__ is
 # substituted, which we don't care about here. Having the .py suffix makes it
 # easier to import the file.
diff --git a/src/ukify/ukify.py b/src/ukify/ukify.py
index d4772d2bae..83423fc720 100755
--- a/src/ukify/ukify.py
+++ b/src/ukify/ukify.py
@@ -20,7 +20,6 @@ import subprocess
 import tempfile
 import typing
 
-import pefile
 
 __version__ = '{{GIT_VERSION}}'
 
@@ -66,6 +65,8 @@ def shell_join(cmd):
 
 
 def pe_executable_size(filename):
+    import pefile
+
     pe = pefile.PE(filename)
     section = pe.sections[-1]
     return section.VirtualAddress + section.Misc_VirtualSize