forked from rc/aircox
save diffusion error
This commit is contained in:
parent
c1bf1c34f0
commit
4eef440a2a
|
@ -208,6 +208,11 @@ class Diffusion(BaseRerun):
|
||||||
str_ += ' ({})'.format(_('rerun'))
|
str_ += ' ({})'.format(_('rerun'))
|
||||||
return str_
|
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):
|
#def save(self, no_check=False, *args, **kwargs):
|
||||||
#if self.start != self._initial['start'] or \
|
#if self.start != self._initial['start'] or \
|
||||||
# self.end != self._initial['end']:
|
# self.end != self._initial['end']:
|
||||||
|
@ -217,10 +222,8 @@ class Diffusion(BaseRerun):
|
||||||
self.episode = self.initial.episode
|
self.episode = self.initial.episode
|
||||||
self.program = self.episode.program
|
self.program = self.episode.program
|
||||||
|
|
||||||
def save_initial(self):
|
def save_initial(self, *args, **kwargs):
|
||||||
self.program = self.episode.program
|
self.program = self.episode.program
|
||||||
if self.episode != self._initial['episode']:
|
|
||||||
self.rerun_set.update(episode=self.episode, program=self.program)
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def duration(self):
|
def duration(self):
|
||||||
|
|
|
@ -198,7 +198,7 @@ class BaseRerun(models.Model):
|
||||||
if self.is_rerun:
|
if self.is_rerun:
|
||||||
self.save_rerun()
|
self.save_rerun()
|
||||||
else:
|
else:
|
||||||
self.save_initial()
|
self.save_initial(*args, **kwargs)
|
||||||
super().save(*args, **kwargs)
|
super().save(*args, **kwargs)
|
||||||
|
|
||||||
def save_rerun(self):
|
def save_rerun(self):
|
||||||
|
|
Loading…
Reference in New Issue
Block a user