diff options
| -rwxr-xr-x | bin/disk_info | 23 |
1 files changed, 15 insertions, 8 deletions
diff --git a/bin/disk_info b/bin/disk_info index 75d142c..8b2cc0a 100755 --- a/bin/disk_info +++ b/bin/disk_info @@ -6,7 +6,7 @@ # # Copyright 2015 Canonical Ltd. # -# Authors: +# Authors: # Nelson Chu <Nelson.Chu@itri.org.tw> # Jeff Lane <jeff@ubuntu.com> # @@ -26,7 +26,8 @@ import sys import xml.etree.ElementTree as ET from subprocess import check_output -def get_item(disk,attribute): + +def get_item(disk, attribute): try: attribute_name = disk.find(attribute).text except AttributeError: @@ -34,6 +35,7 @@ def get_item(disk,attribute): return attribute_name + def main(): hwinfo_xml = check_output(['lshw', '-c', 'disk', '-xml']) root = ET.fromstring(hwinfo_xml) @@ -48,15 +50,20 @@ def main(): result = {} for disk in disk_list: if disk.get('id') == 'disk': - print("Name: {}".format(get_item(disk,'logicalname'))) - print("\t{k:15}\t{v}".format(k="Description:",v=get_item(disk,'description'))) - print("\t{k:15}\t{v}".format(k="Vendor:",v=get_item(disk,'vendor'))) - print("\t{k:15}\t{v}".format(k="Product:",v=get_item(disk,'product'))) + print("Name: {}".format(get_item(disk, 'logicalname'))) + print("\t{k:15}\t{v}".format(k="Description:", + v=get_item(disk, 'description'))) + print("\t{k:15}\t{v}".format(k="Vendor:", + v=get_item(disk, 'vendor'))) + print("\t{k:15}\t{v}".format(k="Product:", + v=get_item(disk, 'product'))) try: - disk_size = ("%dGB" % (int(disk.find('size').text) / (1000**3))) + disk_size = ("%dGB" % ( + int(disk.find('size').text) / (1000**3))) except: disk_size = "Unknown" - print("\t{k:15}\t{v}".format(k="Size:",v=disk_size)) + print("\t{k:15}\t{v}".format(k="Size:", + v=disk_size)) return 0 if __name__ == '__main__': |