From f26850291824b1897e7c51aff3cad8ee5edafafa Mon Sep 17 00:00:00 2001
From: Louis Abel <label@rockylinux.org>
Date: Fri, 15 Nov 2024 15:08:03 -0700
Subject: [PATCH] add magic

---
 iso/empanadas/empanadas/util/shared.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/iso/empanadas/empanadas/util/shared.py b/iso/empanadas/empanadas/util/shared.py
index 79db257..5880dfb 100644
--- a/iso/empanadas/empanadas/util/shared.py
+++ b/iso/empanadas/empanadas/util/shared.py
@@ -14,6 +14,7 @@ import xmltodict
 import productmd.treeinfo
 import productmd.composeinfo
 import pycdlib
+import magic
 import empanadas
 import kobo.shortcuts
 from empanadas.common import Color
@@ -1342,6 +1343,19 @@ class Idents:
 
         return False
 
+    @staticmethod
+    def get_magic(m):
+        """
+        Gets magic data of a given file
+        """
+        try:
+            meta = magic.detect_from_filename(m)
+        except ValueError as exc:
+            print(exc)
+            return False
+
+        return meta
+
     @staticmethod
     def get_vol_id(i):
         """
@@ -1353,6 +1367,14 @@ class Idents:
         except pycdlib.pycdlibexception.PyCdlibInvalidISO as exc:
             print(exc)
             return False
+        # This is for s390x. a temporary hack
+        except pycdlib.pycdlibexception.PyCdlibInvalidInput as exc:
+            print(exc)
+            print('Trying magic instead')
+            magic_data = Idents.get_magic(i)
+            if magic_data.mime_type == 'application/x-iso9660-image':
+                volume_id = magic_data.name.split("'")[1]
+                return volume_id
 
         pvd = iso.pvd
         volume_id = pvd.volume_identifier.decode('UTF-8').strip()