docker-playground/pip_venv_autogluon/app.py

import pandas as pd 
import numpy as np 
import matplotlib.pyplot as plt
import plotly.graph_objs as go
import matplotlib
import plotly
import papermill
import hdfs

def print_version():
    import sys
    from hdfs_kerberos_no_verify import NoVerifyKerberosClient as KerberosClient
    print("python version", sys.version)
    print("pandas version ", pd.__version__)
    print("numpy version ", np.__version__)
    # print("autogluon.tabular", autogluon.tabular.__version__)
    print("matplotlib version ", matplotlib.__version__)

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()
    print_version()
    venv_size()