From 5cf725a75b8ef2aa746caba5dc7153b703d65a57 Mon Sep 17 00:00:00 2001 From: ks123 Date: Wed, 11 Mar 2020 16:59:49 +0200 Subject: (Space Cog): Add `self.rovers` using to `.space mars` command check. --- bot/seasons/evergreen/space.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/bot/seasons/evergreen/space.py b/bot/seasons/evergreen/space.py index 5f629dc7..df15d7a4 100644 --- a/bot/seasons/evergreen/space.py +++ b/bot/seasons/evergreen/space.py @@ -164,8 +164,13 @@ class Space(Cog): """ # Check does user provided correct rover rover = rover.lower() - if rover not in ["curiosity", "opportunity", "spirit"]: - await ctx.send(f"Invalid rover `{rover}`. Rovers: `Curiosity`, `Opportunity`, `Spirit`") + if rover not in self.rovers: + await ctx.send( + ( + f"Invalid rover `{rover}`.\n" + f"**Rovers:** `{'`, `'.join(f'{r.capitalize()}' for r in self.rovers)}`" + ) + ) return # Create API request parameters, try to parse date -- cgit v1.2.3