diff options
Diffstat (limited to 'bot.py')
-rw-r--r-- | bot.py | 20 |
1 files changed, 20 insertions, 0 deletions
@@ -19,6 +19,18 @@ class Cupido(commands.Bot): self.task = None self.reaction_map = {} self.vote_map = {} + self.read_file() + + def read_file(self): + try: + with open(config.TUTOR_FILE_PATH, 'r') as f: + self.tutors = {int(line) for line in f.read().split('\n')} + except FileNotFoundError: + self.tutors = set() + + def write_file(self): + with open(config.TUTOR_FILE_PATH, 'w') as f: + f.write('\n'.join(str(t) for t in self.tutors)) async def await_coroutine(self, co): if self.task is not None: @@ -160,6 +172,14 @@ class Cupido(commands.Bot): selected=vote)[0] return await message.edit(embed=embeds[0]) + def add_tutor(self, tutor): + self.tutors.add(tutor) + self.write_file() + + def remove_tutor(self, tutor): + self.tutors.remove(tutor) + self.write_file() + def main(): token = getenv(config.TOKEN_ENV_VAR) |