первый коммит

chatbot/admin.py

@@ -0,0 +1,5 @@
+from django.contrib import admin
+import models
+
+admin.site.register(models.MessageHistory)
+admin.site.register(models.TelegramBot)

chatbot/apps.py

@@ -0,0 +1,5 @@
+from django.apps import AppConfig
+
+
+class ChatbotConfig(AppConfig):
+    name = 'chatbot'

chatbot/migrations/0001_initial.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.9 on 2017-01-15 20:55
+from __future__ import unicode_literals
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='MessageHistory',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('message', models.CharField(max_length=255)),
+                ('date_sent', models.DateTimeField(auto_now_add=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='TelegramBot',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('chat_id', models.PositiveIntegerField(default=0)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+    ]

chatbot/models.py

@@ -0,0 +1,19 @@
+from django.db import models
+from djing.settings import AUTH_USER_MODEL
+
+
+class TelegramBot(models.Model):
+    user = models.ForeignKey(AUTH_USER_MODEL)
+    chat_id = models.PositiveIntegerField(default=0)
+
+    def __unicode__(self):
+        return self.user.get_full_name() + ' - ' + str(self.chat_id)
+
+
+class MessageHistory(models.Model):
+    user = models.ForeignKey(AUTH_USER_MODEL)
+    message = models.CharField(max_length=255)
+    date_sent = models.DateTimeField(auto_now_add=True)
+
+    def __unicode__(self):
+        return self.message

chatbot/telebot.py

@@ -0,0 +1,141 @@
+# -*- 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)
+

chatbot/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

chatbot/views.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.