pythontemplate.py

# vim:fileencoding=utf-8:expandtab:tabstop=8:shiftwidth=2:softtabstop=2:textwidth=120:noautoindent:nocindent:nosmartindent
##
## ~ $ PYTHONOPTIMIZE=0 python3 -i <your-script.py> ## when you are developing
## ~ $ python3 -I -E -O <your-script.py>  ## when you want to deploy...this removes the if __debug__ and assert checks!
## ~ $ python3 -I -E -OO <your-script.py> ## even more optimization... removes docstrings!

import re, os, sys, sqlite3, datetime, logging, typing, functools, pathlib, inspect, dataclasses as dc,pytest
import numpy as np, pandas as pd, pyarrow as pa, duckdb as ddb
import matplotlib.pyplot as plt,seaborn as sns
import hypothesis as hy, hypothesis.strategies as st
ddb.execute("SET GLOBAL pandas_analyze_sample = 100_000")

if __debug__: import pdb

T = typing.TypeVar("T")
P = typing.ParamSpec("P")

logging.basicConfig(filename=f"log-{str(datetime.datetime.now().date())}.log", level=logging.DEBUG, encoding="utf-8", format="{asctime} - {levelname} - {message!r}", style="{",  datefmt="%Y.%m.%dT%H:%M:%S%z")

def grid(axis="both"):
  "Show the grid...kinda like graph paper"
  plt.minorticks_on()
  plt.grid(which="major", linestyle="-", alpha=0.75, axis=axis)
  plt.grid(which="minor", linestyle=":", alpha=0.5, axis=axis)

def log(func: typing.Callable[P,T]) -> typing.Callable[P,T]:
  @functools.wraps(func)
  def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
    signature = ", ".join([f"{a!r}" for a in args] + [f"{k}={v!r}" for k,v in kwargs.items()])
    logger = logging.getLogger()
    logger.debug(f"function {func.__name__}{inspect.signature(func)} called ==> {func.__name__}({signature})")
    try:
      result = func(*args, **kwargs)
      return result
    except Exception as e:
      logger.exception(f"Exception raised in {func.__name__}. exception: {e!s}")
      raise e
  return wrapper


@log
def f1(**kwargs: typing.Dict[typing.Any,typing.Any]) -> typing.Any:
  print("key\tvalue")
  print("---\t-----")
  for k,v in kwargs.items():
    print(f"{k=}",f"{v=}",sep="\t")

Num = typing.TypeVar('Num', int, float, complex)
@log
def sum(a:Num, b:Num) -> Num: return a+b

@log
def prod(a:Num, b:Num) -> Num: return a*b

@log
@hy.settings(max_examples=500, verbosity=hy.Verbosity.verbose)
@hy.given(a=st.one_of(st.integers(), st.floats()), b=st.one_of(st.integers(), st.floats()))
def test_sum[T: int|float](a:T, b:T) -> None:
  """ This ought to fail for quite a few cases!!!

      Included here just to demonstrate how to use hypothesis for PBT.
  """
  assert a + b == sum(a,b)
  assert sum(a,b) >= a
  assert sum(a,b) >= b

@log
@pytest.mark.parametrize(("test_input","expected"),[("3+5",8),("2+4",6),pytest.param("6*9",42,marks=pytest.mark.xfail)])
def test_eval(test_input, expected): ## the args have to match args listed in the pytest.mark.parametrize tuple!
  assert eval(test_input) == expected

@log
def main():
  logging.info("In main")
  print("hello world!")
  f1(fname="vijay",lname="lulla",addr="mythical city",salary=5.2)
  f1(x=[1,2,3,4,5],y=[1,2,3,4,5])
  million, scaler = 1_000_000, 0.89
  if __debug__: pdb.set_trace()
  print(f"${million:_d} US = \u20ac{million*scaler:4_.2f}")
  print(f"{sum(2,18.5)=}\n{prod(2,18.5)=}")

  today = datetime.datetime.now()
  print(f"{today:%B %d, %Y}\n"
        f"{today=:%B %d, %Y}\n"
        f"{ today = :%B %d, %Y}" ) ## NOTE: whitespace preserved!

if __name__ == "__main__":
  main()