feat: handle Subscripts (slice in queries)

Author: richardhapb

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "django-check"
-version = "0.1.2"
+version = "0.1.3"
 rust-version = { workspace = true }
 license = { workspace = true }
 authors = { workspace = true }

crates/django-check/Cargo.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "djch"
-version = "0.1.2"
+version = "0.1.3"
 description = "Static N+1 query detector for Django ORM – fast Rust-powered checker"
 # For PyPI long description (shows formatted on project page)
 readme = {file = "README.md", content-type = "text/markdown"}

crates/django-check/pyproject.toml

@@ -83,6 +83,8 @@ impl<'a> QuerySetResolver<'a> {
             Expr::Call(call) => self.resolve_call(call),
             Expr::Attribute(attr) => self.resolve_attribute(attr),
             Expr::Name(name) => self.symbols.lookup(&name.id).cloned(),
+            // Handle slice like myqueryset[0:100]
+            Expr::Subscript(subscript) => self.resolve(&subscript.value),
             _ => None,
         }
     }
@@ -1510,4 +1512,41 @@ for t in tier1s:
         let diags = run_pass(source);
         assert_eq!(diags.len(), 1);
     }
+
+    #[test]
+    fn detect_query_using_subscript() {
+        let source = r#"
+class User(Model):
+     name = models.CharField()
+
+class Payment(Model):
+    user = models.ForeignKey(User, related_name="payments")
+    amount = models.FloatField()
+
+payments = Payment.objects.all().order_by("-amount")
+top_payments = payments[0:10]
+for payment in top_payments:
+    print(payment.user.name)
+        "#;
+        let diags = run_pass(source);
+        assert_eq!(diags.len(), 1);
+    }
+
+    #[test]
+    fn detect_query_using_subscript_in_for_stmt() {
+        let source = r#"
+class User(Model):
+     name = models.CharField()
+
+class Payment(Model):
+    user = models.ForeignKey(User, related_name="payments")
+    amount = models.FloatField()
+
+payments = Payment.objects.all().order_by("-amount")
+for payment in payments[0:10]:
+    print(payment.user.name)
+        "#;
+        let diags = run_pass(source);
+        assert_eq!(diags.len(), 1);
+    }
 }