add mediaGroup func

This commit is contained in:
KatykhinAA
2024-07-17 00:15:15 +03:00
parent 88889fe87c
commit a3b53d26e2
9 changed files with 183 additions and 39 deletions

View File

@@ -605,9 +605,11 @@ class BotDB:
f"Запуск функции update_username_and_full_name: user_id={user_id}, username={username}, full_name={full_name}")
try:
self.connect()
self.cursor.execute("UPDATE our_users SET username = ?, full_name = ? WHERE user_id = ?", (username, full_name, user_id,))
self.cursor.execute("UPDATE our_users SET username = ?, full_name = ? WHERE user_id = ?",
(username, full_name, user_id,))
self.conn.commit()
self.logger.info(f"Функция update_username_and_full_name. Данные пользователя: user_id={user_id} успешно обновлены")
self.logger.info(
f"Функция update_username_and_full_name. Данные пользователя: user_id={user_id} успешно обновлены")
return True
except sqlite3.Error as error:
self.logger.error(f"Ошибка в функции update_username_and_full_name: {error}")
@@ -823,18 +825,69 @@ class BotDB:
finally:
self.close()
def add_post_from_telegram_in_db(self, group_message_id, message_id, is_last_message_id, content_name, text):
self.logger.info(
f"Запуск функции add_post_from_telegram_in_db: group_message_id={group_message_id}, message_id={message_id}"
f", is_last_message_id={is_last_message_id}, content_name={content_name}, text={text}")
try:
self.connect()
self.cursor.execute(
"INSERT INTO posts_from_telegram (group_message_id, message_id, is_last_message_id, content_name, text)"
"VALUES (?, ?, ?, ?, ?)", (group_message_id, message_id, is_last_message_id, content_name, text))
self.conn.commit()
self.logger.info(f"Функция add_post_from_telegram_in_db отработала успешно")
return True
except Exception as e:
self.logger.error(f"Ошибка в функции add_post_from_telegram_in_db при добавлении поста в базу данных: {e}")
return False
def get_post_content_from_telegram_by_last_id(self, last_post_id: int):
self.logger.info(
f"Запуск функции get_post_content_from_telegram_by_last_id, идентификатор поста {last_post_id}")
try:
self.connect()
main_post_sql = self.cursor.execute("SELECT group_message_id "
"FROM posts_from_telegram WHERE message_id = ?", (last_post_id,))
main_post = main_post_sql.fetchone()[0]
except Exception as e:
self.logger.error(f"Ошибка в функции get_post_content_from_telegram_by_last_id {str(e)}")
return False
try:
result = self.cursor.execute("SELECT content_name "
"FROM posts_from_telegram WHERE group_message_id = ?", (main_post,))
post_content = result.fetchall()
self.logger.info(f"Функция get_post_content_from_telegram_by_last_id получила текст поста: {post_content}")
return post_content
finally:
self.close()
def get_post_text_from_telegram_by_last_id(self, last_post_id: int):
self.logger.info(f"Запуск функции get_post_text_from_telegram_by_last_id, идентификатор поста {last_post_id}")
try:
self.connect()
result = self.cursor.execute("SELECT text "
"FROM posts_from_telegram WHERE message_id = ?", (last_post_id,))
text = result.fetchone()[0]
self.logger.info(f"Функция get_post_text_from_telegram_by_last_id получила text")
return text
except Exception as e:
self.logger.error(f"Ошибка в функции get_post_text_from_telegram_by_last_id {str(e)}")
def add_audio_record(self, file_name, author_id, date_added, listen_count, file_id):
"""Добавляет информацию о войсе юзера в БД"""
self.logger.info(
f"Запуск функции add_audio_record (file_name = {file_name}, author_id = {author_id}, date_added = {date_added}")
f"Запуск функции add_audio_record (file_name = {file_name}, author_id = {author_id},"
f" date_added = {date_added}")
try:
self.connect()
result = self.cursor.execute(
"INSERT INTO `audio_message_reference` (file_name, author_id, date_added, listen_count, file_id) VALUES (?, ?, ?, ?, ?)",
"INSERT INTO `audio_message_reference` (file_name, author_id, date_added, listen_count, file_id) "
"VALUES (?, ?, ?, ?, ?)",
(file_name, author_id, date_added, listen_count, file_id))
self.conn.commit()
self.logger.info(
f"Аудио успешно добавлено в БД (file_name = {file_name}, author_id = {author_id}, date_added = {date_added}")
f"Аудио успешно добавлено в БД (file_name = {file_name}, author_id = {author_id}, "
f"date_added = {date_added}")
return None
except sqlite3.Error as error:
print(error)
@@ -871,7 +924,8 @@ class BotDB:
try:
self.connect()
result = self.cursor.execute(
"SELECT `file_id` FROM `audio_message_reference` WHERE `author_id` = ? ORDER BY date_added DESC LIMIT 1",
"SELECT `file_id` FROM `audio_message_reference` WHERE `author_id` = ? "
"ORDER BY date_added DESC LIMIT 1",
(user_id,))
return result.fetchone()[0]
except sqlite3.Error as error:
@@ -884,7 +938,11 @@ class BotDB:
try:
self.connect()
result = self.cursor.execute(
"SELECT `file_name` FROM `audio_message_reference` WHERE `author_id` = ? ORDER BY date_added DESC LIMIT 1",
"SELECT `file_name` "
"FROM `audio_message_reference` "
"WHERE `author_id` = ? "
"ORDER BY date_added "
"DESC LIMIT 1",
(user_id,))
return result.fetchone()[0]
except sqlite3.Error as error: