docker-playground/simple_conda/app.py

def venv_size():
    import os
    import shutil

    def get_directory_size(directory):
        total_size = 0
        for path, dirs, files in os.walk(directory):
            for f in files:
                fp = os.path.join(path, f)
                total_size += os.path.getsize(fp)
        return total_size

    def format_size(size):
        power = 2**10
        n = 0
        power_labels = {0 : '', 1: 'KB', 2: 'MB', 3: 'GB', 4: 'TB'}
        while size > power:
            size /= power
            n += 1
        return f"{size:.2f} {power_labels[n]}"

    directory = '/venv'
    size = get_directory_size(directory)
    human_readable_size = format_size(size)
    print(f"The size of {directory} is {human_readable_size}")


def get_current_user():
    import getpass
    current_user = getpass.getuser()
    print(current_user)

if __name__ == "__main__":
    get_current_user()
    venv_size()