Das deutsche Python-Forum

Code: Alles auswählen

#!/usr/bin/env python3
from ast import literal_eval
from pathlib import Path

from psutil import disk_partitions, disk_usage

SELF_FILE_PATH = Path(__file__)


def format_size(value):
    power = 1024
    for unit in ["", "K", "M", "G", "T"]:
        if value < power:
            break
        value /= power

    return f"{value:.1f}{unit}"


def main():
    with SELF_FILE_PATH.open("a", encoding="utf-8") as self_file:
        for partition in disk_partitions():
            usage = disk_usage(partition.mountpoint)
            #
            # Do whatever ``awk ....`` in the example was supposed to do here.
            #
            text = (
                f"{usage.percent:4.1f}%\t{format_size(usage.free)}"
                f"\t{partition.mountpoint}"
            )
            #
            # Need to write the `repr()` form of `text` because the file must be
            # a syntactycally valid Python source file.
            #
            self_file.write(f"{text!r}\n")

    with SELF_FILE_PATH.open("r", encoding="utf-8") as lines:
        #
        # Skip lines up to and including special comment.
        #
        for line in lines:
            if line.startswith("# DATA STARTS HERE -"):
                break

        for line in lines:
            print(literal_eval(line))


if __name__ == "__main__":
    main()

# DATA STARTS HERE ------------------------------------------------------