Can flag TweetItems and PhotoItems as deleted

seantc · seantc · commit 3b71d611fc78 · 2016-04-07T18:51:51.000Z
diff --git a/testsite/items/models.py b/testsite/items/models.py
@@ -12,7 +12,9 @@ class Meta:
 
 class PhotoItem(ItemAbstract):
     image = models.ImageField()
+    deleted = models.BooleanField(default=False)
 
 
 class TweetItem(ItemAbstract):
     text = models.CharField(max_length=150)
+    deleted = models.BooleanField(default=False)
diff --git a/testsite/stream/models.py b/testsite/stream/models.py
@@ -5,5 +5,11 @@
 class Stream(models.Model):
     user = models.ForeignKey(User)
     created_at = models.DateTimeField()
-    photo_item = models.ForeignKey(PhotoItem, blank=True, null=True)
-    tweet_item = models.ForeignKey(TweetItem, blank=True, null=True)
+    photo_item = models.ForeignKey(PhotoItem,
+        on_delete=models.SET_NULL,
+        blank=True,
+        null=True)
+    tweet_item = models.ForeignKey(TweetItem,
+        on_delete=models.SET_NULL,
+        blank=True,
+        null=True)
