rose.py

from flask import Flask, request
import os
import requests
import json

app = Flask(__name__)

BEGINNING = 'BEGINNING'
ASKED = 'ASKED'
CONFUSED = 'CONFUSED'
SATISFIED = 'SATISFIED'

INITIAL_BODY_1 = u'''<p>Well hello there, it sounds like you're thirsty as hell and in need of a refreshing drink. Might I interest you in a ros\xe9?</p>
<p>I've quite a selection, but how about you tell me a little about yourself so I can refer you to the proper drink?</p>
'''

INITIAL_BODY_2 = '''Are you:
<ol>
  <li>On a yacht or feel you deserve to be?</li>
  <li>Going out with the bros and got no time for masculo-normative bullshit?</li>
  <li>A basic white girl?</li>
</ol>
'''

BOAT_RESPONSE = 'I\'m on a boat'
BRO_RESPONSE = 'BRO'
BASIC_RESPONSE = 'Basic AF'
NONE_RESPONSE = '... None of these'
REALLY_NONE_RESPONSE = 'Yes, really none of these'

def button(response):
    return {'value': response, 'label': response}

INITIAL_RESPONSE_BUTTONS = [button(BOAT_RESPONSE), button(BRO_RESPONSE), button(BASIC_RESPONSE), button(NONE_RESPONSE)]
CONFUSED_RESPONSE_BUTTONS = [button(BOAT_RESPONSE), button(BRO_RESPONSE), button(BASIC_RESPONSE), button(REALLY_NONE_RESPONSE)]

CONVERSATIONS = {}

API_HOST = os.environ.get('API_HOST')
MESSAGE_ENDPOINT_FORMAT = API_HOST + "/open/conversations/%s/messages"

AUTH_TOKEN = os.environ.get('AUTH_TOKEN')
AUTH = 'bearer %s' % AUTH_TOKEN

def post_message(conversationId, message):
    r = requests.post(MESSAGE_ENDPOINT_FORMAT % conversationId, json=message, headers={'Authorization': AUTH})
    r.raise_for_status()

def make_message(orgId, body, type, buttons=[]):
    return {'orgId': orgId, 'body': body, 'type': type, 'buttons': buttons}

def respond_to_probable_button_response(conversationId, orgId, body):
    if body == BOAT_RESPONSE:
        post_message(conversationId, make_message(orgId, '<a href="https://drizly.com/meiomi-rose/p61784">This</a> should satisfy you, bourgeois animal', 'chat'))
        return SATISFIED
    elif body == BRO_RESPONSE:
        post_message(conversationId, make_message(orgId, '<a href="https://drizly.com/charles-and-charles-rose/p2516">BROOOOOOO</a>', 'chat'))
        return SATISFIED
    elif body == BASIC_RESPONSE:
        post_message(conversationId, make_message(orgId, '<a href="https://drizly.com/white-girl-rose/p20394">Here</a> you go girl', 'chat'))
        return SATISFIED
    elif body == NONE_RESPONSE:
        post_message(conversationId, make_message(orgId, 'Ughh.. what? Wait, are you sure?', 'chat', CONFUSED_RESPONSE_BUTTONS))
        return CONFUSED
    elif body == REALLY_NONE_RESPONSE:
        post_message(conversationId, make_message(orgId, '??? Ok...', 'chat'))
        return SATISFIED
    else:
        post_message(conversationId, make_message(orgId, 'Sssshhhh sshhhhh, don\'t fight it, just answer', 'chat', INITIAL_RESPONSE_BUTTONS))
        return ASKED

def handle_state_edge(state, orgId, message):
    conversationId = message.get('conversationId')
    body = message.get('body')

    if state == BEGINNING:
        post_message(conversationId, make_message(orgId, INITIAL_BODY_1, 'chat'))
        post_message(conversationId, make_message(orgId, INITIAL_BODY_2, 'chat', INITIAL_RESPONSE_BUTTONS))
        return ASKED
    if state == ASKED:
        return respond_to_probable_button_response(conversationId, orgId, body)
    elif state == CONFUSED:
        return respond_to_probable_button_response(conversationId, orgId, body)
    elif state == SATISFIED:
        return state

@app.route('/', methods=['POST'])
def accept_message():
    event = request.get_json()    
    if event and event.get('type') == 'new_message':
        orgId = event.get('orgId')
        message = event.get('data')
        conversationId = message.get('conversationId')

        if 'contact' == message.get('author').get('type'):
            if (CONVERSATIONS.has_key(conversationId)):
                state = CONVERSATIONS[conversationId]
                CONVERSATIONS[conversationId] = handle_state_edge(state, orgId, message)
            elif 'rose' in message.get('body').lower():
                CONVERSATIONS[conversationId] = handle_state_edge(BEGINNING, orgId, message)
    return json.dumps(CONVERSATIONS)

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