touchmpl.sh

#!/usr/bin/bash

filename=$1

if [[ $# -eq 0 ]]; then
    >&2 echo "Error: no arguments provided"
    >&2 echo "USAGE: $(basename $0) [NEW_FILE_NAME]"
    exit 1
fi

touch ${filename}.py
chmod u+x ${filename}.py

template() {
    cat <<'EOF'
#!/usr/bin/env python3
"""
Description

Input:
    ...
Output:
    ...
Prerequisites:
    ...
"""

import sys
import argparse
import numpy as np
import pandas as pd
import polars as pl
import seaborn as sns
import matplotlib.pyplot as plt
from pathlib import Path
from matplotlib import rcParams
import matplotlib.font_manager as fm
import matplotlib.ticker as ticker
from matplotlib.ticker import FormatStrFormatter


#=============================================
# set font
arial_font = "/home/james/Downloads/arial.ttf"
arial_font_bold = "/home/james/Downloads/Arial Bold.ttf"
fm.fontManager.addfont(arial_font)
fm.fontManager.addfont(arial_font_bold)
rcParams["font.sans-serif"] = "Arial"
rcParams["font.family"] = "Arial"
rcParams["font.size"] = 10

# =============================================================
def parse_args() -> argparse.Namespace:
    """Argument parsing"""
    parser = argparse.ArgumentParser(
        description=__doc__,
        formatter_class=argparse.RawDescriptionHelpFormatter,
    )

    parser.add_argument(
        "-i",
        "--input",
        dest="input",
        type=Path,
        metavar="IN",
        required=True,
        help="Path to input file [Required]",
    )

    parser.add_argument(
        "-o",
        "--outdir",
        dest="outdir",
        type=Path,
        required=False,
        default="plots",
        metavar="DIR",
        help="Output target directory [Optional][Default: $(pwd)/plots]",
    )

    parser.add_argument(
        "-D",
        "--dpi",
        dest="dpi",
        type=int,
        required=False,
        default=300,
        metavar="N",
        help="Resolution in dpi for output .png [Default: 300]",
    )

    parser.add_argument(
        "-F",
        "--arial_fonts",
        dest="arial_fonts",
        type=Path,
        required=False,
        default="./auxfiles",
        metavar="TFF",
        help="Path to Arial fonts dir",
    )

    args = parser.parse_args()

    return args


# or import from sys
file = Path(sys.argv[1])

#=============================================
# uses argparse to take input

def custom_plot(file, filename):
    """Plot"""

    # read data
    df = pl.read_csv(
        file,
        separator="\t",
        has_header=False,
        new_columns=["col1", "col2"]
    )
    print(df)

    # make plot
    fig, ax = plt.subplots(figsize=(5,5))

    # write plot
    plt.savefig(
        f"./plot_{filename}.png",
        format="png",
        dpi=300,
        bbox_inches="tight",
    )
    plt.tight_layout()
    plt.show()
    plt.close("all")


custom_plot(args.input, args.input.stem)

# =============================================================
def funca(args):
    """Description

    ---
    Args:
        arg1 (dtype): description

    Returns:
        dtype: description
    """
    # stuff
    print("Hello world")


# =============================================================
def main() -> None:
    """Workflow:
    ---
    main
     ├── args
     └── func
     │
    """
    # get args
    args = parse_args()

    # func
    funca(args)


if __name__ == "__main__":
    sys.exit(main())
EOF
}

template >"${filename}.py"