telegram_bot.py

import json
import os
import sys
import requests
import telegram
from conectors.luis_connector import LUISConnector
from api_calls.redmine import Redmine
from api_calls.cartola import Cartola

OK_RESPONSE = {
    'statusCode': 200,
    'headers': {'Content-Type': 'application/json'},
    'body': json.dumps('ok')
}
ERROR_RESPONSE = {
    'statusCode': 400,
    'body': json.dumps('Ocorreu um erro!')
}

ABRIR_CHAMADO = 'Abrir Chamado'
CONSULTAR_CHAMADO = 'Consultar Chamado'
SAUDACAO = 'Saudacao'
LIGA = 'Consultar Liga'


def configure_telegram():
    TELEGRAM_TOKEN = os.environ.get('TELEGRAM_TOKEN')
    return telegram.Bot(TELEGRAM_TOKEN)


def publish_telegram_msg(event, context):

    bot = configure_telegram()
    if event.get('httpMethod') == 'POST' and event.get('body'):
        update = telegram.Update.de_json(json.loads(event.get('body')), bot)
        # print(update)
        chat_id = update.message.chat.id
        text = update.message.text
        name = update.message.from_user.first_name

        response = create_msg_response(text, name)
        if response != '':
            bot.sendMessage(chat_id=chat_id, text=response)
            return OK_RESPONSE

    return ERROR_RESPONSE


def create_msg_response(input, first_name):
    response = ''
    intencao, entity, query = tratar_intencao(input)
    if intencao == SAUDACAO:
        response = tratar_saudacao(first_name)
    elif intencao == LIGA:
        response = tratar_cartola_liga(input)
    elif intencao == CONSULTAR_CHAMADO:
        response = tratar_consultar_chamado(entity, first_name)
    elif intencao == ABRIR_CHAMADO:
        response = tratar_abrir_chamado(input)
    else:
        response = tratar_nao_reconhecimento(first_name)
    return response


def tratar_intencao(input):
    luis_conn = LUISConnector(model_file=None)
    intent = luis_conn._process(input)
    # print(intent)
    intencao = intent.intents[0].intent
    entity = ''
    if len(intent.entities) > 0:
        #entity = intent.entities[-1].resolution['value']
        entity = intent.entities[-1].entity
    return intencao, entity, intent.query


def tratar_cartola_liga(consulta):
    cartola = Cartola()
    frase = consulta.split('liga')[1].replace('?', '')
    print(frase)
    ligas = cartola.ligas(frase)
    resp = 'Foram encontradas {} Ligas! \n'.format(len(ligas))
    for liga in ligas:
        resp += '* {} \n\t - descrição: {} \n'.format(
            liga.nome, liga.descricao)
    return resp


def tratar_retorno(input):
    resp = 'O chamado nº: "' + str(input.get('issue').get('id')) + '"' + \
        ' está com status: "' + str(input.get('issue').get('status').get('name')) \
        + '" e possui a descrição: "' + \
        str(input.get('issue').get('subject')) + '".\n'
    return resp


def tratar_saudacao(first_name):
    response = "Olá {}, sou um bot de demonstração!!\n".format(first_name)
    response += "Estou aqui para criar e consultar chamados.\n"
    response += "\n1- Para consultar chamados, utilize a sintaxe:"
    response += "\n       'você poderia consultar o chamado 1'"
    response += "\n\n2- Para cadastrar chamados, utilize a sintaxe:"
    response += "\n       'cadastrar um chamado para: perda de celular'"
    return response


def tratar_consultar_chamado(entity, first_name):
    red = Redmine()
    response = 'Não foi possível consultar seu chamado.'
    print(entity)
    if not entity == '':
        resp = red.execute_get(entity)
        response = "Olá {}, abaixo os dados do seu chamado:\n\n".format(
            first_name)
        response += tratar_retorno(resp)
    return response


def tratar_abrir_chamado(input):
    red = Redmine()
    response = input.split(':')
    if len(response) > 1:
        resp = red.execute_post(response[1])
        print(resp)
        response = "Chamado {} aberto com sucesso\n".format(
            resp.get('issue').get('id'))
    else:
        response = "Não foi possível criar o seu chamado."
    return response


def tratar_nao_reconhecimento(first_name):
    response = "Olá {}, não entendi o que você disse. Poderia repetir?\n" \
        .format(first_name)
    return response


'''
if __name__ == '__main__':
    # a = tratar_intencao('Poderia consultar a liga ete old')
    # b = create_msg_response('Poderia consultar a liga ete old?', 'Evandro')
    # print(a)
    publish_telegram_msg({'resource': '/teleg-bot-demo', 
        'path': '/teleg-bot-demo', 'httpMethod': 'POST', 
        'headers': {'Accept-Encoding': 'gzip, deflate', 
        'CloudFront-Forwarded-Proto': 'https', 'CloudFront-Is-Desktop-Viewer': 
        'true', 'CloudFront-Is-Mobile-Viewer': 'false', 
        'CloudFront-Is-SmartTV-Viewer': 'false', 
        'CloudFront-Is-Tablet-Viewer': 'false', 
        'CloudFront-Viewer-Country': 'GB', 'Content-Type': 
        'application/json', 
        'Host': '29hthkrsx0.execute-api.us-east-2.amazonaws.com', 
        'Via': '1.1 5fe8343a80de49928fae39084e131a25.cloudfront.net (CloudFront)', 
        'X-Amz-Cf-Id': 'IN69kYU2zS1Xi_zGUdgEdmvLH1Sg8ah9hwj9bPA-yRDEU8bkZmUocw==', 
        'X-Amzn-Trace-Id': 'Root=1-5cb6fc3a-2248bb2cd6151a58c47a5f36', 
        'X-Forwarded-For': '149.154.167.229, 54.240.156.99', 
        'X-Forwarded-Port': '443', 'X-Forwarded-Proto': 'https'}, 
        'multiValueHeaders': {'Accept-Encoding': ['gzip, deflate'], 
        'CloudFront-Forwarded-Proto': ['https'], 
        'CloudFront-Is-Desktop-Viewer': ['true'], 'CloudFront-Is-Mobile-Viewer': 
        ['false'], 'CloudFront-Is-SmartTV-Viewer': ['false'], 
        'CloudFront-Is-Tablet-Viewer': ['false'], 'CloudFront-Viewer-Country': ['GB'], 
        'Content-Type': ['application/json'], 
        'Host': ['29hthkrsx0.execute-api.us-east-2.amazonaws.com'], 
        'Via': ['1.1 5fe8343a80de49928fae39084e131a25.cloudfront.net (CloudFront)'], 
        'X-Amz-Cf-Id': ['IN69kYU2zS1Xi_zGUdgEdmvLH1Sg8ah9hwj9bPA-yRDEU8bkZmUocw=='], 
        'X-Amzn-Trace-Id': ['Root=1-5cb6fc3a-2248bb2cd6151a58c47a5f36'], 
        'X-Forwarded-For': ['149.154.167.229, 54.240.156.99'], 
        'X-Forwarded-Port': ['443'], 'X-Forwarded-Proto': ['https']}, 
        'queryStringParameters': None, 'multiValueQueryStringParameters': 
        None, 'pathParameters': None, 'stageVariables': None, 
        'requestContext': {'resourceId': 'ty7i49', 'resourcePath': 
        '/teleg-bot-demo', 'httpMethod': 'POST', 
        'extendedRequestId': 'YRxZMHJACYcF3LQ=', 
        'requestTime': '17/Apr/2019:10:13:14 +0000', 
        'path': '/dev/teleg-bot-demo', 'accountId': '175187370354', 
        'protocol': 'HTTP/1.1', 'stage': 'dev', 'domainPrefix': '29hthkrsx0', 
        'requestTimeEpoch': 1555495994851, 'requestId': 
        '69bb33af-60f9-11e9-a015-1d41eb305e2f', 'identity': 
        {'cognitoIdentityPoolId': None, 'accountId': None, 'cognitoIdentityId': 
        None, 'caller': None, 'sourceIp': '149.154.167.229', 
        'accessKey': None, 'cognitoAuthenticationType': None, 
        'cognitoAuthenticationProvider': None, 'userArn': None, 
        'userAgent': None, 'user': None}, 'domainName': 
        '29hthkrsx0.execute-api.us-east-2.amazonaws.com', 'apiId': '29hthkrsx0'}, 
        'body': '{"update_id":241554614,\n"message":{"message_id":131,"from":{"id":68330001,"is_bot":false,"first_name":"Evandro","last_name":"Franco","username":"evandrofranco","language_code":"pt-br"},"chat":{"id":68330001,"first_name":"Evandro","last_name":"Franco","username":"evandrofranco","type":"private"},"date":1555495019, \
        "text":"criar chamado para: Meu e-mail não está funcionando"}}', 'isBase64Encoded': False}, ''
        )
'''