update lib

ix-dev/stable/emby/migrations/migration_helpers/kubernetes_secrets.py

@@ -0,0 +1,15 @@
+def get_value_from_secret(secrets={}, secret_name="", key=""):
+    if not secrets or not secret_name or not key:
+        raise ValueError("Expected [secrets], [secret_name] and [key] to be set")
+    for secret in secrets.items():
+        curr_secret_name = secret[0]
+        curr_data = secret[1]
+
+        if curr_secret_name.endswith(secret_name):
+            if not curr_data.get(key, None):
+                raise ValueError(
+                    f"Expected [{key}] to be set in secret [{curr_secret_name}]"
+                )
+            return curr_data[key]
+
+    raise ValueError(f"Secret [{secret_name}] not found")

ix-dev/stable/emby/migrations/migration_helpers/resources.py

@@ -2,8 +2,9 @@
 from .cpu import transform_cpu, CPU_COUNT
 
 
-def migrate_resources(resources, gpus=None):
+def migrate_resources(resources, gpus=None, system_gpus=None):
     gpus = gpus or {}
+    system_gpus = system_gpus or []
 
     result = {
         "limits": {
@@ -21,11 +22,38 @@ def migrate_resources(resources, gpus=None):
             {"memory": transform_memory(resources.get("limits", {}).get("memory", ""))}
         )
 
+    gpus_result = {}
     for gpu in gpus.items() if gpus else []:
-        if gpu[1] > 0 and ("amd" in gpu[0] or "intel" in gpu[0]):
-            result.update({"gpus": {"use_all_gpus": True}})
-            break
-        # We cannot migrate NVIDIA GPUs, as we don't know the UUIDs at this point
-        # and schema validation will fail.
+        kind = gpu[0].lower()  # Kind of gpu (amd, nvidia, intel)
+        count = gpu[1]  # Number of gpus user requested
+
+        if count == 0:
+            continue
+
+        if "amd" in kind or "intel" in kind:
+            gpus_result.update({"use_all_gpus": True})
+        elif "nvidia" in kind:
+            sys_gpus = [
+                gpu_item
+                for gpu_item in system_gpus
+                if gpu_item.get("error") is None
+                and gpu_item.get("vendor", None) is not None
+                and gpu_item.get("vendor", "").upper() == "NVIDIA"
+            ]
+            for sys_gpu in sys_gpus:
+                if count == 0:  # We passed # of gpus that user previously requested
+                    break
+                guid = sys_gpu.get("vendor_specific_config", {}).get("uuid", "")
+                pci_slot = sys_gpu.get("pci_slot", "")
+                if not guid or not pci_slot:
+                    continue
+
+                gpus_result.update(
+                    {"nvidia_gpu_selection": {pci_slot: {"uuid": guid, "use_gpu": True}}}
+                )
+                count -= 1
+
+    if gpus_result:
+        result.update({"gpus": gpus_result})
 
     return result