diff --git a/dotfiles/i3/blocks b/dotfiles/i3/blocks index a7480d6..3e439d6 100644 --- a/dotfiles/i3/blocks +++ b/dotfiles/i3/blocks @@ -54,7 +54,7 @@ LABEL= [free_disk_space] interval=10 LABEL= -DISK=/home/flyingscorpio +DISK=$HOME [corona_stats] COUNTRY=France diff --git a/dotfiles/i3/scripts/free_disk_space b/dotfiles/i3/scripts/free_disk_space index b160c5a..be26288 100755 --- a/dotfiles/i3/scripts/free_disk_space +++ b/dotfiles/i3/scripts/free_disk_space @@ -11,6 +11,12 @@ import os def get_disk_free_space(disk: str) -> str: """Return available disk space in GiB.""" + disk = ( + disk.replace("$USER", os.environ["USER"]) + .replace("~", os.environ["HOME"]) + .replace("$HOME", os.environ["HOME"]) + ) + stat = os.statvfs(disk) total = stat.f_blocks * stat.f_bsize / 1024 / 1024 / 1024 available = stat.f_bavail * stat.f_bsize / 1024 / 1024 / 1024 @@ -21,6 +27,6 @@ def get_disk_free_space(disk: str) -> str: return "{} {}".format(avail_format, percent_format) -OUTPUT = os.environ['LABEL'] + get_disk_free_space(os.environ["DISK"]) +OUTPUT = os.environ["LABEL"] + get_disk_free_space(os.environ["DISK"]) print(OUTPUT)