diff --git a/rowers/models.py b/rowers/models.py index 666671ef..02fb531f 100644 --- a/rowers/models.py +++ b/rowers/models.py @@ -1418,10 +1418,12 @@ class TrainingPlan(models.Model): def save(self, *args, **kwargs): manager = self.manager + if manager.rowerplan in ['basic','pro']: - raise ValidationError( - "Basic user cannot have a training plan" - ) + if manager.plantrialexpires < timezone.now().date(): + raise ValidationError( + "Basic user cannot have a training plan" + ) if self.enddate < self.startdate: startdate = self.startdate