fix nested querying

Author: timgraham

django_mongodb/fields/embedded_model.py

@@ -141,7 +141,8 @@ def preprocess_lhs(self, compiler, connection):
 
 def key_transform(self, compiler, connection):
     mql, key_transforms = self.preprocess_lhs(compiler, connection)
-    return f"{mql}.{key_transforms[0]}"
+    transforms = ".".join(key_transforms)
+    return f"{mql}.{transforms}"
 
 
 class KeyTransformFactory:

tests/model_fields_/models.py

@@ -43,3 +43,19 @@ class EmbeddedModel(models.Model):
     someint = models.IntegerField(db_column="custom_column")
     auto_now = models.DateTimeField(auto_now=True)
     auto_now_add = models.DateTimeField(auto_now_add=True)
+
+
+class Address(models.Model):
+    city = models.CharField(max_length=20)
+    state = models.CharField(max_length=2)
+
+
+class Author(models.Model):
+    name = models.CharField(max_length=10)
+    age = models.IntegerField()
+    address = EmbeddedModelField(Address)
+
+
+class Book(models.Model):
+    name = models.CharField(max_length=100)
+    author = EmbeddedModelField(Author)

tests/model_fields_/test_embedded_model.py

@@ -6,6 +6,9 @@
 from django_mongodb.fields import EmbeddedModelField
 
 from .models import (
+    Address,
+    Author,
+    Book,
     DecimalKey,
     DecimalParent,
     EmbeddedModel,
@@ -125,3 +128,9 @@ def test_gte(self):
         self.assertCountEqual(
             EmbeddedModelFieldModel.objects.filter(simple__someint__gte=3), self.objs[3:]
         )
+
+    def test_nested(self):
+        obj = Book.objects.create(
+            author=Author(name="Shakespeare", age=55, address=Address(city="NYC", state="NY"))
+        )
+        self.assertCountEqual(Book.objects.filter(author__address__city="NYC"), [obj])