sbp2-all.py

from flask import Flask, request
import alpaca_trade_api as tradeapi
import requests
import asyncio
import json

# Constants used throughout the script (names are self-explanatory). 
# Must hard code SLACK TOKEN, KEY_ID, SECRET_KEY, channel
WRONG_NUM_ARGS = "ERROR: Incorrect amount of args.  Action did not complete."
BAD_ARGS = "ERROR: Request error.  Action did not complete."
SLACK_TOKEN = "SLACK_TOKEN_HERE"
KEY_ID = "KEY_ID_HERE"
SECRET_KEY = "SECRET_KEY_HERE"

# Set up environment
api = tradeapi.REST(
    KEY_ID,
    SECRET_KEY,
    base_url="https://paper-api.alpaca.markets")

# Initialize the Flask object which will be used to handle HTTP requests
# from Slack
app = Flask(__name__)

# Execute an order.  Must contain 5, 6, or 7 arguments: type, symbol,
# quantity, side, time in force, limit price (optional), and stop price
# (optional).


@app.route("/order", methods=["POST"])
def order_handler():
    args = request.form.get("text").split(" ")
    if(len(args) == 0):
        return WRONG_NUM_ARGS

    async def sub_order(api, request, args):
        if(args[0].lower() == "market"):
            if(len(args) != 5):
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": WRONG_NUM_ARGS}), headers={"Content-type": "application/json"})
            try:
                args[3] = args[3].upper()
                order = api.submit_order(
                    args[3],
                    args[2],
                    args[1],
                    args[0].lower(),
                    args[4])
                price = api.get_barset(args[3], 'minute', 1)[args[3]][0].c
                text = f'Market order of | {args[1]} {args[2]} {args[3]} {args[4]} |, current equity price at {price}.  Order id = {order.id}.'
                response = requests.post(
                    url="https://slack.com/api/chat.postMessage",
                    data={
                        "token": SLACK_TOKEN,
                        "channel": request.form.get("channel_name"),
                        "text": text})
            except Exception as e:
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
        elif(args[0].lower() == "limit"):
            if(len(args) != 6):
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": WRONG_NUM_ARGS}), headers={"Content-type": "application/json"})
            try:
                args[3] = args[3].upper()
                order = api.submit_order(
                    args[3],
                    args[2],
                    args[1],
                    args[0].lower(),
                    args[4],
                    limit_price=args[5])
                text = f'Limit order of | {args[1]} {args[2]} {args[3]} {args[4]} at limit price {args[5]} | submitted.  Order id = {order.id}.'
                response = requests.post(
                    url="https://slack.com/api/chat.postMessage",
                    data={
                        "token": SLACK_TOKEN,
                        "channel": request.form.get("channel_name"),
                        "text": text})
            except Exception as e:
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
        elif(args[0].lower() == "stop"):
            if(len(args) != 6):
                return WRONG_NUM_ARGS
            try:
                args[3] = args[3].upper()
                order = api.submit_order(
                    args[3],
                    args[2],
                    args[1],
                    args[0].lower(),
                    args[4],
                    stop_price=args[5])
                text = f'Stop order of | {args[1]} {args[2]} {args[3]} {args[4]} at stop price {args[5]} | submitted.  Order id = {order.id}.'
                response = requests.post(
                    url="https://slack.com/api/chat.postMessage",
                    data={
                        "token": SLACK_TOKEN,
                        "channel": request.form.get("channel_name"),
                        "text": text})
            except Exception as e:
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
        elif(args[0].lower() == "stop_limit"):
            if(len(args) != 7):
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": WRONG_NUM_ARGS}), headers={"Content-type": "application/json"})
            try:
                args[3] = args[3].upper()
                order = api.submit_order(
                    args[3],
                    args[2],
                    args[1],
                    args[0].lower(),
                    args[4],
                    limit_price=args[5],
                    stop_price=args[6])
                text = f'Stop-Limit order of | {args[1]} {args[2]} {args[3]} {args[4]} at stop price {args[6]} and limit price {args[5]} | submitted.  Order id = {order.id}.'
                response = requests.post(
                    url="https://slack.com/api/chat.postMessage",
                    data={
                        "token": SLACK_TOKEN,
                        "channel": request.form.get("channel_name"),
                        "text": text})
            except Exception as e:
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
        else:
            return BAD_ARGS
    asyncio.run(sub_order(api, request, args))
    return ""

# Lists certain things.  Must contain 1 argument: orders, positions, or
# streams.


@app.route("/list", methods=["POST"])
def list_handler():
    args = request.form.get("text").split(" ")
    if(len(args) == 0):
        return WRONG_NUM_ARGS
    if(args[0] == "positions"):
        try:
            positions = api.list_positions()
            if(len(positions) == 0):
                return "No positions."
            positions = map(
                lambda x: (f'Symbol: {x.symbol}, Qty: {x.qty}, Side: {x.side}, Entry price: {x.avg_entry_price}, Current price: {x.current_price}'),
                positions)
            return "Listing positions...\n" + '\n'.join(positions)
        except Exception as e:
            response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
    elif(args[0] == "orders"):
        try:
            orders = api.list_orders(status="open")
            if(len(orders) == 0):
                return "No orders."
            orders = map(
                lambda x: (f'Symbol: {x.symbol}, Qty: {x.qty}, Side: {x.side}, Type: {x.type}, Time in Force: {x.time_in_force}, Amount Filled: {x.filled_qty}{(f", Stop Price = {x.stop_price}","")[x.stop_price == None]}{(f", Limit Price = {x.limit_price}","")[x.limit_price == None]}, Order id = {x.id}'),
                orders)
            return "Listing orders...\n" + '\n'.join(orders)
        except Exception as e:
            response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
    elif(args[0] == "streams"):
        text = "Listing active streams...\n"
        try:
            for stream in streams:
                if(streams[stream] is not None):
                    text += (stream + "\n")
            if(text == "Listing active streams...\n"):
                return "No active streams."
            return text
        except Exception as e:
            response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
    else:
        return BAD_ARGS

# Clears positions or orders.  Must contain 1 argument: positions or orders


@app.route("/clear", methods=["POST"])
def clear_handler():
    args = request.form.get("text").split(" ")
    if(len(args) == 0):
        return WRONG_NUM_ARGS
    if(args[0] == "positions"):
        async def sub_clear_positions(api, request):
            try:
                positions = api.list_positions()
                positions = map(lambda x: [x.symbol, x.qty, x.side], positions)
                for position in positions:
                    api.submit_order(position[0], abs(
                        int(position[1])), "sell" if position[2] == "long" else "buy", "market", "day")
                text = "Position clearing orders sent."
                response = requests.post(
                    url="https://slack.com/api/chat.postMessage",
                    data={
                        "token": SLACK_TOKEN,
                        "channel": request.form.get("channel_name"),
                        "text": text})
            except Exception as e:
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": f'ERROR: {str(e)}'}), headers={"Content-type": "application/json"})
        asyncio.run(sub_clear_positions(api, request))
        return ""
    elif(args[0] == "orders"):
        async def sub_clear_orders(api, request):
            try:
                orders = api.list_orders()
                orders = map(lambda x: x.id, orders)
                for order in orders:
                    api.cancel_order(order)
                text = "Order cancels sent."
                response = requests.post(
                    url="https://slack.com/api/chat.postMessage",
                    data={
                        "token": SLACK_TOKEN,
                        "channel": request.form.get("channel_name"),
                        "text": text})
            except Exception as e:
                response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                    {"text": f'ERROR: {str(e)}'}), headers={"Content-type": "application/json"})
        asyncio.run(sub_clear_orders(api, request))
        return ""
    else:
        return BAD_ARGS

# Cancels order by id.  Must take one argument: order_id


@app.route("/cancel_order", methods=["POST"])
def cancel_order_handler():
    args = request.form.get("text").split(" ")
    if(len(args) != 1):
        return WRONG_NUM_ARGS
    try:
        api.cancel_order(args[0])
        text = f'Order canceled.  Order id = {args[0]}'
        return text
    except Exception as e:
        return f"ERROR: {str(e)}"

# Cancels most recent order.  Takes no arguments.


@app.route("/cancel_recent_order", methods=["POST"])
def cancel_recent_order_handler():
    args = request.form.get("text").split(" ")
    if(len(args) != 0 and not (len(args) == 1 and args[0].strip() == "")):
        return WRONG_NUM_ARGS
    try:
        orders = api.list_orders(status="open", limit=1)
        if(len(orders) == 0):
            return "No orders to cancel."
        api.cancel_order(orders[0].id)
        text = f'Most recent order cancelled.  Order id = {orders[0].id}'
        return text
    except Exception as e:
        return f"ERROR: {str(e)}"

# Gets basic account info.  Takes no arguments.


@app.route("/account_info", methods=["POST"])
def account_info_handler():
    args = request.form.get("text").split(" ")
    if(len(args) != 0 and not (len(args) == 1 and args[0].strip() == "")):
        return WRONG_NUM_ARGS
    try:
        account = api.get_account()
        text = f'Account info...\nBuying power = {account.buying_power}\nEquity = {account.equity}\nPortfolio value = {account.portfolio_value}\nShorting enabled? = {account.shorting_enabled}'
        return text
    except Exception as e:
        return f"ERROR: {str(e)}"

# Gets Polygon price specified stock symbols.  Must include one or more
# arguments representing stock symbols. Must have live account to access.


@app.route("/get_price_polygon", methods=["POST"])
def get_price_polygon_handler():
    args = request.form.get("text").split(" ")
    if(len(args) == 1 and args[0].strip() == ""):
        return WRONG_NUM_ARGS

    async def sub_get_price_polygon(api, request, args):
        try:
            text = "Listing prices..."
            args = map(lambda x: x.upper(), args)
            for symbol in args:
                quote = api.polygon.last_quote(symbol)
                text += f'\n{symbol}: Bid price = {quote.bidprice}, Ask price = {quote.askprice}'
            response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                {"text": text}), headers={"Content-type": "application/json"})
        except Exception as e:
            response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
    asyncio.run(sub_get_price_polygon(api, request, args))
    return ""

# Gets price specified stock symbols.  Must include one or more arguments
# representing stock symbols.


@app.route("/get_price", methods=["POST"])
def get_price_handler():
    args = request.form.get("text").split(" ")
    if(len(args) == 1 and args[0].strip() == ""):
        return WRONG_NUM_ARGS

    async def sub_get_price(api, request, args):
        try:
            text = "Listing prices..."
            args = map(lambda x: x.upper(), args)
            bars = api.get_barset(args, "minute", 1)
            for bar in bars:
                text += f'\n{bar}: Price = {bars[bar][0].c}, Time = {bars[bar][0].t}'
            response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                {"text": text}), headers={"Content-type": "application/json"})
        except Exception as e:
            response = requests.post(url=request.form.get("response_url"), data=json.dumps(
                {"text": f"ERROR: {str(e)}"}), headers={"Content-type": "application/json"})
    asyncio.run(sub_get_price(api, request, args))
    return ""

if __name__ == "__main__":
    app.run(port=3000)