You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
130 lines
4.8 KiB
130 lines
4.8 KiB
# -*- 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
|
|
}
|
|
|
|
# отвечаем пользователю
|
|
def _sent_reply(self, text):
|
|
self.sender.sendMessage(text)
|
|
|
|
# задаём вопрос пользователю, и ожидаем ответ в fn
|
|
def _question(self, text, fn):
|
|
assert callable(fn)
|
|
self._dialog_fn = fn
|
|
if text is not None:
|
|
self._sent_reply(text)
|
|
|
|
# сохраняем сообщение в базе
|
|
def _message_log(self, msg):
|
|
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):
|
|
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):
|
|
content_type, chat_type, chat_id = glance(msg)
|
|
if content_type != 'text':
|
|
return
|
|
self._chat_id = chat_id
|
|
text = msg['text'].lower()
|
|
|
|
# выполняем комманды если они есть
|
|
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):
|
|
try:
|
|
profile = UserProfile.objects.get(username=username)
|
|
self._current_user = profile
|
|
try:
|
|
TelegramBot.objects.get(user=profile)
|
|
except TelegramBot.DoesNotExist:
|
|
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
|
|
|
|
# пингуем адрес
|
|
def ping(self, ip=None):
|
|
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):
|
|
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)
|
|
|