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.
179 lines
9 KiB
Python
179 lines
9 KiB
Python
import sqlite3
|
||
|
||
|
||
class Users:
|
||
def __init__(self, database):
|
||
"""Подключаемся к БД и сохраняем курсор соединения"""
|
||
self.connection = sqlite3.connect(database)
|
||
self.cursor = self.connection.cursor()
|
||
|
||
# на новой установке (например, в Docker с пустым томом) таблиц ещё нет
|
||
with self.connection:
|
||
self.cursor.execute("CREATE TABLE IF NOT EXISTS user (id INTEGER PRIMARY KEY, user_id INTEGER NOT NULL)")
|
||
self.cursor.execute("CREATE TABLE IF NOT EXISTS \"group\" (id INTEGER PRIMARY KEY, group_id INTEGER NOT NULL)")
|
||
|
||
# КОМАНДЫ USER
|
||
def user_exists(self, user_id):
|
||
"""Проверяем, есть ли уже пользователь в базе"""
|
||
with self.connection:
|
||
result = self.cursor.execute('SELECT * FROM `user` WHERE `user_id` = ?', (user_id,)).fetchall()
|
||
return bool(len(result))
|
||
|
||
def add_user(self, user_id):
|
||
"""Добавляем нового пользователя"""
|
||
with self.connection:
|
||
return self.cursor.execute("INSERT INTO `user` (`user_id`) VALUES(?)", (user_id,))
|
||
|
||
def get_user_id(self, user_id):
|
||
"""Получаем короткое айди юзера"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `id` FROM `user` WHERE `user_id` = ?', (user_id,)).fetchone()[0]
|
||
|
||
def get_tg_id(self, user_id):
|
||
"""Получаем длинное айди юзера"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `user_id` FROM `user` WHERE `id` = ?', (user_id,)).fetchone()[0]
|
||
|
||
# КОМАНДЫ GROUP
|
||
def group_exists(self, group_id):
|
||
"""Проверяем, есть ли уже группа в базе"""
|
||
with self.connection:
|
||
result = self.cursor.execute('SELECT * FROM `group` WHERE `group_id` = ?', (group_id,)).fetchall()
|
||
return bool(len(result))
|
||
|
||
def add_group(self, group_id):
|
||
"""Добавляем новую группу"""
|
||
with self.connection:
|
||
return self.cursor.execute("INSERT INTO `group` (`group_id`) VALUES(?)", (group_id,))
|
||
|
||
def get_group_id(self, group_id):
|
||
"""Получаем короткое айди группы"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `id` FROM `group` WHERE `group_id` = ?', (group_id,)).fetchone()[0]
|
||
|
||
def get_vk_id(self, group_id):
|
||
"""Получаем длинное айди группы"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `group_id` FROM `group` WHERE `id` = ?', (group_id,)).fetchone()[0]
|
||
|
||
# ЗАКРЫТИЕ ВЫЗОВА
|
||
def close(self):
|
||
"""Закрываем соединение с БД"""
|
||
self.connection.close()
|
||
|
||
|
||
class Base:
|
||
def __init__(self, database):
|
||
"""Подключаемся к БД и сохраняем курсор соединения"""
|
||
self.connection = sqlite3.connect(database)
|
||
self.cursor = self.connection.cursor()
|
||
|
||
with self.connection:
|
||
self.cursor.execute("CREATE TABLE IF NOT EXISTS user (user_id INTEGER NOT NULL, status BOOLEAN NOT NULL "
|
||
"DEFAULT (True), count INTEGER NOT NULL DEFAULT (0), groups TEXT, token TEXT)")
|
||
self.cursor.execute("CREATE TABLE IF NOT EXISTS \"group\" (group_id INTEGER NOT NULL, type BOOLEAN NOT NULL, "
|
||
"count INTEGER NOT NULL DEFAULT (0), last_post INTEGER NOT NULL)")
|
||
|
||
# КОМАНДЫ USER
|
||
def infoUser_exists(self, user_id):
|
||
"""Проверяем, есть ли данные уже в базе"""
|
||
with self.connection:
|
||
result = self.cursor.execute('SELECT * FROM `user` WHERE `user_id` = ?', (user_id,)).fetchall()
|
||
return bool(len(result))
|
||
|
||
def add_infoUser(self, user_id):
|
||
"""Добавляем информацию о пользователе"""
|
||
with self.connection:
|
||
return self.cursor.execute("INSERT INTO `user` (`user_id`) VALUES(?)", (user_id, ))
|
||
|
||
def all_activUser(self):
|
||
"""Список активных пользователей"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `user_id`, `groups` FROM `user` WHERE `status` = 1 AND `count` > 0 '
|
||
'AND `token` NOT NULL').fetchall()
|
||
|
||
def all_subUser(self):
|
||
"""Список пользователей подписанных на группу"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `user_id`, `groups` FROM `user` WHERE `status` = 1 AND `count` > 0').fetchall()
|
||
|
||
def get_status(self, user_id):
|
||
"""Получаем статус рассылки уведомлений"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `status` FROM `user` WHERE `user_id` = ?', (user_id,)).fetchone()[0]
|
||
|
||
def update_status(self, user_id):
|
||
"""Обновляем статус рассылки уведомлений"""
|
||
with self.connection:
|
||
status = self.cursor.execute('SELECT `status` FROM `user` WHERE `user_id` = ?', (user_id,)).fetchone()[0]
|
||
return self.cursor.execute("UPDATE `user` SET `status` = ? WHERE `user_id` = ?", (not status, user_id))
|
||
|
||
def get_user_groups(self, user_id):
|
||
"""Получаем группы на которые подписан пользователь"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `groups` FROM `user` WHERE `user_id` = ?', (user_id,)).fetchone()[0]
|
||
|
||
def update_user_groups(self, user_id, groups):
|
||
"""Обновляем группы на которые подписан пользователь"""
|
||
with self.connection:
|
||
return self.cursor.execute("UPDATE `user` SET `groups` = ? WHERE `user_id` = ?", (groups, user_id))
|
||
|
||
def get_token(self, user_id):
|
||
"""Получаем access token пользователя"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `token` FROM `user` WHERE `user_id` = ?', (user_id,)).fetchone()[0]
|
||
|
||
def update_token(self, user_id, token):
|
||
"""Обновляем access token пользователя"""
|
||
with self.connection:
|
||
return self.cursor.execute("UPDATE `user` SET `token` = ? WHERE `user_id` = ?", (token, user_id))
|
||
|
||
def get_countUser(self, user_id):
|
||
"""Получаем количество подписок пользователя"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `count` FROM `user` WHERE `user_id` = ?', (user_id,)).fetchone()[0]
|
||
|
||
def update_countUser(self, user_id, num):
|
||
"""Обновляем количество подписок пользователя"""
|
||
with self.connection:
|
||
count = self.cursor.execute('SELECT `count` FROM `user` WHERE `user_id` = ?', (user_id,)).fetchone()[0]
|
||
return self.cursor.execute("UPDATE `user` SET `count` = ? WHERE `user_id` = ?", (count + num, user_id))
|
||
|
||
# КОМАНДЫ GROUP
|
||
def infoGroup_exists(self, group_id):
|
||
"""Проверяем, есть ли данные уже в базе"""
|
||
with self.connection:
|
||
result = self.cursor.execute('SELECT * FROM `group` WHERE `group_id` = ?', (group_id,)).fetchall()
|
||
return bool(len(result))
|
||
|
||
def add_infoGroup(self, group_id, tp, last_post):
|
||
"""Добавляем информацию о группе"""
|
||
with self.connection:
|
||
return self.cursor.execute("INSERT INTO `group` (`group_id`, `type`, `last_post`) VALUES(?, ?, ?)",
|
||
(group_id, tp, last_post))
|
||
|
||
def all_notifGroup(self):
|
||
"""Список отслеживаемых групп"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `group_id` FROM `group` WHERE `count` > 0').fetchall()
|
||
|
||
def get_postGroup(self, group_id):
|
||
"""Получаем номер последнего поста"""
|
||
with self.connection:
|
||
return self.cursor.execute('SELECT `last_post` FROM `group` WHERE `group_id` = ?', (group_id,)).fetchone()[0]
|
||
|
||
def update_postGroup(self, group_id, last_post):
|
||
"""Обновляем номер последнего поста"""
|
||
with self.connection:
|
||
return self.cursor.execute("UPDATE `group` SET `last_post` = ? WHERE `group_id` = ?", (last_post, group_id))
|
||
|
||
def update_countGroup(self, group_id, num):
|
||
"""Обновляем количество подписок на группу"""
|
||
with self.connection:
|
||
count = self.cursor.execute('SELECT `count` FROM `group` WHERE `group_id` = ?', (group_id,)).fetchone()[0]
|
||
return self.cursor.execute("UPDATE `group` SET `count` = ? WHERE `group_id` = ?", (count + num, group_id))
|
||
|
||
# ЗАКРЫТИЕ ВЫЗОВА
|
||
def close(self):
|
||
"""Закрываем соединение с БД"""
|
||
self.connection.close()
|