def __init__(self):
self.__disable = False
self.__is_intelligent = False
self._vk_session = vk_api.VkApi(
BOT_SIGN_IN['login'],
BOT_SIGN_IN['password'],
captcha_handler=self.__captcha_handler
)
try:
self._vk_session.authorization()
except vk_api.AuthorizationError as error_msg:
print(error_msg)
return
self._logger = Logger()
self._db = DBConnector()
self._news = NewsParser()
self._math = BotMath()
self._vk = self._vk_session.get_api()
self._ignore_msgs = self._logger.read_history()
self._bot_name = self._get_user_name(None)
self.__ans_list = []
self.__msg_list = []
self.__intell__ans_list = []
self.__intell__msg_list = []
self.__update_messages()
def __init__(self,
max_words,
maxlen,
embeded_size,
embedding_matrix=None,
new=True,
path=None):
self.max_words = max_words
self.maxlen = maxlen
self.parser = NewsParser()
self.cleaner = Cleaner()
self.tokenizer = Tokenizer(self.max_words)
self.embeded_size = embeded_size
self.embedding_matrix = embedding_matrix
if new:
self._model = self.create_model()
else:
self._model = load_model(path)
class Network:
def __init__(self,
max_words,
maxlen,
embeded_size,
embedding_matrix=None,
new=True,
path=None):
self.max_words = max_words
self.maxlen = maxlen
self.parser = NewsParser()
self.cleaner = Cleaner()
self.tokenizer = Tokenizer(self.max_words)
self.embeded_size = embeded_size
self.embedding_matrix = embedding_matrix
if new:
self._model = self.create_model()
else:
self._model = load_model(path)
def connect_to_db(self, path):
self.conn = sqlite3.connect(path)
self.c = self.conn.cursor()
def create_model(self):
inputs = Input(shape=(self.maxlen, ))
x = Embedding(self.max_words,
self.embeded_size,
weights=[self.embedding_matrix],
trainable=False)(inputs)
x = Bidirectional(
LSTM(32, return_sequences=True, dropout=0.1,
recurrent_dropout=0.1))(x)
x = GlobalMaxPool1D()(x)
x = BatchNormalization()(x)
x = Dense(20, activation='relu')(x)
x = Dropout(0.1)(x)
outputs = Dense(1, activation='sigmoid')(x)
model = Model(inputs=inputs, outputs=outputs)
model.compile(loss='binary_crossentropy',
optimizer='adam',
metrics=['acc'])
return model
def fit_data(self,
X,
y,
epochs,
batch_size,
validation_split,
save=False,
path=None):
self._model.fit(X,
y,
epochs=epochs,
batch_size=batch_size,
validation_split=validation_split)
if save:
if path == None:
path = input('Path for weights:\n')
self._model.save(path)
def predict_news(self, topic, page_size, database=True):
self.parser.get_data(topic=topic, page_size=page_size)
for page in range(page_size):
data = self.cleaner.clean_data(self.parser.titles[page])
pad = self.cleaner.to_pad(data, self.tokenizer, self.maxlen)
ans = self._model.predict(pad)
if ans <= 0.5:
label = 0
else:
label = 1
if database:
self.c.execute(
'INSERT INTO news VALUES(?, ?, ?, ?)',
(self.parser.titles[page], self.parser.desc[page],
self.parser.content[page], label))
self.conn.commit()
return label
def test(htmls):
for key, html in htmls.items():
NewsParser().parse(html)
class MainBot(object):
def __init__(self):
self.__disable = False
self.__is_intelligent = False
self._vk_session = vk_api.VkApi(
BOT_SIGN_IN['login'],
BOT_SIGN_IN['password'],
captcha_handler=self.__captcha_handler
)
try:
self._vk_session.authorization()
except vk_api.AuthorizationError as error_msg:
print(error_msg)
return
self._logger = Logger()
self._db = DBConnector()
self._news = NewsParser()
self._math = BotMath()
self._vk = self._vk_session.get_api()
self._ignore_msgs = self._logger.read_history()
self._bot_name = self._get_user_name(None)
self.__ans_list = []
self.__msg_list = []
self.__intell__ans_list = []
self.__intell__msg_list = []
self.__update_messages()
def __update_messages(self):
u"""
Метод используемый для обновления списков с айдишниками сообщений.
Используется после добавления/удаления нового сообщения.
"""
self.__ans_list = self._db.select_ids(is_question_answer=True)
self.__msg_list = self._db.select_ids()
self.__intell__ans_list = self._db.select_ids(is_question_answer=True, is_intelligent=True)
self.__intell__msg_list = self._db.select_ids(is_intelligent=True)
@staticmethod
def __captcha_handler(captcha):
u"""
При возникновении капчи вызывается эта функция и ей передается объект
капчи. Через метод get_url можно получить ссылку на изображение.
Через метод try_again можно попытаться отправить запрос с кодом капчи
"""
key = input("Enter Captcha {0}: ".format(captcha.get_url())).strip()
# Пробуем снова отправить запрос с капчей
return captcha.try_again(key)
def _add_msg_to_ignore(self, msg):
u"""
Добавляем в логгер сообщения, адресованные боту.
Эти сообщения будут проигнорированы.
"""
if 'chat_id' in msg:
self._ignore_msgs.append('%s-%s-%s' % (msg['chat_id'], msg['id'], msg['user_id']))
else:
self._ignore_msgs.append('NONE-%s-%s' % (msg['id'], msg['user_id']))
self._logger.write_history(self._ignore_msgs)
def _is_msg_in_ignore(self, msg):
u"""
Проверяем есть ли сообщение в логгере.
Если есть, то игнорим его.
"""
# result = False
if 'chat_id' in msg:
result = '%s-%s-%s' % (msg['chat_id'], msg['id'], msg['user_id']) in self._ignore_msgs
else:
result = 'NONE-%s-%s' % (msg['id'], msg['user_id']) in self._ignore_msgs
return result
def _send(self, msg, text):
u"""
Метод отправки собщения с текстом text.
"""
if 'chat_id' in msg:
self._vk.messages.send(
chat_id=msg['chat_id'],
message=text
)
else:
self._vk.messages.send(
user_id=msg['user_id'],
message=text
)
def _get_random_message(self, msg_list):
u"""
Получаем рандомное сообщение из списка айдишников сообщений.
"""
return self._db.get_message(random.choice(msg_list))
def _send_message_from_db(self, msg):
u"""
Метод отправляющий сообщение в чат с БД.
"""
if '?' in msg['body']:
text = self._get_random_message(self.__ans_list) if not self.__is_intelligent \
else self._get_random_message(self.__intell__ans_list)
else:
text = self._get_random_message(self.__msg_list) if not self.__is_intelligent \
else self._get_random_message(self.__intell__msg_list)
self._send(msg, u'%s, %s' % (self._get_user_name(msg['user_id']), text))
@staticmethod
def _prepare_msg(msg):
u"""
Метод, используемый для подготовки сообщений для записи их в БД.
Добавляет к спец. символам сообщения дополнительный слеш, с целью не потерять труктуру сообщения.
"""
return msg.replace('\'', '\\\'')
def _remember_new_data(self, msg):
u"""
Запись нового сообщения в БД, обновляем списки айдишников сообщений.
"""
self._db.add_new_row(
msg['user_id'],
self._prepare_msg(msg['body'][len(self._bot_name) + 9:]),
is_intelligent=self.__is_intelligent
)
user_name = self._get_user_name(msg['user_id'])
text = self._prepare_msg(msg['body'][len(self._bot_name) + 9:])
self._send(
msg,
LITERALS['remember_data'].replace('{user_name}', user_name).replace('{message}', text)
)
print(u'Remember new data: from %s, msg: %s' % (msg['user_id'], text))
self.__update_messages()
def _forget_data(self, msg):
u"""
Помечаем сообщение как удаленое, обновляем списки айдишников сообщений.
"""
self._db.del_row(self._prepare_msg(msg['body'][len(self._bot_name) + 8:]))
user_name = self._get_user_name(msg['user_id'])
text = self._prepare_msg(msg['body'][len(self._bot_name) + 9:])
self._send(
msg,
LITERALS['forget_data'].replace('{user_name}', user_name).replace('{message}', text)
)
print(u'Remove data: from %s, msg: %s' % (msg['user_id'], text))
# self.__ans_list = self._db.select_ids()
self.__update_messages()
def _get_help(self):
u"""
Вывод справки по боту.
"""
return u"""
Моя текущая версия: version
Список моих команд на сегодня:
* Выучить что-то новое: name запомни <предложение/фраза>
* Выдать вероятность события: name инфа <фраза/название/событие>
* Выбирает случайного участника беседы: name кто <предложение/фраза>
* Сменить режим общения бота: name смени режим
* Получить текущий режим бота: name режим
* Получить список новостей: name новости
* Посчитать пример: name м
* Справка по математике: name м помощь
* Получить ТВ-программу на ближайшее время: name телепрограмма
* Просто попиздеть: name <предложение/фраза>
* [Для админов] Забыть что-то старое: name забудь <предложение/фраза>
* [Для админов] Выключить бота: name завали ебало
* [Для админов] Включить бота : name камбекнись
Список разработчиков:
* Илья - http://vk.com/id96996256
Список админов:
* Илья - http://vk.com/id96996256
* Дмитрий - http://vk.com/id77698338
""".replace('name', self._bot_name).replace('version', VERSION)
def _get_bot_id(self):
u"""
Получаем айди вк бота.
"""
return int(self._vk_session.token['user_id'])
def _get_user_name(self, user_id):
u"""
Получаем имя пользователя по его айди.
"""
return u'%s' % self._vk.users.get(user_ids=[user_id])[0]['first_name']
def _get_user_lastname(self, user_id):
u"""
Получаем фамилию пользователя по его айди.
"""
return u'%s' % self._vk.users.get(user_ids=[user_id])[0]['last_name']
def _get_chat_users_ids(self, chat_id):
u"""
Получаем список пользователей чата по айдишнику чата.
"""
users = self._vk.messages.getChatUsers(chat_id=chat_id)
users.remove(self._get_bot_id())
idx = random.randint(0, len(users) - 1)
return u'%s %s' % (self._get_user_name(users[idx]), self._get_user_lastname(users[idx]))
def __command_bot_off(self, msg):
u"""
Команда приостановления дейтельности бота.
"""
if msg['body'][:len(self._bot_name) + 13] == LITERALS['commands']['bot_off']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
user_name = self._get_user_name(msg['user_id'])
self._send(msg, LITERALS['commands']['bot_off']['answer'].replace('{user_name}', user_name))
self.__disable = True
return True
return False
def __command_bot_on(self, msg):
u"""
Команда возобновления дейтельности бота.
"""
if msg['body'][:len(self._bot_name) + 13] == LITERALS['commands']['bot_on']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
user_name = self._get_user_name(msg['user_id'])
self._send(msg, LITERALS['commands']['bot_on']['answer'].replace('{user_name}', user_name))
self.__disable = False
return True
return False
def __command_bot_help(self, msg):
u"""
Команда получения справки.
"""
if msg['body'][:len(self._bot_name) + 8] == LITERALS['commands']['help']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
self._send(msg, '%s, %s' % (self._get_user_name(msg['user_id']), self._get_help()))
return True
return False
def __command_bot_forget(self, msg):
u"""
Команда "удаления" сообщения бота.
"""
if msg['body'][:len(self._bot_name) + 8] == LITERALS['commands']['forget']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
self._forget_data(msg)
return True
return False
def __command_bot_remember(self, msg):
u"""
Команда, для запоминания сообщения.
"""
if msg['body'][:len(self._bot_name) + 9] == LITERALS['commands']['remember']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
self._remember_new_data(msg)
return True
return False
def __command_bot_probability(self, msg):
u"""
Команда получения процента.
"""
if msg['body'][:len(self._bot_name) + 6] == LITERALS['commands']['probability']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
probability = random.randint(0, 100)
if probability == 100:
text = LITERALS['commands']['probability']['answer']['max'].replace('{percent}', str(probability))
elif probability > 75:
text = LITERALS['commands']['probability']['answer']['high'].replace('{percent}', str(probability))
elif probability == 0:
text = LITERALS['commands']['probability']['answer']['min'].replace('{percent}', str(probability))
elif probability < 15:
text = LITERALS['commands']['probability']['answer']['low'].replace('{percent}', str(probability))
else:
text = LITERALS['commands']['probability']['answer']['mid'].replace('{percent}', str(probability))
self._send(msg, u'%s, %s' % (self._get_user_name(msg['user_id']), text))
return True
return False
def __command_bot_message(self, msg):
u"""
Команда отправки ботом сообщения.
"""
if msg['body'][:len(self._bot_name)] == self._bot_name \
and not self._is_msg_in_ignore(msg) and len(msg['body']) > 5:
self._add_msg_to_ignore(msg)
self._send_message_from_db(msg)
return True
return False
def __command_bot_change_mod(self, msg):
u"""
Команда смены режима общения бота.
"""
if msg['body'][:len(self._bot_name) + 12] == LITERALS['commands']['change_mod']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
self.__is_intelligent = not self.__is_intelligent
user_name = self._get_user_name(msg['user_id'])
if self.__is_intelligent:
text = LITERALS['commands']['change_mod']['answer']['intelligency'].replace('{user_name}', user_name)
else:
text = LITERALS['commands']['change_mod']['answer']['default'].replace('{user_name}', user_name)
self._send(msg, text)
return True
return False
def __command_bot_current_mod(self, msg):
u"""
Команда получения текущего режима общения.
"""
if msg['body'][:len(self._bot_name) + 6] == LITERALS['commands']['current_mod']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
user_name = self._get_user_name(msg['user_id'])
if self.__is_intelligent:
text = LITERALS['commands']['current_mod']['answer']['intelligency'].replace('{user_name}', user_name)
else:
text = LITERALS['commands']['current_mod']['answer']['default'].replace('{user_name}', user_name)
self._send(msg, text)
return True
return False
def __command_bot_news(self, msg):
u"""
Команда получения актуальных новостей по РФ и СПб с сайта http://yandex.ru.
"""
if msg['body'][:len(self._bot_name) + 8] == LITERALS['commands']['news']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
self._send(msg, u'%s,\n%s' % (self._get_user_name(msg['user_id']), self._news.get_news()))
return True
return False
def __command_bot_tv_list(self, msg):
u"""
Команда получения актуальной телепрограммы с сайта http://yandex.ru.
"""
if msg['body'][:len(self._bot_name) + 14] == LITERALS['commands']['tv_list']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
self._send(msg, u'%s,\n%s' % (self._get_user_name(msg['user_id']), self._news.get_tv_list()))
return True
return False
def __command_bot_math(self, msg):
u"""
Команда для вычисления математических выражений.
"""
if msg['body'][:len(self._bot_name) + 9] == LITERALS['commands']['math']['cmd']['help']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
user_name = self._get_user_name(msg['user_id'])
help_text = self._math.get_help()
self._send(
msg,
LITERALS['commands']['math']['answer'].replace('{user_name}', user_name).replace('{help}', help_text)
)
return True
if msg['body'][:len(self._bot_name) + 3] == LITERALS['commands']['math']['cmd']['calc']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
self._send(msg, u'%s, %s' % (
self._get_user_name(msg['user_id']),
self._math.calculate(msg['body'][len(self._bot_name) + 3:])
))
return True
return False
def __command_bot_who_is(self, msg):
u"""
Команда вывода ответа на вопрос "кто ...?"
"""
if 'chat_id' in msg:
if msg['body'][:len(self._bot_name) + 4] == LITERALS['commands']['who_is']['cmd']\
.replace('{bot_name}', self._bot_name):
self._add_msg_to_ignore(msg)
user_name = self._get_chat_users_ids(msg['chat_id'])
answer = LITERALS['commands']['who_is']['answer']
self._send(msg, u'%s, %s' % (
self._get_user_name(msg['user_id']),
answer[random.randint(0, len(answer) - 1)].replace('{user_name}', user_name)
))
return True
return False
def __analyze_messages(self, msg_list=None):
u"""
Основной метод анализа входящих сообщений.
Бот анализирует только те сообщения, которые адресованы именно ему.
Анализ идет исходя из правил русского языка: "ИмяБота[,] текст сообщения"
"""
if msg_list:
for msg in msg_list:
if not self._is_msg_in_ignore(msg):
if not self.__disable:
if msg['user_id'] in ADMIN_LIST:
self.__command_bot_off(msg)
self.__command_bot_forget(msg)
self.__command_bot_help(msg)
self.__command_bot_remember(msg)
self.__command_bot_probability(msg)
self.__command_bot_change_mod(msg)
self.__command_bot_current_mod(msg)
self.__command_bot_news(msg)
self.__command_bot_tv_list(msg)
self.__command_bot_math(msg)
self.__command_bot_who_is(msg)
# априори последняя команда, иначе все что после выполнено не будет.
self.__command_bot_message(msg)
else:
if msg['user_id'] in ADMIN_LIST:
self.__command_bot_on(msg)
def __main(self):
u"""
Основной метод бота для работы.
Читает первые 5 входящих сообщений.
"""
while True:
self.__analyze_messages(self._vk.messages.get(count=5)['items'])
# current_chat_title = vk.messages.getChat(chat_id=target_chat_id)['title']
# if current_chat_title != target_chat_title:
# print('Changed:', current_chat_title, 'to', target_chat_title)
# vk.messages.editChat(chat_id=target_chat_id, title=target_chat_title)
time.sleep(2)
def start_bot(self, debug=False):
u"""
Метод, запускающий бота.
"""
print(u"Starting work...")
while True:
if debug:
self.__main()
else:
try:
self.__main()
except Exception as error_msg:
print('Oops! I\'m restarting. Error: %s' % error_msg)