主页 / 问题 / 1163798
Accepted
Noname
Asked:2020-08-09 14:38:52 +0000 UTC2020-08-09 14:38:52 +0000 UTC

如何检查 sqlite3 中的值？

  • 772

我这里有问题，我需要检查sqlite中的值。我正在通过aiogram制作一个机器人，它需要检查表中是否有用户id，如果没有，则添加它。我得到了这个代码 -

@dp.message_handler(commands=['start'])
async def send_welcome(message: types.Message):
    for row in c:
        if row['chat_id'] == message['from']['id']:
            await message.reply('test: no')
        elif row['chat_id'] != message['from']['id']:
            data = [message['from']['id'], str(message['from']['username'])]
            c.execute("INSERT INTO имя таблицы VALUES ?,?", data)
            conn.commit()
            conn.close()
            await message.reply('test: ok')

Sqlite3 和 Python 未发现错误，但代码也无法正常工作。你能帮我吗？提前致谢

python

2 个回答

  1. Best Answer

    下面是一个例子，让你理解逻辑：

    # postgresql
def search_db_users(id_tlgrm):
    conn = psycopg2.connect(dbname=db_name, user=db_user, password=db_password, host=db_host)
    cursor = conn.cursor()
    cursor.execute("SELECT id_tlgrm FROM users WHERE id_tlgrm = %s", (id_tlgrm,))
    idt = cursor.fetchone()
    if idt is None:
        return None
    else:
        return idt[0]


# sqlite
def search_id_tlg(id_tlgrm):
    conn = sqlite3.connect(db)
    cursor = conn.execute("SELECT id_tlgrm FROM users WHERE id_tlgrm=?", (id_tlgrm,)).fetchone()
    if cursor is None:
        conn.close()
        return None
    else:
        conn.close()
        idt = cursor[0]
        return idt


# pytelegrambotapi
@bot.message_handler(commands=['start'])
def start(message):
    if message.chat.id == search_db_users(message.chat.id):
        bot.send_message(message.chat.id, 'где-то я тебя видел')
    else:
        bot.send_message(message.chat.id, 'будем знакомы')

    函数search_id_tlg()接受 Telgram id - search_id_tlg(message.chat.id) request SELECT id_tlgrm FROM users WHERE id_tlgrm=?= select id_tlgrmfrom userswhere id_tlgrm= id_tlgrm:

    • 在第一种情况下 ( SELECT id_tlgrm) - 列
    • 在第二个 ( WHERE id_tlgrm) - 所需的（如果在数据库中可用）
    • 在第三个 ( (id_tlgrm,)) - 我们在数据库中查找的内容，message.chat.id
    • 2

  2. 你可以这样做。

    @dp.message_handler(commands=['start'])
async def start_message(message: types.Message):
      users_id = str(message.from_user.id)
      conn = sqlite3.connect('*название базы*')
      cur = conn.cursor()
      cur.execute(
          "SELECT count(*) FROM users WHERE user_id = ?", (users_id,))
      info = cur.fetchone()[0]
      if info == 0:
         # выполняем если id нема в базе
      else:
         # выполняем если id есть в базе
    • 0

相关问题