From 4eef440a2ad2dea387835faec85fde72931ec78b Mon Sep 17 00:00:00 2001
From: bkfox <thomas bkfox net>
Date: Wed, 5 Oct 2022 01:13:45 +0200
Subject: [PATCH] save diffusion error

---
 aircox/models/episode.py | 9 ++++++---
 aircox/models/program.py | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/aircox/models/episode.py b/aircox/models/episode.py
index 87721c7..0d8d956 100644
--- a/aircox/models/episode.py
+++ b/aircox/models/episode.py
@@ -208,6 +208,11 @@ class Diffusion(BaseRerun):
             str_ += ' ({})'.format(_('rerun'))
         return str_
 
+    def save(self, *args, **kwargs):
+        super().save(*args, **kwargs)
+        if self.is_initial and self.episode != self._initial['episode']:
+            self.rerun_set.update(episode=self.episode, program=self.program)
+
     #def save(self, no_check=False, *args, **kwargs):
         #if self.start != self._initial['start'] or \
         #        self.end != self._initial['end']:
@@ -217,10 +222,8 @@ class Diffusion(BaseRerun):
         self.episode = self.initial.episode
         self.program = self.episode.program
 
-    def save_initial(self):
+    def save_initial(self, *args, **kwargs):
         self.program = self.episode.program
-        if self.episode != self._initial['episode']:
-            self.rerun_set.update(episode=self.episode, program=self.program)
 
     @property
     def duration(self):
diff --git a/aircox/models/program.py b/aircox/models/program.py
index 0efc2e5..e39c1f5 100644
--- a/aircox/models/program.py
+++ b/aircox/models/program.py
@@ -198,7 +198,7 @@ class BaseRerun(models.Model):
         if self.is_rerun:
             self.save_rerun()
         else:
-            self.save_initial()
+            self.save_initial(*args, **kwargs)
         super().save(*args, **kwargs)
 
     def save_rerun(self):