Guest User

Untitled

a guest
Nov 5th, 2020
20
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.47 KB | None | 0 0
  1. @dp.message_handler(content_types=[ContentType.CONTACT], state=UserStates.get_number)
  2. async def save_contact_handler(msg: types.Contact, state):
  3. number = msg.contact.phone_number
  4. uid = msg.from_user.id
  5. await msg.answer('(♻️) Ищу тебя в базе...')
  6. if (find_user(number) == True):
  7. await state.finish()
  8. update_user(number, 'tg_id', uid)
  9. if (if_admin(uid) == True):
  10. print("[INFO] Администратор {0} вошел в сессию.".format(number))
  11. await UserStates.verified_admin.set()
  12. await msg.answer('(✅) Привет, {0}! \nТы являешься администратором, держи специальное меню!'.format(get_info(uid, 'fname')), reply_markup=admin_request)
  13. else:
  14. print("[INFO] Пользователь {0} вошел в сессию.".format(number))
  15. await UserStates.verified_user.set()
  16. await msg.answer('(✅) Привет, {0}! Держи меню.'.format(get_info(uid, 'fname')), reply_markup=menu_request)
  17. else:
  18. await msg.answer('(❌) К сожалению мы не нашли ваш аккаунт в базе сотрудников.')
  19.  
  20. def find_user(number):
  21. cursor = connect_str.cursor()
  22. cursor.execute(f"SELECT * FROM users WHERE number = '{number}'")
  23. cursor.fetchall()
  24. rows = cursor.rowcount
  25. if (rows > 0):
  26. return True
  27. else:
  28. return False
Advertisement
Add Comment
Please, Sign In to add comment