From 79e88874b14e278884f0e0ea12e8b2ead80bd804 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alin=20Tr=C4=83istaru?= <alin.traistaru+github@gmail.com>
Date: Sat, 29 Apr 2017 10:19:06 +0300
Subject: [PATCH] linux: display total size of disk

---
 scripts/usb.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/scripts/usb.py b/scripts/usb.py
index e24d7f2..2897edb 100644
--- 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 = ""