diff options
Diffstat (limited to 'src/utils/disk.py')
| -rw-r--r-- | src/utils/disk.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/utils/disk.py b/src/utils/disk.py new file mode 100644 index 0000000..4f11a82 --- /dev/null +++ b/src/utils/disk.py @@ -0,0 +1,19 @@ +# +# Copyright (C) 2022 Soleta Networks <info@soleta.eu> +# +# This program is free software: you can redistribute it and/or modify it under +# the terms of the GNU Affero General Public License as published by the +# Free Software Foundation; either version 3 of the License, or +# (at your option) any later version. + +import os + +def get_disks(): + """ + Walks /sys/block/ and returns files starting with 'sd', + 'nvme' or 'vd' + """ + return sorted([ dev for dev in os.listdir('/sys/block/') + if dev.startswith('sd') + or dev.startswith('nvme') + or dev.startswith('vd')]) |