This commit is contained in:
KatykhinAA
2024-07-15 22:28:36 +03:00
parent 1a0344d0e8
commit f7f6fa5482
7 changed files with 79 additions and 11 deletions

View File

@@ -562,6 +562,62 @@ class BotDB:
finally:
self.close()
def get_username_and_full_name(self, user_id: int):
"""
Получает full_name и username пользователя по ID из базы
Args:
date (str): Новая дата изменения.
user_id (int): Идентификатор пользователя в Telegram.
Returns:
username (str): username пользователя
full_name (str): full_name пользователя
"""
self.logger.info(
f"Запуск функции check_username_and_first_name: user_id={user_id}")
try:
self.connect()
self.cursor.execute("SELECT username FROM our_users WHERE user_id = ?", (user_id,))
username = self.cursor.fetchone()[0]
self.cursor.execute("SELECT full_name FROM our_users WHERE user_id = ?", (user_id,))
full_name = self.cursor.fetchone()[0]
self.logger.info(
f"Функция check_username_and_first_name успешно отработала: user_id={user_id}, username={username}, full_name={full_name}")
return username, full_name
except sqlite3.Error as error:
self.logger.error(f"Ошибка в функции get_username_and_first_name: {error}")
return None
finally:
self.close()
def update_username_and_full_name(self, user_id: int, username: str, full_name: str):
"""
Обновляет full_name и username пользователя
Args:
username (str): username пользователя
full_name (str): full_name пользователя
user_id (int): Идентификатор пользователя в Telegram
Returns:
True (bool): Если обновления прошли успешно
sqlite3. Error: Если произошла ошибка при выполнении запроса.
"""
self.logger.info(
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.conn.commit()
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}")
raise
finally:
self.close()
def update_date_for_user(self, date: str, user_id: int):
"""
#TODO: Не возвращается ошибка sqlite3. Error. Тест не перехватывает. Возвращается no such table: our_users
@@ -785,6 +841,7 @@ class BotDB:
return None
except sqlite3.Error as error:
print(error)
raise
finally:
self.close()