Postgres upsert template

postgres_upsert.py

from jinja2 import Template


_POSTGRES_UPSERT_TEMPLATE = Template(
    """
    --beginsql
    INSERT INTO {{ target_table }}
    {{ select_statement }}
    ON CONFLICT ({{ primary_keys|join(', ') }}) DO UPDATE SET
    {%- for col in target_cols %}
    {%- if loop.first %}
    {{ col }} = EXCLUDED.{{ col }}
    {%- else %}
    , {{ col }} = EXCLUDED.{{ col }}
    {%- endif -%}
    {%- endfor %}
    --endsql
    """
)