kdb-bot/src/modules/base/command/user_group.py

@@ -67,8 +67,6 @@ class UserGroup(DiscordCommandABC):
             'ontime': self._t.transform('modules.base.user.atr.ontime')
         }
 
-        self._atr_list = [(key, self._atr_dict[key]) for key in self._atr_dict]
-
     @commands.hybrid_group()
     @commands.guild_only()
     async def user(self, ctx: Context):
@@ -179,7 +177,8 @@ class UserGroup(DiscordCommandABC):
 
     @get.autocomplete('atr')
     async def get_autocomplete(self, interaction: discord.Interaction, current: str) -> List[app_commands.Choice[str]]:
-        return [app_commands.Choice(name=key, value=value) for key, value in self._atr_list]
+        atr_list = [self._atr_dict["xp"], self._atr_dict["ontime"]]
+        return [app_commands.Choice(name=atr, value=atr) for atr in atr_list]
 
     @user.command()
     @commands.guild_only()
@@ -218,8 +217,8 @@ class UserGroup(DiscordCommandABC):
 
     @set.autocomplete('atr')
     async def set_autocomplete(self, interaction: discord.Interaction, current: str) -> List[app_commands.Choice[str]]:
-        atr_list = [('xp', self._atr_dict['xp'])]
-        return [app_commands.Choice(name=key, value=value) for key, value in atr_list]
+        atr_list = [self._atr_dict["xp"]]
+        return [app_commands.Choice(name=atr, value=atr) for atr in atr_list]
 
     @user.command()
     @commands.guild_only()
@@ -255,4 +254,5 @@ class UserGroup(DiscordCommandABC):
 
     @remove.autocomplete('atr')
     async def remove_autocomplete(self, interaction: discord.Interaction, current: str) -> List[app_commands.Choice[str]]:
-        return [app_commands.Choice(name=value, value=key) for key, value in self._atr_list]
+        atr_list = [self._atr_dict["xp"], self._atr_dict["ontime"]]
+        return [app_commands.Choice(name=atr, value=atr) for atr in atr_list]