rc/aircox
13
1
Fork 1
Code Issues 15 Pull Requests Projects Releases Wiki Activity

save diffusion error

Browse Source
This commit is contained in:
bkfox 2022-10-05 01:13:45 +02:00
parent c1bf1c34f0
commit 4eef440a2a
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
aircox/models/episode.py
View File

@ -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):

2
aircox/models/program.py
View File

@ -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):