allow user-code switch to extract pyc files

binref · Sep 28, 2024 · 22690fc · 22690fc
1 parent 433f4d7
commit 22690fc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/refinery/units/formats/archive/xtpyi.py b/refinery/units/formats/archive/xtpyi.py
@@ -603,11 +603,12 @@ def unpack(self, data):
  for position in positions:
  self.log_info(F'magic signature found at offset 0x{position:0{width}X}')
  self.log_warn(F'found {len(positions) - 1} potential PyInstaller epilogue markers; using last one.')
- decompile = self.args.decompile or self.args.user_code
+ decompile = self.args.decompile
+ uc_target = PiType.USERCODE if decompile else PiType.SOURCE
  archive = PyInstallerArchiveEpilogue(view, positions[-1], mode, decompile)
  for name, file in archive.files.items():
  if self.args.user_code:
- if file.type != PiType.USERCODE:
+ if file.type != uc_target:
  continue
  if name.startswith('pyiboot'):
  continue