Как исправить ошибку с заглавными буквами в названиях столбцов PostgreSQL

При работе с PostgreSQL разработчики часто сталкиваются с ситуацией, когда база данных создаёт столбцы с заглавными буквами (например, ISOCODE, TCGDP). Это приводит к ошибкам при выполнении запросов UPDATE, особенно если код написан без учёта чувствительности Postgres к регистру. В этой статье мы разберём, почему возникает проблема и как правильно писать функцию обновления.

Почему PostgreSQL видит заглавные буквы?

По стандарту SQL, PostgreSQL автоматически приводит все имена таблиц и столбцов к нижнему регистру, если они не взяты в двойные кавычки. Если вы создали таблицу с кавычками ("ISOCODE"), то имена остаются в исходном регистре. В результате запрос UPDATE table SET isocode = 1 не сработает, так как столбец на самом деле называется "ISOCODE" (с заглавными).

Ошибка в исходном коде

В представленном фрагменте кода функция update_by_id использует key.lower() для имени столбца id, но не обрабатывает остальные поля (kwargs). Если названия столбцов в таблице содержат заглавные буквы, Postgres не найдет их, и возникнет исключение.

Как правильно написать функцию обновления

Чтобы функция работала с любым регистром столбцов, необходимо явно оборачивать все имена полей в двойные кавычки. Ниже представлен исправленный код:

def update_by_id(table, id, key='id', **kwargs):
    try:
        # Формируем SET с кавычками для каждого ключа
        set_clause = ', '.join(f'"{k}" = %s' for k in kwargs)
        # Кавычки для ключевого поля (id)
        query = f'UPDATE "{table}" SET {set_clause} WHERE "{key}" = %s'
        update(query, *(tuple(kwargs.values()) + (id,)))
    except Exception as e:
        print(f"Ошибка обновления записи в таблице {table}: {e}")
        print(f"ID: {id}, Данные: {kwargs}")
        raise

Что изменилось?

• Добавлены двойные кавычки вокруг имени таблицы и всех столбцов.
• Убран key.lower(), так как теперь кавычки сохраняют оригинальный регистр.
• Использованы f-строки для читаемости.

Альтернативное решение: переименование столбцов

Если вы контролируете структуру базы данных, проще всего переименовать столбцы в нижний регистр:

ALTER TABLE your_table RENAME COLUMN "ISOCODE" TO isocode;
ALTER TABLE your_table RENAME COLUMN "TCGDP" TO tcgdp;

После этого можно использовать код без кавычек. Однако помните: если таблица уже содержит данные, переименование безопасно и не затрагивает строки.

Рекомендации по стилю кода

Чтобы избежать подобных проблем в будущем:

• Всегда создавайте таблицы с именами столбцов в нижнем регистре без кавычек.
• Если используете кавычки, последовательно применяйте их во всех запросах.
• Для динамических запросов используйте функции экранирования идентификаторов (например, psycopg2.sql.Identifier).

Следуя этим советам, вы избавитесь от ошибок, связанных с регистром, и ваш код будет работать стабильно с любой структурой БД.