mirror of
https://github.com/EDeev/vkrepost_to_tg.git
synced 2026-10-07 20:49:50 +03:00
Egor Deev
a38819dee7
- фоновая рассылка падала на каждом новом посте: NameError: name 'types' is not defined (видно в логе сервера) — бот не пересылал ничего; - пост с одной картинкой не отправлялся: send_media_group принимает от 2 до 10 элементов; больше 10 фото — тоже ошибка; теперь одиночное фото — send_photo, альбом обрезается до 10; - подпись длиннее 1024 символов ломала отправку — длинный текст уходит отдельными сообщениями (с делением по строкам, чтобы не рвать HTML); - если первым вложением была ссылка или опрос, текст поста терялся (подпись ставилась только на вложение с индексом 0); - текст и названия из ВК подставлялись в HTML без экранирования; ссылки оформлялись Markdown-синтаксисом ([..](..)) при разметке HTML; - post_message[0] у одиночного сообщения, logging.danger, /notif без подписок — падения; - у видео без photo_800 разбор поста падал; «Лайков» у видео на самом деле были просмотры; - синхронные вызовы vk_api и time.sleep блокировали бота — теперь в отдельном потоке и asyncio.sleep; ошибка у одного подписчика не мешает остальным. Новое: /logout — удалить сохранённый токен VK; сообщение со ссылкой-токеном удаляется из переписки. Токены бота и VK — из переменных окружения, таблицы SQLite создаются при первом запуске (пустые базы убраны из репозитория), ссылка «АВТОР» — на EDeev.
306 lines
16 KiB
Python
306 lines
16 KiB
Python
import asyncio
|
||
import logging
|
||
|
||
from aiogram import types, F, Router
|
||
from aiogram.types import Message, CallbackQuery, ContentType
|
||
from aiogram.filters import Command
|
||
|
||
from emoji import emojize
|
||
|
||
from init import bot, du, db
|
||
from config import checkUrl, loginUrl, serviceToken
|
||
from vk_scripts import VkParser
|
||
from scripts import login, pars_post, send_post
|
||
|
||
router = Router()
|
||
|
||
|
||
@router.message(Command("start", "help"))
|
||
async def start(msg: Message) -> None:
|
||
login(msg.chat.id, du, db)
|
||
|
||
buttons = [[types.InlineKeyboardButton(text="КОМАНДЫ", callback_data="com"),
|
||
types.InlineKeyboardButton(text="АВТОР", callback_data="auth")]]
|
||
keyboard = types.InlineKeyboardMarkup(inline_keyboard=buttons)
|
||
|
||
await msg.answer(text=f'<b>Portal in VK</b> - это бот для перепоста постов со страниц в социальной сети ВКонтакте. '
|
||
f'Для начала работы вам нужно всего лишь вызвать команду <b>/add</b> и добавить к ней '
|
||
f'короткое имя личной или публичной страницы. После этого, бот начнёт присылать каждый новый '
|
||
f'пост с этой страницы форматируя данные с неё под сообщение в Telegram. С полным '
|
||
f'функционалом бота, вы можете ознакомиться по кнопке <b>КОМАНДЫ</b>\n\n'
|
||
|
||
f'Если вы хотите ставить лайки или получать уведомления от закрытых страниц на которые '
|
||
f'подписаны в социальной сети, вам необходимо скинуть ссылку из поисковой строки в чат, '
|
||
f'после подтверждения на <a href="{loginUrl}">сайте</a>', reply_markup=keyboard)
|
||
|
||
|
||
# ИЛАЙН КЛАВИАТУРА HELP
|
||
@router.callback_query(F.data == "com")
|
||
async def function(call: CallbackQuery) -> None:
|
||
await call.message.answer(text='<b>| КОМАНДЫ |</b>\n\n'
|
||
'<b>/like</b> - лайк на пост, который вы отметили\n'
|
||
'<b>/list</b> - список страниц от, которых вы получаете уведомления\n'
|
||
'<b>/notif</b> - отписка или подписка от всех уведомлений\n'
|
||
'<b>/last_post</b> - получение последнего поста по короткому имени страницы\n'
|
||
'<b>/add</b> и <b>/del</b> - добавление и удаление странички из списка подписок\n'
|
||
'<b>/update</b> - ответом на пост: прислать его заново с актуальными данными\n'
|
||
'<b>/logout</b> - удалить сохранённый токен VK\n')
|
||
|
||
|
||
@router.callback_query(F.data == "auth")
|
||
async def author(call: CallbackQuery) -> None:
|
||
await call.message.answer(text='<b>| АВТОР |</b>\n\n<b>>></b> Этот бот не коммерческий проект, для получения '
|
||
'постов из социальной сети ВКонтакте. Бот работает на сервисном токене VK API, '
|
||
'пока вы не предоставите собственный. Ваш токен нужен для получения постов со '
|
||
'страниц, доступ к которым есть исключительно у вас. Также это даст вам '
|
||
'возможность ставить через Telegram лайки на посты в самой социальной сети!'
|
||
|
||
'\n\nЯ же пишу подобные небольшие проекты, о которых вы можете узнать '
|
||
'больше на моём <a href="https://github.com/EDeev">GitHub</a>.')
|
||
|
||
|
||
# КОМАНДЫ
|
||
@router.message(Command("notif"))
|
||
async def notification(msg: Message) -> None:
|
||
user_id = login(msg.chat.id, du, db)
|
||
status = db.get_status(user_id)
|
||
|
||
# у пользователя может ещё не быть ни одной подписки
|
||
groups = db.get_user_groups(user_id)
|
||
for group in map(int, groups.split(";") if groups else []):
|
||
db.update_countGroup(group, -1 if status else 1)
|
||
|
||
if status: await msg.answer("Получение постов из ВК выключено!")
|
||
else: await msg.answer("Получение постов из ВК включено!")
|
||
db.update_status(user_id)
|
||
|
||
|
||
@router.message(Command("add", "del"))
|
||
async def add_del(msg: Message) -> None:
|
||
user_id = login(msg.chat.id, du, db)
|
||
|
||
domain = msg.text.split()[-1]
|
||
command = msg.text.split()[0]
|
||
|
||
if "/" not in domain:
|
||
try:
|
||
token = db.get_token(user_id)
|
||
if token: vk = VkParser(token)
|
||
else: vk = VkParser(serviceToken)
|
||
|
||
group_id, typ, last_post = await asyncio.to_thread(vk.login, domain)
|
||
|
||
if not du.group_exists(group_id):
|
||
du.add_group(group_id)
|
||
group_id = du.get_group_id(group_id)
|
||
|
||
if not db.infoGroup_exists(group_id):
|
||
db.add_infoGroup(group_id, typ, last_post)
|
||
|
||
groups = db.get_user_groups(user_id)
|
||
user_groups = groups.split(";") if groups else []
|
||
|
||
if str(group_id) not in user_groups:
|
||
if command == "/del":
|
||
await msg.answer("Вы и так не подписаны на эту группу!")
|
||
else:
|
||
if db.get_countUser(user_id) < 10:
|
||
user_groups.append(str(group_id))
|
||
db.update_user_groups(user_id, ";".join(user_groups))
|
||
|
||
db.update_countUser(user_id, 1)
|
||
if db.get_status(user_id): db.update_countGroup(group_id, 1)
|
||
|
||
await msg.answer("Группа успешно добавлена в список уведомлений!")
|
||
else:
|
||
await msg.answer("У вас добавлено максимально количество групп!")
|
||
else:
|
||
if command == "/add":
|
||
await msg.answer("Вы уже подписаны на эту страницу!")
|
||
else:
|
||
user_groups.remove(str(group_id))
|
||
db.update_user_groups(user_id, ";".join(user_groups))
|
||
|
||
db.update_countUser(user_id, -1)
|
||
if db.get_status(user_id): db.update_countGroup(group_id, -1)
|
||
|
||
await msg.answer("Группа успешно удалена из списка уведомлений!")
|
||
except Exception:
|
||
logging.info("Нет доступа к страницы:", exc_info=True)
|
||
await msg.answer("<b>Произошла ошибка!</b> Проверьте правильность написания <b>короткого адреса</b> "
|
||
"страницы, <b>имеете ли вы доступ</b> к этой странице и есть ли на ней <b>хотя бы "
|
||
"один пост</b>!")
|
||
else:
|
||
if command == "/add":
|
||
await msg.answer("Команда введена некорректно, такой страницы не существует или она закрыта для "
|
||
"просмотра!\n\nКорректный ввод команд: /add <i>example</i>")
|
||
elif command == "/del":
|
||
await msg.answer("Команда введена некорректно, такой страницы не существует или она закрыта для "
|
||
"просмотра!\n\nКорректный ввод команд: /del <i>example</i>")
|
||
|
||
|
||
@router.message(Command("list"))
|
||
async def lst(msg: Message) -> None:
|
||
user_id = login(msg.chat.id, du, db)
|
||
groups = db.get_user_groups(user_id)
|
||
user_groups = list(map(int, groups.split(";") if groups else []))
|
||
|
||
if user_groups:
|
||
group_ids = [du.get_vk_id(id) for id in user_groups]
|
||
|
||
vk = VkParser(serviceToken)
|
||
peoples, groups = await asyncio.to_thread(vk.info, group_ids)
|
||
|
||
text = "<b><i>* Список страниц, от которых вы получаете уведомления!</i></b>\n"
|
||
|
||
if peoples:
|
||
text += "\n<b>Личные страницы пользователей</b>\n"
|
||
for i, people in enumerate(peoples):
|
||
text += f'{emojize(":green_circle:") if people[2] else emojize(":red_circle:")} ' \
|
||
f'{i + 1}. {people[1]} <a href="https://vk.com/{people[0]}">@{people[0]}</a>\n'
|
||
|
||
if groups:
|
||
text += "\n<b>Группы / сообщества / паблики</b>\n"
|
||
for i, group in enumerate(groups):
|
||
text += f'{i + 1}. {group[1]} <a href="https://vk.com/{group[0]}">@{group[0]}</a>\n'
|
||
|
||
await msg.answer(text, disable_web_page_preview=True)
|
||
else:
|
||
await msg.answer("Вы <b>не добавили ни одной страницы</b> для отслеживания публикаций!")
|
||
|
||
|
||
@router.message(Command("like"))
|
||
async def like(msg: Message) -> None:
|
||
user_id = login(msg.chat.id, du, db)
|
||
token = db.get_token(user_id)
|
||
|
||
if token and (msg.reply_to_message is not None):
|
||
url = ''
|
||
|
||
if msg.reply_to_message.entities is not None:
|
||
for i in list(msg.reply_to_message.entities):
|
||
i = dict(i)
|
||
if i['type'] == 'text_link':
|
||
url = i['url']; break
|
||
elif msg.reply_to_message.caption_entities is not None:
|
||
for i in list(msg.reply_to_message.caption_entities):
|
||
i = dict(i)
|
||
if i['type'] == 'text_link':
|
||
url = i['url']; break
|
||
|
||
if url:
|
||
try:
|
||
[group_id, last_post] = list(map(int, url.split('wall')[-1].split('_')))
|
||
|
||
vk = VkParser(token)
|
||
await asyncio.to_thread(vk.like, group_id, last_post)
|
||
|
||
await msg.answer('Вы поставили лайк 😉')
|
||
except Exception:
|
||
logging.error("Не удалось поставить лайк:", exc_info=True)
|
||
await msg.answer('По <b>неизвестной причине</b> не удалось поставить лайк!')
|
||
else:
|
||
await msg.answer('Вы <b>ответили не на тот пост</b>, для отправки лайка!')
|
||
else:
|
||
await msg.answer('Вы <b>не отметили пост</b> который хотите лайкнуть или <b>не прислали свой токен</b> для '
|
||
'возможности лайкать посты!')
|
||
|
||
|
||
@router.message(Command("update"))
|
||
async def update(msg: Message) -> None:
|
||
user_id = msg.chat.id
|
||
short_id = login(msg.chat.id, du, db)
|
||
|
||
if msg.reply_to_message is not None:
|
||
url = ''
|
||
|
||
if msg.reply_to_message.entities is not None:
|
||
for i in list(msg.reply_to_message.entities):
|
||
i = dict(i)
|
||
if i['type'] == 'text_link':
|
||
url = i['url']; break
|
||
elif msg.reply_to_message.caption_entities is not None:
|
||
for i in list(msg.reply_to_message.caption_entities):
|
||
i = dict(i)
|
||
if i['type'] == 'text_link':
|
||
url = i['url']; break
|
||
|
||
if url:
|
||
try:
|
||
[group_id, last_post] = list(map(int, url.split('wall')[-1].split('_')))
|
||
|
||
token = db.get_token(short_id)
|
||
if token: vk = VkParser(token)
|
||
else: vk = VkParser(serviceToken)
|
||
|
||
output = await asyncio.to_thread(vk.last_post, post=f"{group_id}_{last_post}")
|
||
text, audio, media = pars_post(output)
|
||
|
||
await bot.delete_message(chat_id=user_id, message_id=msg.reply_to_message.message_id)
|
||
await bot.delete_message(chat_id=user_id, message_id=msg.message_id)
|
||
|
||
await send_post(bot, user_id, text, audio, media)
|
||
|
||
except Exception:
|
||
logging.error("Ошибка обновления поста:", exc_info=True)
|
||
await msg.answer(text='По неизвестной причине не удалось обновить пост!')
|
||
else: await msg.answer(text='Вы ответили не на тот пост!')
|
||
else: await msg.answer(text='Вы не отметили пост который хотите обновить!')
|
||
|
||
|
||
# ВЗАИМОДЕЙСТВИЕ С ДАННЫМИ ВК ПОСТОВ
|
||
@router.message(Command("last_post"))
|
||
async def last_post(msg: Message) -> None:
|
||
user_id = msg.chat.id
|
||
short_id = login(user_id, du, db)
|
||
|
||
domain = msg.text.split()[-1]
|
||
if "/" not in domain:
|
||
try:
|
||
token = db.get_token(short_id)
|
||
if token: vk = VkParser(token)
|
||
else: vk = VkParser(serviceToken)
|
||
|
||
output = await asyncio.to_thread(vk.last_post, domain=domain)
|
||
text, audio, media = pars_post(output)
|
||
|
||
await send_post(bot, user_id, text, audio, media)
|
||
|
||
except Exception:
|
||
logging.info("Некорректное/недоступное короткое имя:", exc_info=True)
|
||
await msg.answer("Короткое <b>имя страницы неверное</b> или у вас <b>нет доступа</b> к этой странице!")
|
||
else: await msg.answer("Команда введена некорректно! Корректный ввод: /last_post <b>example</b>")
|
||
|
||
|
||
@router.message(Command("logout"))
|
||
async def logout(msg: Message) -> None:
|
||
user_id = login(msg.chat.id, du, db)
|
||
|
||
if db.get_token(user_id):
|
||
db.update_token(user_id, None)
|
||
await msg.answer("Токен VK удалён, бот снова работает на сервисном ключе. Полностью отозвать доступ "
|
||
"можно в настройках VK: <b>Приложения и сайты</b>.")
|
||
else:
|
||
await msg.answer("Сохранённого токена нет.")
|
||
|
||
|
||
# ПОЛУЧЕНИЕ ТОКЕНА ПОЛЬЗОВАТЕЛЯ
|
||
@router.message(F.content_type == ContentType.TEXT)
|
||
async def url(msg: Message) -> None:
|
||
user_id = login(msg.chat.id, du, db)
|
||
|
||
if checkUrl in msg.text:
|
||
try:
|
||
token = msg.text.split("&")[0].split("=")[-1]
|
||
await asyncio.to_thread(VkParser(token).check)
|
||
|
||
db.update_token(user_id, token)
|
||
# ссылка с токеном не должна оставаться в переписке
|
||
try: await msg.delete()
|
||
except Exception: pass
|
||
await msg.answer("Токен успешно сохранён! Удалить его можно командой /logout")
|
||
except Exception:
|
||
logging.info("Некорректная ссылка:", exc_info=True)
|
||
await msg.answer("Ваша <b>ссылка не корректна</b>, попробуйте повторить копирование!")
|
||
else: await msg.answer("Ваше <b>сообщение не корректно</b>, необходима ссылка, что отобразилась в строке "
|
||
"поиске после подтверждения доступа!")
|