Ticket #14163: inheritance_pk.diff

django/db/models/base.py

@@ -399 +399 @@
         return getattr(self, meta.pk.attname)
 
     def _set_pk_val(self, value):
-        return setattr(self, self._meta.pk.attname, value)
+        ret = setattr(self, self._meta.pk.attname, value)
+        parent_link = self._meta.pk
+        while (isinstance(parent_link, OneToOneField) and parent_link.rel.parent_link):
+            parent_link = parent_link.rel.get_related_field()
+            setattr(self, parent_link.attname, value)
+        return ret
 
     pk = property(_get_pk_val, _set_pk_val)
 

tests/modeltests/model_inheritance/models.py

@@ -119 +119 @@
     def __unicode__(self):
         return u"%s the parking lot" % self.name
 
+class CarInLot(models.Model):
+    parking_lot = models.ForeignKey(ParkingLot)
+
 #
 # Abstract base classes with related models where the sub-class has the
 # same name in a different app and inherits from the same abstract base
@@ -345 +348 @@
 >>> Restaurant.objects.get(lot__name='Well Lit')
 <Restaurant: Ristorante Miron the restaurant>
 
+# Setting the pk of a model with OneToOneField, where parent_link=True, will
+# also set the parent model's pk.
+
+>>> ir2 = ItalianRestaurant(pk=1)
+>>> ir2.id == ir2.pk
+True
+>>> ir2.pk = 2
+>>> ir2.pk == ir2.id
+True
+
+>>> park3 = ParkingLot(pk=1)
+>>> park3.pk == park3.id
+True
+>>> park3.parent_id == park3.pk
+True
+>>> cil = CarInLot(parking_lot=park3)
+>>> cil.parking_lot_id == 1
+True
+
 # The update() command can update fields in parent and child classes at once
 # (although it executed multiple SQL queries to do so).
 >>> Restaurant.objects.filter(serves_hot_dogs=True, name__contains='D').update(name='Demon Puppies', serves_hot_dogs=False)