milestone.py

#!/usr/bin/env python3

import json, re, sys
from itertools import groupby
from lib.config import *
from lib import sharesight
from lib import webhook
from lib import util
from lib import yahoo

# BUGS:
# - need to handle stock splits: flush cache if we see a split in the data
# https://finance.yahoo.com/quote/TSLA/history?period1=0&period2=1707609600&filter=split

def lambda_handler(chat_id=config_telegramChatID, specific_stock=None, service=None, interactive=False):
	def prepare_milestone_payload(service, market_data):
		emoji = '⭐'
		payload = []
		payloadhigh = []
		payloadlow = []
		payloadprofit = []
		payloadcashflow = []
		payloadprofitneg = []
		payloadcashflowneg = []
		oldhigh = float()
		oldlow = float()
		records = util.json_load('finbot_milestone.json', persist=True)
		if records is None:
			records = {}
		for ticker in tickers:
			try:
				oldhigh = records[ticker]['high']
				oldlow = records[ticker]['low']
			except (KeyError, TypeError):
				records[ticker] = {}
				records[ticker]['high'] = float()
				records[ticker]['low'] = float()
				try:
					oldhigh, oldlow = yahoo.historic_high(ticker) # first run to prime cache file
				except (KeyError, TypeError):
					continue
			try:
				newhigh = round(market_data[ticker]['fiftyTwoWeekHigh'], 2)
				newlow = round(market_data[ticker]['fiftyTwoWeekLow'], 2)
			except (KeyError, ValueError):
				continue

			newprofit = False
			newcashflow = False
			detail = yahoo.fetch_detail(ticker)
			if ticker in detail:
				if 'net_income' in detail[ticker] and detail[ticker]['net_income'] > 0:
					newprofit = True
				if 'free_cashflow' in detail[ticker] and detail[ticker]['free_cashflow'] > 0:
					newcashflow = True
			try:
				oldprofit = records[ticker]['profit']
				oldcashflow = records[ticker]['cashflow']
			except (KeyError, TypeError):
				oldprofit = newprofit
				oldcashflow = newcashflow

			title = market_data[ticker]['profile_title']
			ticker_link = util.finance_link(ticker, market_data[ticker]['profile_exchange'], service)
			emoji = util.flag_from_ticker(ticker)
			records[ticker]['high'] = newhigh
			records[ticker]['low'] = newlow
			records[ticker]['profit'] = newprofit
			records[ticker]['cashflow'] = newcashflow
			if newhigh > oldhigh:
				print("DEBUG", ticker, newhigh, "is higher than", oldhigh) if debug else None
				payloadhigh.append(f"{emoji} {title} ({ticker_link})")
			if newlow < oldlow:
				print("DEBUG", ticker, newlow, "is lower than", oldlow) if debug else None
				payloadlow.append(f"{emoji} {title} ({ticker_link})")
			if oldprofit != newprofit:
				print("DEBUG", ticker, "profitability changed", oldprofit, newprofit) if debug else None
				if newprofit:
					payloadprofit.append(f"{emoji} {title} ({ticker_link})")
				else:
					payloadprofitneg.append(f"{emoji} {title} ({ticker_link})")
			if oldcashflow != newcashflow:
				print("DEBUG", ticker, "cashflow changed", oldcashflow, newcashflow) if debug else None
				if newcashflow:
					payloadcashflow.append(f"{emoji} {title} ({ticker_link})")
				else:
					payloadcashflowneg.append(f"{emoji} {title} ({ticker_link})")
		payloadlow.sort()
		payloadhigh.sort()
		payloadprofit.sort()
		payloadcashflow.sort()
		payloadprofitneg.sort()
		payloadcashflowneg.sort()
		if payloadhigh:
			message = 'Record high:'
			message = webhook.bold(message, service)
			payload.append(message)
			payload = payload + payloadhigh
		if payloadlow:
			payload.append("")
			message = 'Record low:'
			message = webhook.bold(message, service)
			payload.append(message)
			payload = payload + payloadlow
		if payloadprofit:
			payload.append("")
			message = 'Became profitable'
			message = webhook.bold(message, service)
			payload.append(message)
			payload = payload + payloadprofit
		if payloadcashflow:
			payload.append("")
			message = 'Became cashflow positive'
			message = webhook.bold(message, service)
			payload.append(message)
			payload = payload + payloadcashflow
		if payloadprofitneg:
			payload.append("")
			message = 'Became unprofitable'
			message = webhook.bold(message, service)
			payload.append(message)
			payload = payload + payloadprofitneg
		if payloadcashflowneg:
			payload.append("")
			message = 'Became cashflow negative'
			message = webhook.bold(message, service)
			payload.append(message)
			payload = payload + payloadcashflowneg
		payload = [i[0] for i in groupby(payload)] # de-dupe white space
		return payload, records

	# MAIN #

	market_data = {}
	if specific_stock:
		ticker = util.transform_to_yahoo(specific_stock)
		tickers = [ticker]
	else:
		tickers = util.get_holdings_and_watchlist()
		market_data = yahoo.fetch(tickers)

	# Prep and send payloads
	if not webhooks:
		print("Error: no services enabled in .env", file=sys.stderr)
		sys.exit(1)
	else:
		for service, url in webhooks.items():
			payload, records = prepare_milestone_payload(service, market_data)
			if service == "telegram":
				url = url + "sendMessage?chat_id=" + str(chat_id)
			webhook.payload_wrapper(service, url, payload, chat_id)

	if records:
		util.json_write('finbot_milestone.json', records, persist=True)

	# make google cloud happy
	return True

if __name__ == "__main__":
	lambda_handler()