Сегодня меня вновь попросили напомнить человеку о важной вещи в определённое время. Но что делать, если я и про свои-то дела забываю постоянно, а уж тем более про дела кого-то ещё? И тут мне снова помог мой любимый python.
Честно говоря, обычные программы-напоминалки, что в телефоне, что в компьютере, меня не устраивали из-за их ограниченности рамками устройства + они совершенно не решали задачу, когда нужно напомнить о чём-то, но не мне. Решение пришло как-то само-собой. А что, если напоминания будут приходить как сообщения вконтакте? Если я не на рабочем месте — телефон свибрирует своим пуш-ап уведомлением, а за компьютером всё ещё очевиднее. Цель — написать скрипт, который читает мои сообщения о напоминании и в заданное время напоминает кому нужно о том, что, собственно, требуется. Ну раз идея пришла, я приступаю к её реализации.
Стартуем
Для начала научим наш скрипт логиниться в эту социальную сеть. Всё просто, используем стандартный mechanize.Browser()
br = mechanize.Browser()
br.set_handle_equiv(True)
br.set_handle_redirect(True)
br.set_handle_robots(False)
br.open('https://vk.com/')
br.select_form(nr=0)
br.form['email'] = name
br.form['pass'] = password
br.submit()
Вуаля! Мы зашли на свою страничку вконтакте. Теперь используем классную особенность вконтакта — возможность писать самому себе(кто не в курсе, как это делается — перейдите по ссылке vk.com/im?sel=id, где id — Ваш id в социальной сети. В моём случае это был 38591009).
Первым делом прочитаем эти сообщения. Для этого мне нужно узнать свой id(он есть в коде главной страницы, причём повторяется множество раз).
self_username = 'username'
def get_self_page_id(br):
br.open('https://vk.com/'+self_username)
return br.response().read().split('<form action="/wall')[1].split('?')[0]
def check_messages(br):
br.open('https://vk.com/im?sel='+get_self_page_id(br))
response = br.response().read()
Мы видим последние 20 сообщений из тех, что сами же себе и отсылали. Нам этого достаточно. Каждое сообщение имеет свой уникальный(для пользователя) номер, нам это очень полезно. Дальше нужно с ними немного поиграть, чтобы разделить все сообщения, отделить текст от порядкового номера и научить скрипт понимать, какие сообщения новые, а какие уже не актуальны.
first_start = True
msg_numbers = [] #номера сообщений. Глобальная переменная, будет хранить номера сообщений, прочитанных в предыдущей итерации.
def play_with_messages(br, response):
global first_start
all_messages = response.split('class="messages bl_cont">')[1].split('<div id="mfoot"')[0].split('<a name="msg')
all_numbers = []
global msg_numbers
for msg in all_messages:
if msg != all_messages[0]:
msg_num = msg.split('">')[0]
all_numbers.append(msg_num)
if first_start:
msg_numbers = all_numbers
first_start = False
new_numbers = set(all_numbers) - set(all_numbers).intersection(set(msg_numbers))
for num in new_numbers:
reply_to_message(br, get_message_text(response, num)) #вызов функции ответа на сообщение. Опишу её позже.
msg_numbers = all_numbers
Начинаем внутренний диалог
Отлично. Теперь мы знаем какие сообщения поступили мне от меня недавно. Осталось их понять и сделать что-то в ответ. Займёмся сначала первой задачей:
def reply_to_message(br, message):
if message.find('напомнить') == -1:
print 'nothing'
else:
print 'I obey, my lord'
ms_words = message.split(' ')
user = 'self'
time_s = datetime.datetime.now().strftime('%H:%M')
day_s = str(datetime.date.today())
msg = 'something went wrong'
times = message.split('|')
if len(times) == 1:
times = '1'
else:
times = int(times[1])
if ms_words[1] == 'в':
user = 'self'
time_s = ms_words[2]
msg = message.split('текст ')[1].split('|')[0]
elif ms_words[1] == 'день':
user = 'self'
time_s = ms_words[4]
day_s = ms_words[2]
msg = message.split('текст ')[1].split('|')[0]
elif ms_words[2] == 'в':
user = get_page_id(br, ms_words[1])
time_s = ms_words[3]
msg = message.split('текст ')[1].split('|')[0]
elif ms_words[2] == 'день':
user = get_page_id(br, ms_words[1])
time_s = ms_words[5]
day_s = ms_words[3]
msg = message.split('текст ')[1].split('|')[0]
let_it_do(user, time_s, day_s, msg, times) #вызов функции, которая знает, что делать с полученными из сообщения значениями.
Здесь я спличу полученные сообщения и заношу в переменные соответствующие значения. В общем, отвечаю на вопросы «кому напомнить?», «что напомнить?», «когда и сколько раз это сделать?». Синтаксис сообщения/команды выбрал не сложный: напомнить [кому] [дата] в [время] текст [текст сообщения]|[сколько раз]. Вот пример:«напомнить tenoclock в 14:10 текст Очередной тест | 4»
Так наш робот видит внутренний диалог
Для хранения заданий я выбрал базу данных sqlite3. Нагрузка у нас минимальная, разворачивается она совершенно без усилий. Теперь приступим к записи заданий в базу данных, по пути проверяя валидность даты и времени. Выглядит это вот так:
def valid_time(time_text):
try:
datetime.datetime.strptime(time_text, '%H:%M')
return True
except ValueError:
send_message(br_fake, get_self_page_id(br), 'неверный формат времени')
return False
def valid_date(date_text):
try:
datetime.datetime.strptime(date_text, '%Y-%m-%d')
return True
except ValueError:
send_message(br_fake, get_self_page_id(br), 'неверный формат даты')
return False
def let_it_do(user, time_s, day_s, message, times):
if valid_time(time_s) and valid_date(day_s):
c = conn.cursor()
c.execute("INSERT INTO reminder (time, date, user, message, times) VALUES (?,?,?,?,?)",(time_s, day_s, user, message, str(times)))
conn.commit()
Финишная прямая
Мы уже близки к финалу! Задания наш робот получил, себе их записал. По сути, осталось только их выполнить. Тут я столкнулся с небольшой трудностью. Скрипт постоянно читает мои сообщения ко мне, поэтому, если он будет отправлять их в этот-же диалог, то в непрочитанных у меня ничего висеть не будет. А это плохо. Проблема решилась заведением фэйкового аккаунта для этого случая. Теперь если скрипт напоминает мне о чём-то, он пишет со второго аккаунта, если же нужно напомнить кому-то другому, то он пишет от моего имени, дабы людей не пугать.
Собственно вот пара функций, которые отвечают за чтение из базы и отсылку сообщений:
def check_answers():
conn = sqlite3.connect('reminder.db')
rows = get_rows(conn)
for row in rows:
print row[5]
c = conn.cursor()
if row[3] == 'self':
pass
send_message(br_fake, get_self_page_id(br), row[4].encode('utf-8'))
else:
send_message(br, row[3], row[4].encode('utf-8'))
if row[5] == '1' or row[5] == 1:
c.execute("DELETE FROM reminder WHERE id = ?;", str(row[0]))
else:
time_s = (datetime.datetime.now()+datetime.timedelta(seconds=60)).strftime('%H:%M')
num = int(row[5]) - 1
c.execute("UPDATE reminder SET time = ?, times = ? WHERE id = ?",(time_s, str(num), row[0]))
conn.commit()
def send_message(br, id, message):
br.open('https://vk.com/im?sel='+id)
br.select_form(nr=0)
br.form['message'] = message
br.submit()
Ну и после отсылки сообщений скрипт удаляет запись из базы, если она не актуальна(если нужно повторить ещё сколько-то раз, то переносит время напоминания на минуту вперёд и уменьшает количество оставшихся отправок)
Подводим итоги
Так скрипт, который укладывается в 200 строк кода, решает проблему напоминаний себе и другим, используя социальную сеть вконтакте. Целиком его можно скачать отсюда. Если он вдруг кому-то нужен, то советую не собирать из кусков статьи, здесь только функциональные вещи. Некоторые вспомогательные штуки остались за кадром. Я запустил его на одном из своих . Пока, вроде как, удобно.
Робот указывает мне, что делать. В воскресенье! Дожили
После несложных модификаций сюда так-же можно включить любые другие функции управления системой, если скрипт запущен на удалённом компьютере. Поставить тот же торрент на скачивание, например. А так, в целом, можно реализовать веб-сервис, который будет заниматься напоминаниями для всех, кто попросит(фактически бота, как в, уже забытых сейчас, irc и icq) Но эти вещи уже не относятся к данной статье. Буду очень рад, если кому-то это было полезным.
Всем спасибо за внимание.
Автор: tenoclock
