48k.eu-mirror
/
ogclient
mirror of https://git.48k.eu/ogclient
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

utils: add get_partition_device 

Maps a disk and partition number to the corresponding device path
('/dev/sdXY', '/dev/nvmeXnYpZ', ...)
more_events
Jose M. Guisado 2022-08-30 16:58:24 +02:00
parent 87e738b33f
commit 52838e2ce6
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/utils/disk.py
View File

@ -8,6 +8,9 @@
import os import os
import fdisk
def get_disks(): def get_disks():
""" """
Walks /sys/block/ and returns files starting with 'sd', Walks /sys/block/ and returns files starting with 'sd',
@ -17,3 +20,17 @@ def get_disks():
if dev.startswith('sd') if dev.startswith('sd')
or dev.startswith('nvme') or dev.startswith('nvme')
or dev.startswith('vd')]) 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')