Size of Python packages

poetry_package_sizer.sh

#!/bin/bash

# this script will install a single package into a virtual environment using poetry,
# check the size of the installed dependencies, and clean up.
#
# usage:
# ./poetry_package_sizer.sh "notebook pandas polars altair bokeh matplotlib panel plotly streamlit fastapi flask django rdkit torch"

# Poetry version: 1.5.1
# Python version (change to yours)
PYTHON=python3.10

PWD=$(pwd)

for PKG in $1
do
  echo -e "\n================================================================================================\n"
  NAME="size$PKG"
  poetry new $NAME
  cd $NAME
  poetry config virtualenvs.in-project true --local
  poetry add $PKG
  cd .venv/lib/$PYTHON/site-packages/

  du -d1 | grep -Ev "pycache|dist-info" | \
    sed -e 's/\.$/TOTAL/g' | sed 's/\.\///g' | sort -nk1 | \
    awk '{printf "%32s", $2}{printf "%12s\n", $1}'

  cd $PWD
  rm -rf $NAME
done