File: pp.sh
   1 #!/bin/sh
   2 
   3 # The MIT License (MIT)
   4 #
   5 # Copyright (c) 2026 pacman64
   6 #
   7 # Permission is hereby granted, free of charge, to any person obtaining a copy
   8 # of this software and associated documentation files (the "Software"), to deal
   9 # in the Software without restriction, including without limitation the rights
  10 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  11 # copies of the Software, and to permit persons to whom the Software is
  12 # furnished to do so, subject to the following conditions:
  13 #
  14 # The above copyright notice and this permission notice shall be included in
  15 # all copies or substantial portions of the Software.
  16 #
  17 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  20 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  22 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  23 # SOFTWARE.
  24 
  25 
  26 # pp [options...] [python expressions...]
  27 #
  28 # Print Python evaluates and prints the results of all the python expressions
  29 # given.
  30 #
  31 # The options are, available both in single and double-dash versions
  32 #
  33 #   -h, -help       show this help message
  34 #   -m, -math       import everything from `math` globally, for convenience
  35 #   -v, -verbose    show an ANSI-styled banner for each command being run
  36 
  37 
  38 verbose=0
  39 imports='import itertools, math, string, statistics'
  40 
  41 while [ $# -gt 0 ]; do
  42     case "$1" in
  43         -h|--h|-help|--help)
  44             awk '/^# +pp /, /^$/ { gsub(/^# ?/, ""); print }' "$0"
  45             exit 0
  46         ;;
  47 
  48         -m|--m|-math|--math)
  49             imports="${imports}"'; from math import *'; shift; continue
  50         ;;
  51 
  52         -v|--v|-verbose|--verbose) verbose=1; shift; continue ;;
  53 
  54         -) break ;;
  55 
  56         --) shift; break ;;
  57 
  58         -*)
  59             printf "unsupported option '%s'\n" "$1" >&2
  60             exit 1
  61         ;;
  62     esac
  63 
  64     break
  65 done
  66 
  67 res=0
  68 
  69 for arg in "$@"; do
  70     if [ "${verbose}" -eq 1 ]; then
  71         printf "\033[7m%s\033[0m\n" "${arg}" >&2
  72     fi
  73 
  74     python -c "${imports}; print(${arg})"
  75     code=$?
  76 
  77     if [ "${code}" -ne 0 ]; then
  78         res="${code}"
  79     fi
  80 done
  81 
  82 exit "${res}"