firstblood.py

"""
FIRSTBLOOD - first GroupMe API bot

Mostly for testing API connectivity & commands
"""
import os
from ApiClient.client import GroupMeApiClient
from TradingViewScreenshotTesting.tvchartbot import TradingViewScraper


GROUP_ID = 12375272     # nerd chat

"""
Client setup

Keyfile can have 2 lines, containing:
line 1 | Groupme Access Token
line 2 | (optional) Bot ID
"""
# PATH_TO_KEYFILE = "../groupme_keys.txt"     # assumes keyfile is in parent dir
#
# with open(PATH_TO_KEYFILE, "r") as f:
#     TOKEN = f.readline().strip()
#     BOTID = f.readline().strip()
#     print(TOKEN, BOTID)

"""
Client setup alternate method:
using environment variables

Meant for use with Heroku CONFIG_VARS:
TOKEN
BOT_ID
"""
TOKEN = os.environ.get('TOKEN')
BOTID = os.environ.get('BOT_ID')
CHARTBOTID = os.environ.get('CHARTBOT_ID')

c = GroupMeApiClient(TOKEN, BOTID)
c_chartbot = GroupMeApiClient(TOKEN, CHARTBOTID)

tvbot = TradingViewScraper()
tvbot.start()


def incoming_message(data):
    """
    Main method invoked when a new message arrives from the group.
    The (flask) app receives a POST with the new message, and it is immediately sent here.
    :param data: dict of JSON data from GroupMe API's POST request
    :return:
    """

    # Chartbot
    if data['name'] != "Chartbot" and data['group_id'] == "36792321":
        # parse message
        msg = data['text'].strip().lower()

        if msg.startswith('c '):
            splittext = data['text'].strip().lower().split()[1:4]
            if len(splittext) is 0:
                return c_chartbot.postfrombot("Error: you must provide an argument. Type 'c help' for a list of commands.")
            elif splittext[0] is "help":
                return c_chartbot.postfrombot("Chart command: 'c <ticker> <exchange> <time>\n"
                                   "Examples:   'c ltc'\n"
                                   "            'c ltcbtc binance'\n"
                                   "            'c xrpusd bitfinex 1d'\n"
                                   "            'c xbtusd bitmex 15'")
            fchartpic = tvbot.get_chart_screenshot_binary(splittext)
            url = c_chartbot.get_pic_upload_url(fchartpic)
            c_chartbot.postfrombot(" ", picture_url=url)
    else:
        # parse message
        msg = data['text'].strip().lower()

        # simple mirror response commands
        if msg == "ride on cowboy":
            respond_rideoncowboy()
        if msg == "jeremy is a":
            jeremyisa()
        # echo command
        if msg.startswith('echo'):
            echo(data['text'])
        # Tradingview Chart screenshot
        if msg.startswith('c '):
            chart(data['text'])


def postfrombot(message):
    c.postfrombot(message)

def postpic(fpicture_binary):
    """
    Uploads & posts a picture, with its filename as the parameter.
    Picture file must be in same directory as firstblood.py.
    :param fpicture_binary: file-like object of the desired image to be posted
    :return:
    """
    url = c.get_pic_upload_url(fpicture_binary)
    c.postfrombot(" ", picture_url=url)


def respond_rideoncowboy():
    resp = "Hell Yeah, brother"
    resp2 = "*takes swig of Busch Lite*"
    print("respond_rideoncowboy should be working...")
    c.postfrombot(resp)
    c.postfrombot(resp2)
    # c.makeCall('bots', 'Post', text=resp)
    # c.makeCall('bots', 'Post', text=resp2)
    # c.makeCall('messages', 'Create', groupId=GROUP_ID, text=resp)
    # c.makeCall('messages', 'Create', groupId=GROUP_ID, text=resp2)


def jeremyisa():
    c.postfrombot('PEEPEEPOOPOOOOOO')
    # c.makeCall('bots', 'Post', text='PEEPEEPOOPOOOOOO')


def echo(text):
    """
    Echoes the message after the command by <multiplier> number of times.
    Invoked with "echo 2x <message>" where 2x is the multiplier; can be up to 20.
    :param text: raw text from latest group message
    :return:
    """
    indexofx = text.find('x')
    multiplier = int(text[5:indexofx])
    if multiplier > 10:
        multiplier = 10     # cap the number of repeats at 20x
    message_to_repeat = text[indexofx+2:]
    for i in range(multiplier):
        c.postfrombot(message_to_repeat)
        # c.makeCall('bots', 'Post', text=text)


def chart(text):
    splittext = text.strip().lower().split()[1:4]
    if len(splittext) is 0:
        return postfrombot("Error: you must provide an argument. Type 'c help' for a list of commands.")
    elif splittext[0] is "help":
        return postfrombot("Chart command: 'c <ticker> <exchange> <time>\n"
                           "Examples:   'c ltc'\n"
                           "            'c ltcbtc binance'\n"
                           "            'c xrpusd bitfinex 1d'\n"
                           "            'c xbtusd bitmex 15'")

    fchartpic = tvbot.get_chart_screenshot_binary(splittext)
    postpic(fchartpic)