diff --git a/aircox/models/episode.py b/aircox/models/episode.py index 2c54a2a..4225e64 100644 --- a/aircox/models/episode.py +++ b/aircox/models/episode.py @@ -39,8 +39,6 @@ class Episode(Page): return super().get_absolute_url() def save(self, *args, **kwargs): - if self.cover is None: - self.cover = self.program.cover if self.parent is None: raise ValueError('missing parent program') super().save(*args, **kwargs) diff --git a/aircox/models/page.py b/aircox/models/page.py index c0f48a8..c41434e 100644 --- a/aircox/models/page.py +++ b/aircox/models/page.py @@ -108,6 +108,9 @@ class Page(models.Model): self.pub_date = tz.now() elif not self.is_published: self.pub_date = None + + if not self.cover and self.parent: + self.cover = self.parent.cover super().save(*args, **kwargs) def get_absolute_url(self):