Как-то меня осенила мысль: «Зачем на работе постоянно прятать вкладку браузера с ВК от начальства, если можно ее замаскировать?»
И несмотря на то, что моему начальству в целом пофиг на социальные сети во время работы, через несколько минут проблема была решена довольно интересным способом, о котором я сейчас вам и расскажу.
Нам понадобятся:
- Стандартные Python модули: os, sys, json, urllib2
- Модуль для авторизации из этой статьи
Писать код мы будем собственно для Python версии 2.7, но при желании его можно оптимизировать и до третьей версии. Для нетерпеливых сразу дам ссылку на GitHub.
Итак, начнем
Первое и самое главное, что нам понадобится, это получить токен. Пример получения:
import vk_auth app_id = "4925055" access_token = vk_auth.auth(YOUR_LOGIN, YOUR_PASSWORD, app_id, "offline,messages")[0]
Предполагается, что выше указанный модуль авторизации лежит в одной директории с нашим скриптом и называется vk_auth.py
Теперь у нас есть токен и мы даже можем сделать какой-нибудь запрос. Давайте для примера получим JSON из 10 наших последних диалогов, преобразуем его в словарь и красиво выведем на экран:
import urllib2 import json import vk_auth app_id = "4925055" login = raw_input(u"Enter your login: ") password = raw_input(u"Enter your pass: ") access_token = vk_auth.auth(login, password, app_id, "offline,messages")[0] dialogs = json.loads(urllib2.urlopen("https://api.vk.com/method/messages.getDialogs?count=10&access_token=%s&v=5.33" % (access_token)).read().decode("utf-8")) layer = "" for item in reversed(dialogs["response"]["items"]): user = json.loads(urllib2.urlopen("https://api.vk.com/method/users.get?user_id=%s&fields=contacts,online&access_token%s&v=5.8" % (item["message"]["user_id"],access_token)).read().decode("utf-8"))["response"][0]; if (user["online"] == 1): user["online"] = "online" else: user["online"] = "offline" layer+= "%s %s [%s] (%s):\n" % (user["first_name"], user["last_name"], user["online"], item["message"]["user_id"]) layer+= "%s \n" % item["message"]["body"] layer+= "--------------------------------------------\n" print (layer)
Теперь мы можем при запуске скрипта ввести свои данные для авторизации и получить список из 10 последних диалогов с пометкой в виде имени собеседника и его статусом.
Но ведь хотелось бы иметь возможность читать сообщения внутри конкретного диалога и отправлять сообщения. Хорошо, для этого мы создадим класс Connect, в котором опишем 3 метода для взаимодействия с API (получение диалогов, получение сообщений из диалога, отправка сообщения в рамках выбранного диалога) и метод-контроллер для возможности управлять вышеупомянутым функционалом.
Собственно весь код:
import os import sys import json import urllib2 import vk_auth os.system("clear") app_id = "4925055" login = raw_input(u"Enter your login: ") password = raw_input(u"Enter your pass: ") class Connect: def __init__(self,app_id,login,password): self.access_token = vk_auth.auth(login, password, app_id, "offline,messages")[0] print(self.access_token) self.layer = "" self.uid = "" self.section = "" def dialog(self, uid=False): if uid != self.uid and uid: self.uid = uid self.layer = "" self.section = "dialog" dialog = json.loads(urllib2.urlopen("https://api.vk.com/method/messages.getHistory?count=20&user_id=%s&access_token=%s&v=5.33" % (self.uid, self.access_token)).read().decode("utf-8")) for item in reversed(dialog["response"]["items"]): user = json.loads(urllib2.urlopen("https://api.vk.com/method/users.get?user_id=%s&fields=contacts&access_token%s&v=5.8" % (item["from_id"],self.access_token)).read().decode("utf-8"))["response"][0]; self.layer+= "%s %s (%s):\n" % (user["first_name"], user["last_name"], item["user_id"]) self.layer+= "%s \n" % item["body"] self.layer+= "-------------------------------------------- \n" def dialogs(self): self.layer = "" self.section = "dialogs" dialogs = json.loads(urllib2.urlopen("https://api.vk.com/method/messages.getDialogs?count=10&access_token=%s&v=5.33" % (self.access_token)).read().decode("utf-8")) for item in reversed(dialogs["response"]["items"]): user = json.loads(urllib2.urlopen("https://api.vk.com/method/users.get?user_id=%s&fields=contacts,online&access_token%s&v=5.8" % (item["message"]["user_id"],self.access_token)).read().decode("utf-8"))["response"][0]; if (user["online"] == 1): user["online"] = "online" else: user["online"] = "offline" self.layer+= "%s %s [%s] (%s):\n" % (user["first_name"], user["last_name"], user["online"], item["message"]["user_id"]) self.layer+= "%s \n" % item["message"]["body"] self.layer+= "--------------------------------------------\n" def send(self, message): message = urllib2.quote(message).encode('utf8') dialog = json.loads(urllib2.urlopen("https://api.vk.com/method/messages.send?user_id=%s&message=%s&access_token=%s&v=5.33" % (self.uid, message, self.access_token)).read().decode("utf-8")) self.controll("r") def controll(self, input): print ("====LOADING====") if input == "d": self.dialogs() elif input == "exit": os.system("clear") sys.exit(0) elif input == "": self.layer = "" elif self.section == "dialogs": if input == "r": self.dialogs() else: self.dialog(input) elif self.section == "dialog": if input == "r": self.dialog() else: self.send(input) os.system("clear") print (self.layer) self.controll("%s" % raw_input("Enter command: \n")) vk = Connect(app_id,login,password) vk.controll("")
Итак, как это работает:
1. Мы вводим свой логин и пароль, которые сохраняются в переменные login и password соотвественно;
2. Создается объект класса Connect, в инициализирующий метод которого передаются логин, пароль и id приложение ВК;
3. В методе инициализации происходит получение токена и сохранение его в локальной переменной класса;
4. Происходит вызов метода controll класса Connect с параметром пустой строки, который очищает экран консоли;
5. Далее скрипт ждет нашей команды.
Команды таковы:
- загрузка и отображение последних 10 диалогов осуществляется указанием «d»
- обновление текущего раздела — «r»
- пустая строка — очищает экран, чтобы никто не увидел ничего лишнего 🙂
- для перехода к конкретному диалогу указываем ID нужного юзера
- находясь внутри диалога, любой текст, не равный «r», «d», «» будет отправлен юзеру
Вот в целом и все. Конечно, этот скрипт не претендует на звание какой-то крутой фичи, но все же в некоторых моментах может оказаться полезным.
На текущий момент работают только текстовые сообщения. Чуть позже добавлю вывод ссылок на изображения, аудио, видео, отображение репостов и поддержку конференций, а если не добавлю, то это может сделать каждый, форкнув этот код.
Дублирую ссылку на GitHub.
Спасибо за внимание!
ссылка на оригинал статьи http://habrahabr.ru/post/270717/
Добавить комментарий