170 похожих чатов

Есть вот такой код, который, используя sqlachemy, берёт все id

из базы данных и сверяет их с тем id, который был передан в функцию.

def register_new_user(user_id, first_name, last_name):
"""This function checks if user exists in the database and adds him there if he doesn't"""
db_user_ids = session.query(TelegramUserProfile.telegram_id)
if user_id in db_user_ids:
return "Welcome back!"
elif user_id not in db_user_ids:
new_user = TelegramUserProfile(
telegram_id=user_id,
telegram_firstname=first_name,
telegram_lastname=last_name
)
session.add_all([new_user])
session.commit()
return "Registered!"
else:
return "Something went wrong while registering..."

Почему даже несмотря на условие if/else мне при вызове функции выдаёт ошибку, которая, судя по всему, связано как раз с этим условием проверки наличия id в бд.

psycopg2.errors.UniqueViolation: duplicate key value violates unique constraint "telegram_user_profile_pkey"
DETAIL: Key (telegram_id)=(579647634) already exists.

Как это решить?
Ну и в целом, не трольте пожалуйста. Я знаю, что эта моя имплементация проверки наличия пользователя в бд далека от идеала. Как можно сделать лучше?

4 ответов

5 просмотров

Проблема возникает при миграции/импорте данных. PostgreSQL перестает понимать столбцы и хочет создать еще одно значение к уникальному столбцу.

Попробуй в try except обернуть

а db_user_ids что содержит? не список кортежей?

а не проще было сделать session.get(TelegramUserProfile, user_id)?

Похожие вопросы

Обсуждают сегодня

я не магистр хаскеля, но разве не может лейзи тип конвертнуться в не-лейзи запросив вычисление содержимого прям при инициализации?
deadgnom32 λ madao
100
в сях есть множество как в питоне? для удаление дубликатов
Linus
25
читать файл максимально быстро? странный вопрос))
zamtmn
53
тоесть, указав return eax, сгенерируется никому ненужная инструкция mov eax,eax ?
Aiwan \ (•◡•) / _bot
24
How to create an OS in C? what to study?
Linus
18
а зачем этот вопрос для удаления из чата?
Mёdkinson Medvezhkin
63
Всем доброго вечера! Хочу поделиться своим злоключением с человеком, который, как оказалось сюда тоже скидывал свое резюме. Жаль, что я вашу группу не нашел раньше… человек ки...
Роман Ахмедзянов
4
а как бы вылезти из ИО, что то типа IO -> Ether или в какую сторону смотреть ? что то туплю
Fedor
14
Компания Elif ищет менеджера проектов, который будет заниматься поиском и ведением новых проектов. Прежде чем приступить к работе, вам нужно пройти наш недельный курс, где вы ...
Elif
5
Привет, кто может сделать юзербота с апи? Задачи: - создавать группы - создавать каналы - задавать для созданных каналов аватарку или эмоджи, имя группы - добавлять в группы...
Lencore
11
Карта сайта