from django.db import connection, migrations

if connection.features.supports_raster:

    from django.contrib.postgres.operations import CreateExtension

    pg_version = connection.ops.postgis_version_tuple()

    class Migration(migrations.Migration):

        # PostGIS 3+ requires postgis_raster extension.

        if pg_version[1:] >= (3,):

            operations = [

                CreateExtension("postgis_raster"),

            ]

        else:

            operations = []

else:

    class Migration(migrations.Migration):

        operations = []