From 58f9262e59baa51989e4ac6620628d4aae2e8e6f Mon Sep 17 00:00:00 2001 From: bkfox Date: Wed, 20 May 2020 12:16:03 +0200 Subject: [PATCH] fix issue #11 --- aircox/models/episode.py | 2 -- aircox/models/page.py | 3 +++ 2 files changed, 3 insertions(+), 2 deletions(-) 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):