diff --git a/bot.py b/bot.py new file mode 100644 index 0000000..fb68ef5 --- /dev/null +++ b/bot.py @@ -0,0 +1,131 @@ +import telebot +from telebot import types +from vk_messages import VkMessage, VkPolling +import vk +import threading +import re + +vk_threads = {} + +vk_tokens = {} + +vk_app_id = "5988748" + +link = 'https://oauth.vk.com/authorize?client_id={}&' \ + 'display=page&redirect_uri=https://oauth.vk.com/blank.html&scope=friends,messages' \ + '&response_type=token&v=5.65'.format(vk_app_id) + +mark = types.InlineKeyboardMarkup() +yes = types.InlineKeyboardButton('ВХОД', url=link) +mark.add(yes) + + +def create_thread(message, vk_token): + a = VkPolling() + t = threading.Thread(name='vk' + str(message.from_user.id), target=a.run, args=(vk_token, bot, message.chat.id,)) + t.setDaemon(True) + t.start() + vk_threads[str(message.from_user.id)] = a + vk_tokens[str(message.from_user.id)] = vk_token + + +bot = telebot.AsyncTeleBot('**********') + + +def extract_unique_code(text): + # Extracts the unique_code from the sent /start command. + try: + text = text[45:] + text = text.split('&') + return text[0] + except: + return None + + +def verifycode(code): + session = vk.Session(access_token=code) + api = vk.API(session) + return dict(api.account.getProfileInfo(fields=[])) + + +def info_extractor(info): + info = info[0].url[8:-3].split('.') + return info + + +def check_thread(message): + for th in threading.enumerate(): + if th.getName() == 'vk' + str(message.from_user.id): + return False + return True + + +def stop_thread(message): + for th in threading.enumerate(): + if th.getName() == 'vk' + str(message.from_user.id): + t = vk_threads[str(message.from_user.id)] + t.terminate() + th.join() + + +@bot.message_handler(commands=['stop']) +def stop_command(message): + if not check_thread(message): + stop_thread(message) + bot.send_message(message.chat.id, 'Успешный выход!').wait() + else: + bot.send_message(message.chat.id, 'Вход не был выполнен!').wait() + + +@bot.message_handler(commands=['start']) +def start_command(message): + if check_thread(message): + bot.send_message(message.chat.id, + 'Привет, этот бот поможет тебе общаться ВКонтакте, войди по кнопке ниже' + ' и отправь мне то, что получишь в адресной строке.', + reply_markup=mark).wait() + else: + bot.send_message(message.chat.id, 'Вход уже выполнен!\n/stop для выхода.').wait() + + +@bot.message_handler(content_types=['text']) +def reply(message): + m = re.search('https://oauth\.vk\.com/blank\.html#access_token=[a-z0-9]*&expires_in=[0-9]*&user_id=[0-9]*', + message.text) + if m: + code = extract_unique_code(m.group(0)) + if check_thread(message): + try: + verifycode(code) + create_thread(message, VkMessage(code)) + bot.send_message(message.chat.id, 'Вход выполнен!').wait() + bot.send_message(message.chat.id, 'Бот позволяет получать и отвечать на текстовые сообщения' + ' из ВКонтакте\nПример личного сообщения:').wait() + bot.send_message(message.chat.id, '*Иван Петров:*\nПривет, я тут классный мессенджер нашёл,' + ' попробуешь? telegram.org/download', parse_mode='Markdown').wait() + bot.send_message(message.chat.id, 'Для сообщений из групповых чатов будет указываться' + ' чат после имени отправителя:').wait() + rp = bot.send_message(message.chat.id, '*Ник Невидов @ My English is perfect:*\n' + 'London is the capital of Great Britain', + parse_mode='Markdown').wait() + bot.send_message(message.chat.id, 'Чтобы ответить, используй Reply на нужное сообщение.' + ' (нет, на эти не сработает, нужно реальное)', + parse_mode='Markdown').wait() + except: + bot.send_message(message.chat.id, 'Неверная ссылка, попробуй ещё раз!').wait() + else: + bot.send_message(message.chat.id, 'Вход уже выполнен!\n/stop для выхода.').wait() + return + + if message.reply_to_message: + info = info_extractor(message.reply_to_message.entities) + if info is not None: + if len(info) - 1: + vk.API(vk_tokens[str(message.from_user.id)].session).messages.send(chat_id=info[1], + message=message.text) + else: + vk.API(vk_tokens[str(message.from_user.id)].session).messages.send(user_id=info[0], + message=message.text) + + +bot.polling(none_stop=True) diff --git a/vk_messages.py b/vk_messages.py new file mode 100644 index 0000000..d4bbcc3 --- /dev/null +++ b/vk_messages.py @@ -0,0 +1,71 @@ +import vk +import time + + +class VkPolling: + def __init__(self): + self._running = True + + def terminate(self): + self._running = False + + def run(self, vk_user, bot, chat_id): + while self._running: + try: + messages = vk_user.get_new_messages() + if messages: + for m in messages: + bot.send_message(chat_id, m, parse_mode='HTML') + except Exception as e: + print('Error: {}'.format(e)) + for i in range(35): + if self._running: + time.sleep(0.1) + else: + break + + +class VkMessage: + def __init__(self, token): + self.session = get_session(token) + self.ts, self.pts = get_tses(self.session) + + def get_new_messages(self): + + api = vk.API(self.session) + new = api.messages.getLongPollHistory(ts=self.ts, pts=self.pts) + msgs = new['messages'] + self.pts = new["new_pts"] + count = msgs[0] + + res = [] + if count == 0: + pass + else: + print(msgs) + messages = msgs[1:] + for m in messages: + if not m['out'] and m['body']: + if 'chat_id' in m: + user = api.users.get(user_ids=m["uid"], fields=[])[0] + data = '​{} {} @ {}:\n{}'.format( + m["uid"], m['chat_id'], m['push_settings']['sound'], user["first_name"], + user["last_name"], m['title'], m["body"].replace('
', '\n')) + res.append(data) + else: + user = api.users.get(user_ids=m["uid"], fields=[])[0] + data = '​{} {}:\n{}'.format( + m["uid"], user["first_name"], user["last_name"], m["body"].replace('
', '\n')) + res.append(data) + return res + + +def get_session(token): + return vk.Session(access_token=token) + + +def get_tses(session): + api = vk.API(session) + + ts = api.messages.getLongPollServer(need_pts=1) + return ts['ts'], ts['pts']