From 9a01764812df1ce706289390b32df98a2cca4c26 Mon Sep 17 00:00:00 2001 From: Xevion Date: Mon, 20 Jan 2020 16:35:58 -0600 Subject: [PATCH] add rebuild parameter, fix multi printing --- app/models.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/app/models.py b/app/models.py index 921b389..00ad675 100644 --- a/app/models.py +++ b/app/models.py @@ -15,10 +15,10 @@ class Season(db.Model): assert 0 <= kwargs.get('id') <= 9, "Season ID must be 0-9 inclusive" super(Season, self).__init__(**kwargs) - def build(self): + def build(self, rebuild=False): """runs build operations on every Episode under this season""" + print(f'Running build() on Season {self.id}') for episode in range(1, episodes[self.id - 1] + 1): - print(f'Running build() on Season {self.id}') ep = Episode.query.filter_by(season_id=self.id, number=episode).first() if ep is None: # Add the episode, then build @@ -29,7 +29,8 @@ class Season(db.Model): db.session.commit() ep.build() else: - # Regardless of whether it existended before hand, the episode will be built. + if rebuild: + ep.build() pass @staticmethod