#!/usr/bin/env python
from __future__ import print_function
import argparse
from pydfu.scanner import Scanner
from pydfu import df as DiskFree

parser = argparse.ArgumentParser(description='Disk and FileSystem Utils')
parser.add_argument('-s', '--scan', action='store_true', default=False)
parser.add_argument('-d', '--df', action='store_true', default=False)

argv = parser.parse_args()

if argv.scan:
    scanner = Scanner()
    scanner.run()

if argv.df:
    df = DiskFree()
    title = [ 'FileSystem', '1K-blocks', 'Used', 'Avail', 'Use%', 'Mounted On' ]
    data = [ fs.for_print() for fs in df ]
    data.insert(0, title)
    col_width = max(len(word) for row in data for word in row)# padding
    for row in data:
        print("".join(word.ljust(col_width) for word in row))


