Доступ к боту телеграмм?

Question

[enabl3]

Всем привет, столкнулся с проблемой.
Сделал регистрацию для доступа к боту, что бы он был не всем доступен.
Храню chat.id в БД.
Пользователь пишет определенную команду, мне от бота прилетает сообщение о том что кто-то хочет зарегаться, и я или подтверждаю или отклоняю регистрацию.
После подтверждения новый chat.id пользователя прописывается в БД, но бот все так же пишет что доступ ограничен.
И так будет до тех пор, пока не перезапущу бота. Что делаю не так?
Доступ сделал так:

@bot.message_handler(func=lambda message: str(message.chat.id) not in users)
def send_deny_message(message):
    bot.send_message(message.chat.id, "Нет доступа!")

Comments

[Алексей С.]

проверяйте находится ли в БД ваш новый chat.it , скорее всего не закрыли транзакцию после добавления записи.

[enabl3]

Алексей С., в том то и дело что новый chat.it добавляется в БД, но применяется только после рестарта бота.
После подключении к БД идет инсерт с сохранением и отключением от БД.

conn = lite.connect(bd)
cursor = conn.cursor()
cursor.execute("INSERT.....")
conn.commit()
conn.close()

[Алексей С.]

enabl3, а кусок на чтение из БД покажите
p.s. Пытался вспомнить из за чего это происходит и не смог, но текущая версия того глючающего кода переписано под одно соединение с базой (то есть conn.close не используется) и при каждом обращении к БД вызывается функция которая делает проверку не разорвалось ли соединение и делает реконнект

try:
  cursor.execute( "SELECT VERSION()" )
  results = cursor.fetchone()
  # Check if anything at all is returned
  if results:
   xconn = True
  else:
   xconn = False
 except MySQLdb.Error as e:
  print ("ERR::CheckConnect : ERROR %d IN CONNECTION: %s" % (e.args[0], e.args[1]) )
  xconn = False
...

... и делает реконнект при необходимости

[enabl3]

Алексей С., список chat.id (users) выбираю так:

conn = lite.connect(bd)
cursor = conn.cursor()
cursor.execute("SELECT chat_id FROM 'base'")
users = str(cursor.fetchall())
conn.commit()
conn.close()

[Алексей С.]

enabl3, попробуйте добавить conn.begin() после conn.cursor()

[enabl3]

Алексей С.,

conn = lite.connect(bd)
cursor = conn.cursor()
conn.begin()
cursor.execute("SELECT chat_id FROM 'base'")
users = str(cursor.fetchall())
conn.commit()
conn.close()

ошибка

conn.begin()
AttributeError: 'sqlite3.Connection' object has no attribute 'begin'

[Алексей С.]

enabl3, Идеи закончились , без полного кода не разобраться )) Попробуйте всё таки убрать закрытие соединения с БД и напишите скрипт проверки не закрылось ли оно само по таймауту

[enabl3]

Алексей С., Прикол в том, что не обновляется инфа, только в этом участке, именно где проверка идет:

@bot.message_handler(func=lambda message: str(message.chat.id) not in users)
def send_deny_message(message):
    bot.send_message(message.chat.id, "Нет доступа!")

[Алексей С.]

enabl3, Эммм , только сейчас сообразил, так в этом куске у тебя только отлуп и прописан , другого и не может быть, функция not "in users" сработала только один раз скорее всего , дальше уже не проверяется, переделывай вызов на приём всех сообщений и уже внутри функции проверяй входит ли он в список users

Answer 1

[Алексей С.]

перепишите код , функцию на приём всех сообщений. Внутри функции делайте запрос к БД и далее проверяйте есть доступ у пользователя или нет, в вашем варианте будут отрабатывать только те кто уже прописан в бд на момент старта программы.

Comments

[enabl3]

С этим у меня и возникли проблемы.

def check_users_in_DB(message):
     conn = lite.connect(bd)
     cursor = conn.cursor()
     cursor.execute("SELECT chat_id FROM 'base'")
     userslist = str(cursor.fetchall())
     conn.close()
     if str(message.chat.id) not in userslist:
         bot.send_message(message.chat.id, "Sorry!")
     else:
         #хз как сюда прикрутить выполнение команды, которую выбрал пользователь

И я так понимаю check_users_in_DB нужно будет поместить в каждый хендлер?

[Алексей С.]

ваша функция должна возращать результат (return true если есть в списке и false если нет ) а не отправлять сама сообщение пользователю.
то есть в хендлер который у вас отвечает за приём всех сообщений, их не должно быть много, скорее всего что то типа : @bot.message_handler(content_types=["text"])
нужно первым делом сделать проверку, а есть ли доступ у юзвера:

if (!check_users_in_DB(message.chat.id) ):
 bot.send .Доступ запрещён, зарегистрируйтесь командой /register..
else: #тут уже ваш код

(аналогично для всех команд)
а до него у вас что то типа такого должно быть : @bot.message_handler(commands=['register']), то есть команда на регистрацию отправляется вам, вы в ответ отвечаете, что то типа "ожидайте" ... ну и т.д . ))