106 lines
2.9 KiB
Python
106 lines
2.9 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# ! /usr/bin/env python
|
||
|
|
|
||
|
|
import logging
|
||
|
|
# import requests
|
||
|
|
import sys
|
||
|
|
import random
|
||
|
|
|
||
|
|
from telegram.ext import (Updater, CommandHandler, MessageHandler,
|
||
|
|
Filters)
|
||
|
|
|
||
|
|
from config import config
|
||
|
|
|
||
|
|
from chatterbot import ChatBot
|
||
|
|
|
||
|
|
|
||
|
|
# Enable logging
|
||
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||
|
|
level=logging.INFO)
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
chatbot = None
|
||
|
|
|
||
|
|
|
||
|
|
# Define a few command handlers. These usually take the two arguments bot and
|
||
|
|
# update. Error handlers also receive the raised TelegramError object in error.
|
||
|
|
def start(bot, update):
|
||
|
|
bot.sendMessage(update.message.chat_id, text='¡¿te rieh?! Que estoy muy loco...')
|
||
|
|
|
||
|
|
|
||
|
|
def help(bot, update):
|
||
|
|
bot.sendMessage(update.message.chat_id, text="""Te Rieh Bot v1.0:
|
||
|
|
No tiene sentido lo que digo... iré endrogao...
|
||
|
|
""")
|
||
|
|
|
||
|
|
|
||
|
|
def error(bot, update, error):
|
||
|
|
logger.warn('Update "%s" caused error "%s"' % (update, error))
|
||
|
|
|
||
|
|
|
||
|
|
def randomchat(bot, update):
|
||
|
|
global chatbot
|
||
|
|
msg = update.message.text.lower()
|
||
|
|
user_name = update.message.from_user.username.lower()
|
||
|
|
reply = chatbot.get_response(msg)
|
||
|
|
|
||
|
|
if random.randint(0, 100) < 5:
|
||
|
|
reply = user_name + ": " + reply
|
||
|
|
bot.sendMessage(update.message.chat_id, text=reply)
|
||
|
|
|
||
|
|
|
||
|
|
def initial_training(bot, update):
|
||
|
|
chatbot.train("chatterbot.corpus.spanish")
|
||
|
|
|
||
|
|
bot.sendMessage(update.message.chat_id, text="Entrenamiento inicial completado.")
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
global chatbot
|
||
|
|
token = config.get('TOKEN')
|
||
|
|
|
||
|
|
if token is None:
|
||
|
|
print("Please, configure your token first")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
updater = Updater(token)
|
||
|
|
dispatcher = updater.dispatcher
|
||
|
|
|
||
|
|
# on different commands - answer in Telegram
|
||
|
|
dispatcher.add_handler(CommandHandler("start", start))
|
||
|
|
dispatcher.add_handler(CommandHandler("help", help))
|
||
|
|
dispatcher.add_handler(CommandHandler("init", initial_training))
|
||
|
|
|
||
|
|
# on noncommand i.e message - echo the message on Telegram
|
||
|
|
dispatcher.add_handler(MessageHandler([Filters.text], randomchat))
|
||
|
|
|
||
|
|
# log all errors
|
||
|
|
dispatcher.add_error_handler(error)
|
||
|
|
|
||
|
|
# Start the chat bot system
|
||
|
|
chatbot = ChatBot('Te Rieh',
|
||
|
|
storage_adapter='chatterbot.storage.MongoDatabaseAdapter',
|
||
|
|
logic_adapters=[
|
||
|
|
'chatterbot.logic.BestMatch'
|
||
|
|
],
|
||
|
|
filters=[
|
||
|
|
'chatterbot.filters.RepetitiveResponseFilter'
|
||
|
|
],
|
||
|
|
database='terieh',
|
||
|
|
database_uri=config.get('database'))
|
||
|
|
|
||
|
|
# Start the Bot
|
||
|
|
updater.start_polling()
|
||
|
|
|
||
|
|
# Run the bot until the you presses Ctrl-C or the process receives SIGINT,
|
||
|
|
# SIGTERM or SIGABRT. This should be used most of the time, since
|
||
|
|
# start_polling() is non-blocking and will stop the bot gracefully.
|
||
|
|
updater.idle()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
print("Starting TeRiehBot")
|
||
|
|
main()
|