из базы данных и сверяет их с тем 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.
Как это решить?
Ну и в целом, не трольте пожалуйста. Я знаю, что эта моя имплементация проверки наличия пользователя в бд далека от идеала. Как можно сделать лучше?
Проблема возникает при миграции/импорте данных. PostgreSQL перестает понимать столбцы и хочет создать еще одно значение к уникальному столбцу.
Попробуй в try except обернуть
а db_user_ids что содержит? не список кортежей?
а не проще было сделать session.get(TelegramUserProfile, user_id)?
Обсуждают сегодня