djing/chatbot/telebot.py

# -*- coding: utf-8 -*-
from telepot import helper, glance, Bot
import os
import socket
from models import TelegramBot
from chatbot.models import MessageHistory
from accounts_app.models import UserProfile

token = '285129725:AAF9Si5_b1n1_cN3vJtwXt0gkgsqKBptut4'


class DjingTelebot(helper.ChatHandler):
    _current_user = None
    _dialog_fn = None
    _chat_id = 0

    def __init__(self, seed_tuple, **kwargs):
        super(DjingTelebot, self).__init__(seed_tuple, **kwargs)
        self.cmds = {
            'ping': self.ping,
            'iam': self.say_me
        }
        print 'Constructor'

    # отвечаем пользователю
    def _sent_reply(self, text):
        print('CALL: _sent_reply')
        self.sender.sendMessage(text)

    # задаём вопрос пользователю, и ожидаем ответ в fn
    def _question(self, text, fn):
        print('CALL: _question')
        assert callable(fn)
        self._dialog_fn = fn
        if text is not None:
            self._sent_reply(text)

    # сохраняем сообщение в базе
    def _message_log(self, msg):
        print('CALL: _message_log')
        if self._current_user is None:
            self._question(None, self.question_name)
            return False
        MessageHistory.objects.create(
            user=self._current_user,
            message=msg
        )
        return True

    # Начинаем диалог
    # @seed - chat_id
    def open(self, initial_msg, seed):
        print('CALL: open')
        content_type, chat_type, chat_id = glance(initial_msg)
        if content_type != 'text':
            return True
        self._chat_id = chat_id
        try:
            tbot = TelegramBot.objects.get(chat_id=seed)
            self._current_user = tbot.user
            self._message_log(initial_msg['text'])
        except TelegramBot.DoesNotExist:
            self._question(u'Давай знакомиться, как тебя зовут? Напиши свой логин из биллинга.',
                           self.question_name)
        return True  # prevent on_message() from being called on the initial message

    # получаем сообщение
    def on_chat_message(self, msg):
        print('CALL: on_chat_message')
        content_type, chat_type, chat_id = glance(msg)
        if content_type != 'text':
            return
        self._chat_id = chat_id
        text = msg['text']

        # выполняем комманды если они есть
        if text in self.cmds.keys():
            self.cmds[text]()
        elif self._dialog_fn is not None:
            self._dialog_fn(text)
            self._dialog_fn = None
        else:
            self._sent_reply(u'Я пока не знаю ответа на это')

        if not self._message_log(text):
            return

    # спрашиваем имя пользователя
    def question_name(self, username):
        print('CALL: question_name')
        try:
            profile = UserProfile.objects.get(username=username)
            self._current_user = profile
            try:
                TelegramBot.objects.get(user=profile)
            except TelegramBot.DoesNotExist:
                print '_chat_id', self._chat_id
                assert self._chat_id != 0
                TelegramBot.objects.create(
                    user=profile,
                    chat_id=self._chat_id
                )
        except UserProfile.DoesNotExist:
            self._question(u'Ты не найден в базе, проверь что правильно указал именно свой ЛОГИН. Попробуй ещё',
                           self.question_name)
            return
        self._sent_reply(u'Да, приятно познакомиться %s, я буду оповещать тебя о событиях в биллинге. Удачной работы ;)'
                         % profile.get_full_name())

    # заканчивается время диалога
    # ex - время ожидания (timeout=ex в pave_event_space)
    def on_close(self, ex):
        self._current_user = None
        self._dialog_fn = None
        self._chat_id = 0
        print 'Destructor'

    # пингуем адрес
    def ping(self, ip=None):
        print('CALL: ping')
        if ip is None:
            self._question(u'Давай пинганём, напиши ip. Нужно будет подождать 10 сек', self.ping)
            return
        try:
            socket.inet_aton(ip)
            ret = os.popen('`which ping` -c 10 ' + ip).read()
            self._sent_reply(ret)
        except socket.error:
            self._question(u'Это не похоже на ip адрес, попробуй ещё', self.ping)

    def say_me(self):
        print('CALL: say_me')
        self._sent_reply(u'Ты ведь %s ?' % self._current_user.get_full_name())


# Просто отправляем текст оповещения указанному админу
def send_notify(msg_text, account):
    tb = TelegramBot.objects.get(user=account)
    tbot = Bot(token)
    tbot.sendMessage(tb.chat_id, msg_text)