Skip to content

Commit

Permalink
;bin: ft, tt reporting scripts (examples)
Browse files Browse the repository at this point in the history
  • Loading branch information
simonmichael committed Jul 6, 2023
1 parent 1cfcb7f commit 37206ab
Show file tree
Hide file tree
Showing 2 changed files with 341 additions and 0 deletions.
161 changes: 161 additions & 0 deletions bin/ft
Original file line number Diff line number Diff line change
@@ -0,0 +1,161 @@
#!/usr/bin/env bash
# * ft - financial scripts, see below
# ** PREAMBLE
# shellcheck shell=bash disable=SC2317
# Customise as needed; consider keeping as a git checkout for merging updates.

set -e

rg="rg -IN --sort=path"
date=$(if [ "$(builtin type -p gdate)" ]; then echo gdate; else echo date; fi)
sed=$(if [ "$(builtin type -p gsed)" ]; then echo gsed; else echo sed; fi)

help() { # show this help
cat <<EOF
--------------------------------------------------------------------------------"; }
ft - finance tool: run financial reports and finance-related scripts
Usage: ft [COMMAND [ARGS]]
Commands:
$($rg '^\w.*\(\) *\{ *#' "$0" | $sed -e 's/() *{//' | column -t -s'#')
OTHERCMD [ARGS] run other hledger commands on the default journal
Add hledger options to customise reports.
EOF
}
# '(...|# \*\* [^#].*)' -or '$3 $1|$2'

# The main hledger file to import to and report on. It should exist.
JOURNAL="$LEDGER_FILE"

#DIR=${FINDIR:-~/finance}
DIR=$(dirname "$JOURNAL")
cd "$DIR"

PERIOD="1/1..tomorrow"
TODAY=$($date +%Y-%m-%d)
#TODAY=$(TZ=UTC $date +%Y-%m-%d)
#YEAR=$($date +%Y)

# ** IMPORT TXNS ------------------------------------------------------------

# where to import most hledger transactions from
IMPORTFILES=(\
bank1-checking.csv.rules \
bank1-savings.csv.rules \
paypal.csv \
)

# hledger .rules files(/globs) used for IMPORTFILES
IMPORTRULESFILES=$($sed -Ee 's/.csv /.csv.rules /g' <<< "${IMPORTFILES[*]}")
IMPORTRULESFILES+=(common.rules unify.rules wf.rules wf*checking.rules wf*savings.rules)

get-csv() { # download auto-downloadable CSVs (paypal)
paypaljson | paypaljson2csv > paypal.csv
}

import-dry() { # import new downloaded transactions to the journal, dry run
hledger -f "$JOURNAL" import "${IMPORTFILES[@]}" --dry-run
}

import() { # import new downloaded transactions to the journal, logging and not printing errors
date >>import.log
hledger -f "$JOURNAL" import "${IMPORTFILES[@]}" 2>>import.log || echo "Failed, check import.log"
echo "Use ledger-mode's M-q to align entries."
}

# ** IMPORT PRICES ------------------------------------------------------------

get-prices() { # [PRICEHISTFETCHOPTS] - download prices for main commodities (default: today's)
(pricehist fetch -o ledger -s "$TODAY" alphavantage EUR/USD "$@" | $sed -E 's/EUR/€/') &
(pricehist fetch -o ledger -s "$TODAY" alphavantage GBP/USD "$@" | $sed -E 's/GBP/£/') &
(pricehist fetch -o ledger -s "$TODAY" alphavantage JPY/USD "$@" | $sed -E 's/JPY/¥/')
# Parallelised for speed; do slowest last.
# Output order varies, can be sorted with LC_COLLATE=C.UTF-8 sort or hledger -f- prices.
}

# ** REPORTS ------------------------------------------------------------
# **** general ------------------------------------------------------------

bs() { # show balance sheet
hledger -f "$JOURNAL" bs --layout bare --pretty --drop 1 -p "$PERIOD" -E -5 "$@"
}

is() { # show income statement
hledger -f "$JOURNAL" is --layout bare --pretty --drop 1 -p "$PERIOD" -S "$@"
}

a() { # show assets
hledger -f "$JOURNAL" bal type:al -H --layout bare --pretty --drop 1 -p "$PERIOD" -E "$@"
}

r() { # show revenues
hledger -f "$JOURNAL" bal type:r --layout bare --pretty --drop 1 -p "$PERIOD" -S --invert "$@"
}

x() { # show expenses
hledger -f "$JOURNAL" bal type:x --layout bare --pretty --drop 1 -p "$PERIOD" -S --invert "$@"
}

ab() { # show assets bar chart
echo "Quarterly net worth:"
hledger-bar -v 200 -f "$JOURNAL" -Q type:al -H "$@"
}

rb() { # show revenues bar chart
echo "Quarterly revenues:"
hledger-bar -v 40 -f "$JOURNAL" -Q type:r --invert "$@"
}

xb() { # show expenses bar chart
echo "Quarterly expenses:"
hledger-bar -v 40 -f "$JOURNAL" -Q type:x --invert "$@"
}

# XXX with partial workaround for https://github.com/gooofy/drawilleplot/issues/4
al() { # show assets line chart
hledger -f "$JOURNAL" plot -- bal --depth=1 ^assets --historical --terminal --rcParams '{"figure.figsize":[8,3]}' --no-today -q --title "hledger assets" "$@" | $sed 's/⠀/ /g'
}

rl() { # show revenues line chart
hledger -f "$JOURNAL" plot -- bal --depth=1 ^revenues --monthly --invert --terminal --rcParams '{"figure.figsize":[8,3]}' --drawstyle 'steps-mid' --no-today -q --title "hledger monthly revenues" "$@" | $sed 's/⠀/ /g'
}

xl() { # show expenses line chart
hledger -f "$JOURNAL" plot -- bal --depth=1 ^expenses --monthly --terminal --rcParams '{"figure.figsize":[8,3]}' --drawstyle 'steps-mid' --no-today -q --title "hledger monthly expenses" "$@" | $sed 's/⠀/ /g'
}

forecast() { # print transactions predicted by forecast rules from last week on
hledger print --auto --forecast=lastweek.. -I tag:_generated "$@"
}

household() { # show a draft month-end household adjustment transaction for last month
env household "$($date -v-1m +%b)"
}

# **** tax ------------------------------------------------------------

# estimated-tax:
# @echo "Federal estimated tax due for this year"
# $(HLEDGER) register liabilities:personal:tax:federal:$(YEAR) --width=130
# @echo State estimated tax due for this year:
# @$(HLEDGER) register liabilities:personal:tax:state:$(YEAR) --width=130
# @echo

# **** business ------------------------------------------------------------

consulting() { # show consulting revenue
hledger -f "$JOURNAL" reg --invert 'revenues:(client1|client2)' -p "$PERIOD" "$@"
}

# **** other ------------------------------------------------------------

bin() { # [PAT] show all scripts in $DIR/bin/[bashrc] (default: ~/finance/)
env bin "$@"
}

# ** END

if [[ $# -eq 0 ]]; then help # no args shows help
elif declare -f "$1" > /dev/null; then "$@"; # arg 1 selects a function above
else hledger -f "$JOURNAL" "$@"; fi # or fall through to hledger
exit
180 changes: 180 additions & 0 deletions bin/tt
Original file line number Diff line number Diff line change
@@ -0,0 +1,180 @@
#!/usr/bin/env bash
# * tt - time scripts, see below
# ** PREAMBLE
# Customise as needed; consider keeping as a git checkout for merging updates.
# Uses hledger, bash, python, a few other unix tools.
# wakelog is a script grepping system logs for sleep/wake events; provide it or comment it.

## #!/usr/bin/env -S osh # -*- sh -*- # or osh - more powerful, less tool support
## shopt -s -errexit strict:all 2>/dev/null || set -e

# shellcheck shell=bash disable=SC2096 disable=SC2317

set -e

rg="rg -IN --sort=path"
date=$(if [ "$(builtin type -p gdate)" ]; then echo gdate; else echo date; fi)
sed=$(if [ "$(builtin type -p gsed)" ]; then echo gsed; else echo sed; fi)
stat=$(if [ "$(builtin type -p gstat)" ]; then echo gstat; else echo stat; fi)

help() { # show this help
line
cat <<EOF
--------------------------------------------------------------------------------
tt - time tool: run time reports and time-related scripts
Usage: tt [COMMAND [ARGS]]
Commands:
$($rg '^\w.*\(\) *\{ *#' "$0" | $sed -e 's/() *{//' | column -t -s'#')
OTHERCMD [ARGS] run other hledger commands on \$TIMELOG
EOF
}

#DIR=${FINDIR:-~/finance}
DIR=$(dirname "$TIMELOG")
cd "$DIR"

# The file where actual time data is logged, for dashboard's stats.
# This might or might not be the top-level $TIMELOG file.
#TIMELOGDATA=$TIMELOG
YEAR=$($date +%Y)
TIMELOGDATA="$DIR"/time-"$YEAR".timedot


# ** REPORTS ------------------------------------------------------------

# This redisplays only when a file listed by `hledger -f $TIMELOG files` is modified.
# To force a per minute display as well, have $TIMELOG include a dummy file (.update)
# and configure a cron job to touch that every minute.
# (This is better than touching the timelog file itself, which confuses editors.)
#
dash() { # show time dashboard, redisplaying when timelog files change
dir=$(dirname "$TIMELOG")
cd "$dir"
opts= #--poll=10 # <- uncomment to fix symlinked files being ignored
# shellcheck disable=SC2086
watchexec $opts --no-vcs-ignore --filter-file=<(hledger -f "$TIMELOG" files | sed -E "s|$dir/||g") -c -r tt status
}

# dash-1m() { # show time dashboard, redisplaying every minute with watch
# watch -n60 -c tt status
# }

status() { # show current time status
curtime=$($date +'%H:%M %Z, %a %b %-e %Y')
modtime=$($date +'%H:%M %Z' -r "$TIMELOGDATA")
modsecs=$($stat -c %Y "$TIMELOGDATA")
nowsecs=$($date +%s)
agesecs=$((nowsecs - modsecs))
agemins=$(python3 -c "print($agesecs/60)")
agehrs=$(python3 -c "print($agesecs/3600.0)")
ageqtrhrs=$(python3 -c "print(round($agesecs/900.0))")
agedots=$(dots "$ageqtrhrs")
printf "Current time: %s\n" "$curtime"
# for osh, whose printf doesn't support floating point yet: env runs the system printf
env printf "Timelog saved: %s, %.0fm / %.1fh / %s ago\n" "$modtime" "$agemins" "$agehrs" "$agedots"

# Show the current day/week/month budget status.
printf "Time plans:\n"
# cf budgets()
# calculate each period's budget from daily budget
hledger -f "$TIMELOG" bal -1 -p 'daily today' --budget=Daily | tail +2
hledger -f "$TIMELOG" bal -1 -p 'weekly this week' --budget=Daily | tail +2
hledger -f "$TIMELOG" bal -1 -p 'monthly this month' --budget=Daily | tail +2
# or use each period's specific budget
# hledger -f "$TIMELOG" bal -p 'daily today' --budget=Daily -1 | tail +2
# hledger -f "$TIMELOG" bal -p 'weekly this week' --budget=Weekly -1 | tail +2
# hledger -f "$TIMELOG" bal -p 'monthly this month' --budget=Monthly -1 | tail +2

echo
hledger -f "$TIMELOG" check -s tags ordereddates || true

# this comes last because it's slow and variable length
echo
printf "Display activity:\n"
wakelog today | tail -n 6
}

what() { # what happened ? Show largest balances first, today and depth 1 by default
hledger -f "$TIMELOG" bal -S -1 -p today "$@"
}

dots() { # print line of N dots, grouped in 4s (suitable for timedot)
n="$1"
ndiv4=$((n/4))
nmod4=$((n-n/4*4))
sep=''
while [[ $ndiv4 -gt 0 ]]; do ndiv4=$((ndiv4-1)); echo -n "$sep...."; sep=' '; done
while [[ $nmod4 -gt 0 ]]; do nmod4=$((nmod4-1)); echo -n "$sep."; sep=''; done
echo
}

#RFLAGS=-tMTA
RFLAGS=-tM

x() { # horizontal time summary this year, monthly by default
hledger -f "$TIMELOG" bal -1 "$RFLAGS" "$@"
}

y() { # vertical time summary this year, monthly by default
hledger -f "$TIMELOG" bal -1 "$RFLAGS" --transpose "$@"
}

rweeks() { # recent weeks' time budgets
printf "\nPast weeks:\n"
timeweekly past
}

weeks() { # this and last week's time budgets
printf "\nLast week, this week:\n"
timeweekly run
}

unused() { # unused & undeclared account names
echo "Unused: (but declared)"
hledger -f "$TIMELOG" acc --unused "$@" --directives | gsed -E 's/:(.)/.\1/g'
echo
echo "Undeclared: (but used)"
hledger -f "$TIMELOG" acc --undeclared "$@" --directives | gsed -E 's/:(.)/.\1/g'
}

unusedcat() { # unused & undeclared account names by category
for a in $(tt acc -1); do line; echo "$a":; tt unused "^$a"; echo; done; line
}

# Budget reports. cf status()

budgets() { # show monthly time budget performance this year
x --budget=daily -M -p jan..tomorrow "$@"
}

budgetsy() { # show monthly time budget performance this year, vertically
# cf status()
y --budget=daily -M -p jan..tomorrow "$@"
}

# dedicated commands needed to set proper week start date for simple headings:
budgetsw() { # show weekly time budget performance this year
# cf status()
y --budget=daily -W -p 3/27..tomorrow "$@"
}

budgetswx() { # show weekly time budget performance this year, horizontally
# cf status()
x --budget=daily -W -p 3/27..tomorrow "$@"
}

addacc() { # add declarations for all undeclared accounts
# shellcheck disable=SC2094
hledger -f "$TIMELOG" accounts --undeclared --directives | sed 's/:/./g' >>"$TIMELOG"
}

hours() { # show a bar chart of daily hours
hledger-bar -v 1 -f "$TIMELOG" -D "$@"
}

# ** END

if [[ $# -eq 0 ]]; then help # no args shows help
elif declare -f "$1" > /dev/null; then "$@"; # arg 1 selects a function above
else hledger -f "$TIMELOG" "$@"; fi # or fall through to hledger
exit

0 comments on commit 37206ab

Please sign in to comment.