Вывод количества рефералов, ЧЯДНТ?

Question

[Straxd]

Обращаюсь к базе через SELECT COUNT, выводит количество: 0, хотя в базе хранится 3 строки с одним и тем же referrer_id.

Файл main:

elif message.text == '‍‍Пригласить друга':
                await bot.send_message(message.from_user.id, 'ID: {}\nhttps://t.me/{}?start={} \nКол-во рефералов: {}'.format(message.from_user.id, cfg.BOT_NICKNAME, message.from_user.id, db.count_referals(message.from_user.id)))

Файл db:

def count_referals (self, user_id):
        with self.connection:
            return self.cursor.execute("SELECT COUNT('id') as count FROM 'users' WHERE 'referrer_id' = ?", (user_id,)).fetchone()[0]

Answer 1

[Сергей Горностаев]

Вы сравниваете строку 'referrer_id' со значением переменной user_id, естественно, что в таблице будет 0 строк удовлетворяющих такому предикату.

Comments

[Straxd]

Ну так правильно, в таблице есть три столбца: id, user_id и referrer_id. Я ищу в столбце referrer_id все совпадения с моим id (user_id) и вывожу у скольких user_id в поле referrer_id стоит мой id.

[Layton]

Straxd, Вам как минимум нужно сменить fetchone на fetchall

[Сергей Горностаев]

Straxd, неправильно. Прямые одинарные кавычки означают строку, а не имя столбца.

[Сергей Горностаев]

Layton, зачем?

[Layton]

Сергей Горностаев, А разве fecthone выведет не только первый попавшийся результат?

[Сергей Горностаев]

Layton, ему и нужен только один результат.

[Layton]

Сергей Горностаев, А, да, извиняюсь. Не на то обратил внимание.

[Straxd]

Сергей Горностаев, убрал все ' " внутри запроса и сработало, спасибо!

[Straxd]

Сергей Горностаев, подскажите тогда почему тут все работает?

def add_user(self, user_id, referrer_id=None):
        with self.connection:
            # --- Пользователь зарегистрировался по реферальной ссылке ---
            if referrer_id != None:
                return self.cursor.execute("INSERT INTO 'users' ('user_id', 'referrer_id') VALUES (?,?)", (user_id, referrer_id))
            # --- Пользователь зарегистрировался не по реферальной ссылке ---
            else:
                return self.cursor.execute("INSERT INTO 'users' ('user_id') VALUES (?)", (user_id,))

[Сергей Горностаев]

Straxd, честно говоря, не знаю. Не должно работать.

[o5a]

Straxd, это просто sqlite позволяет такие вольности там, где не может быть двузначности: при insert явно идут названия таблиц и полей, там не может быть текстовых строк. Поэтому и работает.
Но лучше вообще эти одинарные кавычки не использовать даже там.