From 52838e2ce604ea4edd56f0556a6b50d52b307a8d Mon Sep 17 00:00:00 2001
From: "Jose M. Guisado" <jguisado@soleta.eu>
Date: Tue, 30 Aug 2022 16:58:24 +0200
Subject: utils: add get_partition_device

Maps a disk and partition number to the corresponding device path
('/dev/sdXY', '/dev/nvmeXnYpZ', ...)
---
 src/utils/disk.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'src/utils/disk.py')

diff --git a/src/utils/disk.py b/src/utils/disk.py
index 4f11a82..e26983c 100644
--- a/src/utils/disk.py
+++ b/src/utils/disk.py
@@ -8,6 +8,9 @@
 
 import os
 
+import fdisk
+
+
 def get_disks():
     """
     Walks /sys/block/ and returns files starting with 'sd',
@@ -17,3 +20,17 @@ def get_disks():
                     if dev.startswith('sd')
                     or dev.startswith('nvme')
                     or dev.startswith('vd')])
+
+
+def get_partition_device(disknum, partnum):
+    """
+    Returns the device path, given a disk and partition number
+    """
+    disk = get_disks()[disknum-1]
+    cxt = fdisk.Context(f'/dev/{disk}')
+
+    for pa in cxt.partitions:
+        if pa.partno == partnum - 1:
+            return cxt.partition_to_string(pa, fdisk.FDISK_FIELD_DEVICE)
+
+    raise ValueError('No such partition')
-- 
cgit v1.2.3-18-g5258