iTerm2のステータスバーにディスクの使用量を表示するコンポーネント

storage.py

#!/usr/bin/env python3.7

# iTerm2のステータスバーにストレージの使用量を表示するコンポーネント
# The component that displays storage usage in the iTerm2 status bar

from iterm2 import Connection, StatusBarComponent, StatusBarRPC, run_forever
from iterm2.statusbar import Knob
import os
from decimal import Decimal, ROUND_HALF_UP

# SI接頭辞 SI prefix
si_list = ["", "K", "M", "G", "T", "P", "X", "Z", "Y"]

# 最適なSI接頭辞付きの値に計算します。
# Calculates to the value with the best SI prefix
def calc_si( val ):
    dval = Decimal( val )
    expr = int( dval.log10() )
    si = int( dval.log10() / 3 )
    mantissa = dval / ( 10 ** expr )
    # e.g.: 12,340,000,000 -> 12.34 G
    qval = float( mantissa.quantize( Decimal( '.001' ), rounding = ROUND_HALF_UP ) * ( 10 ** ( expr % 3 ) ) )
    return ( qval, si_list[si] )

async def main( connection ):
    # コンポーネントのメタデータ Metadata
    componet = StatusBarComponent(
        short_description = "Storage",
        detailed_description = "Show storage usage",
        knobs = [],
        exemplar = "💾 50.0 GB / 1.0 TB",
        update_cadence = 300,   # 5分毎更新 5 min.
        identifier = "net.chronoir.iterm2.storage"
    )

    # ストレージの使用量と総容量を計測し、ステータス文字列を生成します。
    # Measures storage usage and total capacity, and generates a status string.
    @StatusBarRPC
    async def storage_status( knobs ):
        try:
            storage = os.statvfs( "/" )
        except:
            print( "ERROR: Failed to measure storage." )
            return "❌[ERROR]"

        storage_used = calc_si( storage.f_frsize * ( storage.f_blocks - storage.f_bfree ) )
        storage_total = calc_si( storage.f_frsize * storage.f_blocks )

        status_str = f"💾 {storage_used[0]} {storage_used[1]}B / {storage_total[0]} {storage_total[1]}B"

        return status_str
    
    # コンポーネントにコルーチンを登録します。 Registers the co-routine to the component.
    await componet.async_register( connection = connection, coro = storage_status, timeout = None )

run_forever( main )