Fix index_together fixer for GIS models module

CHANGELOG.rst

@@ -2,7 +2,10 @@
 Changelog
 =========
 
-* Fix ``check_constraint_condition`` fixer to work when ``django.contrib.gis.db.models`` is used to import ``CheckConstraint``.
+* Make these fixers work when ``django.contrib.gis.db.models`` is used to import objects from ``django.db.models``:
+
+  *  ``check_constraint_condition``
+  *  ``index_together``
 
   `Issue #513 <https://github.com/adamchainz/django-upgrade/issues/513>`__.
 

src/django_upgrade/fixers/index_together.py

@@ -102,9 +102,15 @@ def visit_ClassDef(
     except IndexError:
         indexes = None
 
-    if "models" in state.from_imports["django.db"]:
+    if (
+        "models" in state.from_imports["django.db"]
+        or "models" in state.from_imports["django.contrib.gis.db"]
+    ):
         index_ref = "models.Index"
-    elif "Index" in state.from_imports["django.db.models"]:
+    elif (
+        "Index" in state.from_imports["django.db.models"]
+        or "Index" in state.from_imports["django.contrib.gis.db.models"]
+    ):
         index_ref = "Index"
     else:
         return

tests/fixers/test_index_together.py

@@ -595,3 +595,43 @@ class Meta:
                     indexes = [models.Index(fields=["bill", "tail"])]
         """,
     )
+
+
+def test_models_imported_gis():
+    check_transformed(
+        """\
+        from django.contrib.gis.db import models
+
+        class Duck(models.Model):
+            class Meta:
+                index_together = [["bill", "tail"]]
+                indexes = []
+        """,
+        """\
+        from django.contrib.gis.db import models
+
+        class Duck(models.Model):
+            class Meta:
+                indexes = [models.Index(fields=["bill", "tail"])]
+        """,
+    )
+
+
+def test_index_imported_gis():
+    check_transformed(
+        """\
+        from django.contrib.gis.db.models import Index
+
+        class Duck:
+            class Meta:
+                index_together = [["bill", "tail"]]
+                indexes = []
+        """,
+        """\
+        from django.contrib.gis.db.models import Index
+
+        class Duck:
+            class Meta:
+                indexes = [Index(fields=["bill", "tail"])]
+        """,
+    )