From d9c55539074d2ea20e21b1a2d23a587a5486a49a Mon Sep 17 00:00:00 2001 From: Milo Weinberg Date: Sat, 22 Jun 2024 19:07:09 -0400 Subject: [PATCH] Handle more reward item types --- bot/cogs/core/quests.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/bot/cogs/core/quests.py b/bot/cogs/core/quests.py index 5e6fc610..6c61b61e 100644 --- a/bot/cogs/core/quests.py +++ b/bot/cogs/core/quests.py @@ -201,9 +201,13 @@ async def quest_completed( await self.db.mark_daily_quest_as_done(user_id, quest.key) quest = await self.fetch_user_daily_quest(user_id) - # TODO: Support more types of rewards (quest.reward_item) assert quest.reward_item == "emerald" - await self.db.balance_add(user_id, quest.reward_amount) + if quest.reward_amount == "emerald": + await self.db.balance_add(user_id, quest.reward_amount) + elif quest.reward_item == "Barrel": + await self.db.add_item(user_id, "Barrel", 1024, quest.reward_amount) + else: + raise NotImplementedError(f"Couldn't reward item {quest.reward_item} to user {user_id}") embed = self.get_quest_embed(loc, quest) view = DailyQuestDoneView(bot=self.bot, loc=loc, user_id=user_id)