python

Простой консольный мессенджер для VK на Python

  • суббота, 14 ноября 2015 г. в 02:10:37
http://habrahabr.ru/post/270717/

Как-то меня осенила мысль: «Зачем на работе постоянно прятать вкладку браузера с ВК от начальства, если можно ее замаскировать?»

И несмотря на то, что моему начальству в целом пофиг на социальные сети во время работы, через несколько минут проблема была решена довольно интересным способом, о котором я сейчас вам и расскажу.

Нам понадобятся:

  • Стандартные 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.

Спасибо за внимание!