Powerline disk space widget

disk.py

import os
import humanfriendly


def disk(pl, monitor="/", threshold_gig=10):
        partition = os.statvfs(monitor)
        available = partition.f_bsize * partition.f_bavail
        available_gig = available / 1024 ** 2
        threshold_crit_gig = threshold_gig*0.5
        ret = []
        color_pct = 0
        if available_gig > threshold_gig:
                color_pct = 0
        elif available_gig > threshold_crit_gig:
                color_pct = (available_gig - threshold_crit_gig) * 100 / threshold_gig
        else:
                color_pct = 100
        ret.append({
                'contents': humanfriendly.format_size(available),
                'highlight_groups': ['disk_gradient'],
                'divider_highlight_group': 'background:divider',
                'gradient_level': color_pct,
        })
        return ret