feat: Handle queryset keyworkd in Prefetch object

Author: richardhapb

Cargo.lock

@@ -43,6 +43,8 @@ Runtime tools (django-silk, nplusone) are focuesd in runtime optimiezation. `dja
   * ManyToMany
   * Inheritance
   * Reverse relations (with `related_name` explicit or not)
+  * Complex chained relations like `model__relation__child_relation__depth`
+  * Prefetch usage: `Prefetch("relation", queryset=Relation.objects.select_related("child")`
 
 ## Installation
 
@@ -178,13 +180,11 @@ for user in users:
 
 * Interprocedural analysis requires type hints on QuerySet parameters
 * Limited understanding of:
-  * `Prefetch` objects with basic access support (e.g. `Prefetch("related_model")`)
   * `annotate`, `aggregate`
   * complex custom managers
 
 ## Roadmap
 
-* Complete Prefetch object support
 * Custom queryset method summaries
 * Templates integration
 

README.md

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

crates/django-check/Cargo.toml

@@ -34,12 +34,23 @@ impl QuerySetState {
     }
 }
 
+/// Information about a Prefetch object (e.g., `Prefetch("orders", queryset=...)`)
+#[derive(Debug, Clone, Eq, PartialEq)]
+pub struct PrefetchInfo {
+    /// The lookup string (e.g., "orders" or "orders__items")
+    pub lookup: String,
+    /// The queryset's state if a custom queryset was provided
+    pub queryset_state: Option<QuerySetState>,
+}
+
 #[derive(Debug, Clone, Eq, PartialEq)]
 pub enum DjangoSymbol {
     /// A `QuerySet` instance (e.g., `User.objects.all()`)
     QuerySet(QuerySetState),
     /// A Model instance (e.g., `user` inside a loop)
     ModelInstance(QuerySetState),
+    /// A Prefetch object (e.g., `Prefetch("orders", queryset=...)`)
+    Prefetch(PrefetchInfo),
 }
 
 #[derive(Debug, Clone, Eq, PartialEq)]

crates/django-check_semantic/src/ir/binding.rs

@@ -113,7 +113,7 @@ impl Parser {
             .parse_module(source)
             .map_err(|_| std::io::Error::new(std::io::ErrorKind::InvalidInput, "parsing module"))?;
 
-        let mut pass = ModelGraphPass::new(&filename, &source);
+        let mut pass = ModelGraphPass::new(filename, source);
         Ok(pass.run(parsed.syntax()))
     }