Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- @dp.message_handler(content_types=[ContentType.CONTACT], state=UserStates.get_number)
- async def save_contact_handler(msg: types.Contact, state):
- number = msg.contact.phone_number
- uid = msg.from_user.id
- await msg.answer('(♻️) Ищу тебя в базе...')
- if (find_user(number) == True):
- await state.finish()
- update_user(number, 'tg_id', uid)
- if (if_admin(uid) == True):
- print("[INFO] Администратор {0} вошел в сессию.".format(number))
- await UserStates.verified_admin.set()
- await msg.answer('(✅) Привет, {0}! \nТы являешься администратором, держи специальное меню!'.format(get_info(uid, 'fname')), reply_markup=admin_request)
- else:
- print("[INFO] Пользователь {0} вошел в сессию.".format(number))
- await UserStates.verified_user.set()
- await msg.answer('(✅) Привет, {0}! Держи меню.'.format(get_info(uid, 'fname')), reply_markup=menu_request)
- else:
- await msg.answer('(❌) К сожалению мы не нашли ваш аккаунт в базе сотрудников.')
- def find_user(number):
- cursor = connect_str.cursor()
- cursor.execute(f"SELECT * FROM users WHERE number = '{number}'")
- cursor.fetchall()
- rows = cursor.rowcount
- if (rows > 0):
- return True
- else:
- return False
Advertisement
Add Comment
Please, Sign In to add comment