Add method for updating initramfs accordingly to OS version

Signed-off-by: Katarzyna Treder <katarzyna.treder@h-partners.com>

test_utils/os_utils.py

@@ -8,9 +8,9 @@ import math
 import posixpath
 import re
 import time
-from datetime import timedelta, datetime
 
-from aenum import IntFlag, Enum, IntEnum
+from datetime import timedelta, datetime
+from enum import IntFlag, Enum, IntEnum, StrEnum
 from packaging import version
 from typing import List
 
@@ -28,6 +28,14 @@ DEBUGFS_MOUNT_POINT = "/sys/kernel/debug"
 MEMORY_MOUNT_POINT = "/mnt/memspace"
 
 
+class Distro(StrEnum):
+    UBUNTU = "ubuntu"
+    DEBIAN = "debian"
+    REDHAT = "rhel"
+    OPENEULER = "openeuler"
+    CENTOS = "centos"
+
+
 class DropCachesMode(IntFlag):
     PAGECACHE = 1
     SLAB = 2
@@ -102,6 +110,17 @@ class SystemManagerType(Enum):
     systemd = 1
 
 
+def get_distro():
+    output = TestRun.executor.run(
+        "cat /etc/os-release | grep -e \"^ID=\" | awk -F= '{print$2}' | tr -d '\"'"
+    ).stdout.lower()
+
+    try:
+        return Distro(output)
+    except ValueError:
+        raise ValueError(f"Could not resolve distro name. Command output: {output}")
+
+
 def get_system_manager():
     output = TestRun.executor.run_expect_success("ps -p 1").stdout
     type = output.split('\n')[1].split()[3]