From 28d9085231dc4d13e7ac50b1caeedbd6c0cb3270 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Mon, 13 Jan 2020 17:48:22 -0700
Subject: [PATCH] Fixed getting filesystem UUID

---
 scripts/wk/kit/ufd.py | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/scripts/wk/kit/ufd.py b/scripts/wk/kit/ufd.py
index dcbd572b..2a4be45e 100644
--- a/scripts/wk/kit/ufd.py
+++ b/scripts/wk/kit/ufd.py
@@ -15,7 +15,6 @@ from wk import io, log, std
 from wk.cfg.main import KIT_NAME_FULL, KIT_NAME_SHORT
 from wk.cfg.ufd import BOOT_ENTRIES, BOOT_FILES, ITEMS, ITEMS_HIDDEN, SOURCES
 from wk.exe import run_program
-from wk.hw.obj import Disk
 from wk.os import linux
 
 
@@ -219,6 +218,22 @@ def find_first_partition(dev_path):
   return part_path
 
 
+def get_uuid(path):
+  """Get filesystem UUID via findmnt, returns str."""
+  cmd = [
+    'findmnt',
+    '--noheadings',
+    '--target', path,
+    '--output', 'uuid'
+    ]
+
+  # Run findmnt
+  proc = run_program(cmd, check=False)
+
+  # Done
+  return proc.stdout.strip()
+
+
 def hide_items(ufd_dev, items):
   """Set FAT32 hidden flag for items."""
   first_partition = find_first_partition(ufd_dev)
@@ -392,8 +407,7 @@ def show_selections(args, sources, ufd_dev, ufd_sources):
 def update_boot_entries():
   """Update boot files for UFD usage"""
   configs = []
-  ufd = Disk('/mnt/UFD')
-  uuid = ufd.details.get('uuid')
+  uuid = get_uuid('/mnt/UFD')
 
   # Find config files
   for c_path, c_ext in BOOT_FILES.items():