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 == False and newprofit == True:
			#	print("DEBUG", ticker, "profitability changed", oldprofit, newprofit) if debug else None
			#	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()