Add model metadata to class metadata

Summary: At minimum, we should know if our class is a django model, so we can add special field support for it.

Reviewed By: rchen152

Differential Revision: D83796198

fbshipit-source-id: e5e3807dd4a8fdeaae158d81c8c46ffc7fa6e990

Author: migeed-z

crates/pyrefly_python/src/module_name.rs

@@ -160,6 +160,10 @@ impl ModuleName {
         Self::from_str("django.db.models.enums")
     }
 
+    pub fn django_models() -> Self {
+        Self::from_str("django.db.models.base")
+    }
+
     /// The "unknown" module name, which corresponds to `__unknown__`.
     /// Used for files directly opened or passed on the command line which aren't on the search path.
     pub fn unknown() -> Self {

pyrefly/lib/alt/class/class_metadata.rs

@@ -178,6 +178,16 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
             }
         }
 
+        // TODO Zeina: This pattern is repeated a lot in this file. See if we can refactor it (BE).
+        let has_django_model = bases_with_metadata.iter().any(|(base_class_object, _)| {
+            base_class_object.has_toplevel_qname(ModuleName::django_models().as_str(), "Model")
+        });
+
+        let is_django_model = has_django_model
+            || bases_with_metadata
+                .iter()
+                .any(|(_, metadata)| metadata.is_django_model());
+
         // Compute various pieces of special metadata.
         let has_base_any = contains_base_class_any
             || bases_with_metadata
@@ -318,6 +328,7 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
             total_ordering_metadata,
             dataclass_transform_metadata,
             pydantic_model_kind,
+            is_django_model,
         )
     }
 

pyrefly/lib/alt/types/class_metadata.rs

@@ -58,6 +58,7 @@ pub struct ClassMetadata {
     /// that were passed to the `dataclass_transform` call.
     dataclass_transform_metadata: Option<DataclassTransformKeywords>,
     pydantic_model_kind: Option<PydanticModelKind>,
+    is_django_model: bool,
 }
 
 impl VisitMut<Type> for ClassMetadata {
@@ -91,6 +92,7 @@ impl ClassMetadata {
         total_ordering_metadata: Option<TotalOrderingMetadata>,
         dataclass_transform_metadata: Option<DataclassTransformKeywords>,
         pydantic_model_kind: Option<PydanticModelKind>,
+        is_django_model: bool,
     ) -> ClassMetadata {
         ClassMetadata {
             metaclass: Metaclass(metaclass),
@@ -109,6 +111,7 @@ impl ClassMetadata {
             total_ordering_metadata,
             dataclass_transform_metadata,
             pydantic_model_kind,
+            is_django_model,
         }
     }
 
@@ -130,6 +133,7 @@ impl ClassMetadata {
             total_ordering_metadata: None,
             dataclass_transform_metadata: None,
             pydantic_model_kind: None,
+            is_django_model: false,
         }
     }
 
@@ -157,6 +161,10 @@ impl ClassMetadata {
         self.pydantic_model_kind.is_some()
     }
 
+    pub fn is_django_model(&self) -> bool {
+        self.is_django_model
+    }
+
     pub fn pydantic_model_kind(&self) -> Option<PydanticModelKind> {
         self.pydantic_model_kind.clone()
     }