linux: display total size of disk

7 years ago · 79e88874b1
parent 1782495aea
commit 79e88874b1
1 changed files with 2 additions and 1 deletions
--- a/scripts/usb.py
+++ b/scripts/usb.py
@ -258,7 +258,8 @@ def details_udev(usb_disk_part):
        size_free = shutil.disk_usage(mount_point)[2]

    else:
-        size_total = device.get('UDISKS_PARTITION_SIZE') or ""
+        fdisk_cmd = 'fdisk -l ' + usb_disk_part + ' | grep "^Disk /" | sed -re "s/.*\s([0-9]+)\sbytes.*/\\1/"'
+        size_total = subprocess.check_output(fdisk_cmd, shell=True).strip()
        size_used = ""
        size_free = ""
        mount_point = ""