@@ -9,20 +9,16 @@ def create_user(telegram_id: int) -> None:
99 select_query = select (User .telegram_id ).where (User .telegram_id == telegram_id )
1010 session = Database ().session
1111 if session .execute (select_query ).fetchone ():
12- logger .debug (f"Такой пользователь уже есть в базе! Его Telegram-id - { telegram_id } " )
1312 return
1413 insert_query = User (telegram_id = telegram_id )
1514 session .add (insert_query )
1615 session .commit ()
17- logger .debug (f"Добавлен новый юзер! Telegram-id - { telegram_id } " )
1816
1917
2018def create_user_bot_session (user : User , user_bot_session : str ):
21- logger .debug ("Зашёл в создание сессии" )
2219 session = Database ().session
2320 session .add (Session (user_id = user .id , session = user_bot_session ))
2421 session .commit ()
25- logger .debug ("После создания сессии в бд" )
2622
2723
2824def get_user_by_id_telegram_id (telegram_id : int ) -> User | None :
@@ -36,19 +32,15 @@ def get_user_by_id_telegram_id(telegram_id: int) -> User | None:
3632
3733def check_vip (telegram_id ) -> bool :
3834 select_query = select (User .vip ).where (User .telegram_id == telegram_id )
39- logger .debug ("Отправляю запрос на проверку vip" )
4035 result = bool (Database ().session .execute (select_query ).fetchone ()[0 ])
41- logger .debug (f"result: { result } " )
4236 return result
4337
4438
4539def set_vip (telegram_id ) -> None :
4640 update_query = update (User , values = {User .vip : 1 }).where (User .telegram_id == telegram_id )
47- logger .debug ("Отправляю запрос на установку vip статуса" )
4841 session = Database ().session
4942 session .execute (update_query )
5043 session .commit ()
51- logger .debug ("Vip статус установлен!" )
5244
5345
5446def get_users_with_sessions ():
0 commit comments