Simplify and rename EmbeddedModelField's transform classes

Author: timgraham

django_mongodb_backend/fields/embedded_model.py

@@ -132,7 +132,7 @@ def get_transform(self, name):
         if transform:
             return transform
         field = self.embedded_model._meta.get_field(name)
-        return KeyTransformFactory(name, field)
+        return EmbeddedModelTransformFactory(field)
 
     def validate(self, value, model_instance):
         super().validate(value, model_instance)
@@ -156,10 +156,9 @@ def formfield(self, **kwargs):
         )
 
 
-class KeyTransform(Transform):
-    def __init__(self, key_name, ref_field, *args, **kwargs):
+class EmbeddedModelTransform(Transform):
+    def __init__(self, ref_field, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.key_name = str(key_name)
         self.ref_field = ref_field
 
     def get_lookup(self, name):
@@ -187,8 +186,8 @@ def get_transform(self, name):
     def as_mql(self, compiler, connection, as_path=False):
         previous = self
         columns = []
-        while isinstance(previous, KeyTransform):
-            columns.insert(0, previous.ref_field.column)
+        while isinstance(previous, EmbeddedModelTransform):
+            columns.insert(0, previous.field.column)
             previous = previous.lhs
         if as_path:
             mql = previous.as_mql(compiler, connection, as_path=True)
@@ -204,10 +203,9 @@ def output_field(self):
         return self.ref_field
 
 
-class KeyTransformFactory:
-    def __init__(self, key_name, ref_field):
-        self.key_name = key_name
+class EmbeddedModelTransformFactory:
+    def __init__(self, ref_field):
         self.ref_field = ref_field
 
     def __call__(self, *args, **kwargs):
-        return KeyTransform(self.key_name, self.ref_field, *args, **kwargs)
+        return EmbeddedModelTransform(self.ref_field, *args, **kwargs)

django_mongodb_backend/fields/polymorphic_embedded_model.py

@@ -5,7 +5,7 @@
 from django.db import models
 from django.db.models.fields.related import lazy_related_operation
 
-from .embedded_model import KeyTransformFactory
+from .embedded_model import EmbeddedModelTransformFactory
 from .utils import get_mongodb_connection
 
 
@@ -170,7 +170,7 @@ def get_transform(self, name):
             raise FieldDoesNotExist(
                 f"The models of field '{self.name}' have no field named '{name}'."
             )
-        return KeyTransformFactory(name, field)
+        return EmbeddedModelTransformFactory(field)
 
     def validate(self, value, model_instance):
         super().validate(value, model_instance)