shorts.py

#!/usr/bin/env python3

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

def lambda_handler(chat_id=config_telegramChatID, threshold=config_shorts_percent, specific_stock=None, service=None, interactive=False):
    def prepare_shorts_payload(service, market_data):
        payload = []
        emoji = "🩳"
        for ticker in tickers:
            try:
                short_percent = float(market_data[ticker]['short_percent'])
            except (KeyError, ValueError):
                continue
            if '.AX' in ticker:
                url = 'https://www.shortman.com.au/stock?q=' + ticker.split('.')[0]
                short_interest_link = util.link(url, ticker.split('.')[0], service)
            else:
                url = 'https://finance.yahoo.com/quote/' + ticker + '/key-statistics?p=' + ticker
                short_interest_link = util.link(url, ticker.split('.')[0], service)
            title = market_data[ticker]['profile_title']
            if short_percent > threshold or specific_stock:
                payload.append([emoji, title, f'({short_interest_link})', short_percent])

        def last_element(e):
            return e[-1]
        payload.sort(key=last_element)
        for i, e in enumerate(payload):
            e[-1] = str(round(e[-1])) + '%'
            payload[i] = ' '.join(e)

        if payload:
            if not specific_stock:
                message = f'Tracked stocks shorted over {threshold}%:'
                message = webhook.bold(message, service)
                payload.insert(0, message)
        else:
            if interactive:
                if specific_stock:
                    payload = [f"{emoji}No short interest found for {tickers[0]}"]
                else:
                    payload = [f"{emoji}No stocks shorted over {threshold}%. Try specifying a number."]
        return payload

    # MAIN #

    if specific_stock:
        ticker = util.transform_to_yahoo(specific_stock)
        tickers = [ticker]
    else:
        tickers = util.get_holdings_and_watchlist()

    market_data = yahoo.fetch(tickers)
    for ticker in tickers:
        if '.' not in ticker:
            try:
                market_data = market_data | yahoo.fetch_detail(ticker)
            except TypeError:
                pass
    print("")
    market_data = shortman.fetch(market_data)

    # Prep and send payloads
    if not webhooks:
        print("Error: no services enabled in .env", file=sys.stderr)
        sys.exit(1)
    if interactive:
        payload = prepare_shorts_payload(service, market_data)
        url = webhooks[service]
        if service == "slack":
            url = 'https://slack.com/api/chat.postMessage'
        elif service == "telegram":
            url = url + "sendMessage?chat_id=" + str(chat_id)
        webhook.payload_wrapper(service, url, payload, chat_id)
    else:
        for service, url in webhooks.items():
            payload = prepare_shorts_payload(service, market_data)
            if service == "telegram":
                url = url + "sendMessage?chat_id=" + str(chat_id)
            webhook.payload_wrapper(service, url, payload, chat_id)

    # make google cloud happy
    return True

if __name__ == "__main__":
    lambda_handler()