Game: renamed started_only decorator to game_started

This commit is contained in:
Elnath 2021-06-10 23:50:38 +02:00
parent 19e2811823
commit f6d8bcb955
1 changed files with 19 additions and 19 deletions

View File

@ -8,7 +8,7 @@ import discord
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def started_only(func): def game_started(func):
""" """
Decorator for *methods* of Game that need the game to be started. Decorator for *methods* of Game that need the game to be started.
""" """
@ -55,65 +55,65 @@ class Game:
def is_started(self): def is_started(self):
return self.config["game_started"] return self.config["game_started"]
@started_only @game_started
def get_players_id(self) -> List[int]: def get_players_id(self) -> List[int]:
return self.config["players"] return self.config["players"]
@started_only @game_started
def get_players(self) -> List[discord.Member]: def get_players(self) -> List[discord.Member]:
return [self.guild.get_member(player) for player in self.get_players_id()] return [self.guild.get_member(player) for player in self.get_players_id()]
@started_only @game_started
def get_player_info(self, player: Union[int, discord.Member]): def get_player_info(self, player: Union[int, discord.Member]):
if isinstance(player, discord.Member): if isinstance(player, discord.Member):
player = player.id player = player.id
return self.config["player_info"][str(player)] return self.config["player_info"][str(player)]
@started_only @game_started
def get_gm_role_id(self) -> int: def get_gm_role_id(self) -> int:
return self.config["gm_role"] return self.config["gm_role"]
@started_only @game_started
def get_gm_role(self) -> discord.Role: def get_gm_role(self) -> discord.Role:
return self.guild.get_role(self.get_gm_role_id()) return self.guild.get_role(self.get_gm_role_id())
@started_only @game_started
def is_gm(self, user: discord.Member) -> bool: def is_gm(self, user: discord.Member) -> bool:
return self.get_gm_role() in user.roles return self.get_gm_role() in user.roles
@started_only @game_started
def get_player_role_id(self) -> int: def get_player_role_id(self) -> int:
return self.config["player_role"] return self.config["player_role"]
@started_only @game_started
def get_player_role(self) -> discord.Role: def get_player_role(self) -> discord.Role:
return self.guild.get_role(self.get_player_role_id()) return self.guild.get_role(self.get_player_role_id())
@started_only @game_started
def get_game_category_id(self) -> int: def get_game_category_id(self) -> int:
return self.config["category"] return self.config["category"]
@started_only @game_started
def get_game_category(self) -> discord.CategoryChannel: def get_game_category(self) -> discord.CategoryChannel:
return self.guild.get_channel(self.get_game_category_id()) return self.guild.get_channel(self.get_game_category_id())
@started_only @game_started
def get_announcements_channel_id(self) -> int: def get_announcements_channel_id(self) -> int:
return self.config["announce_chan"] return self.config["announce_chan"]
@started_only @game_started
def get_announcements_channel(self) -> discord.TextChannel: def get_announcements_channel(self) -> discord.TextChannel:
return self.guild.get_channel(self.get_announcements_channel_id()) return self.guild.get_channel(self.get_announcements_channel_id())
@started_only @game_started
def get_votes_channel_id(self) -> int: def get_votes_channel_id(self) -> int:
return self.config["votes_chan"] return self.config["votes_chan"]
@started_only @game_started
def get_votes_channel(self) -> discord.TextChannel: def get_votes_channel(self) -> discord.TextChannel:
return self.guild.get_channel(self.get_votes_channel_id()) return self.guild.get_channel(self.get_votes_channel_id())
@started_only @game_started
def is_vote_running(self) -> bool: def is_vote_running(self) -> bool:
return self.config["vote"] is not None return self.config["vote"] is not None
@ -124,13 +124,13 @@ class Game:
vote_count[self.config["vote"][str(player_id)]] += 1 vote_count[self.config["vote"][str(player_id)]] += 1
return vote_count[True] > vote_count[False] return vote_count[True] > vote_count[False]
@started_only @game_started
def get_player_channel_id(self, player: Union[int, discord.Member]) -> int: def get_player_channel_id(self, player: Union[int, discord.Member]) -> int:
if isinstance(player, discord.Member): if isinstance(player, discord.Member):
player = player.id player = player.id
return self.config["player_info"][str(player)]["channel"] return self.config["player_info"][str(player)]["channel"]
@started_only @game_started
def get_player_channel(self, player: Union[int, discord.Member]) -> discord.TextChannel: def get_player_channel(self, player: Union[int, discord.Member]) -> discord.TextChannel:
return self.guild.get_channel(self.get_player_channel_id(player)) return self.guild.get_channel(self.get_player_channel_id(player))
@ -184,7 +184,7 @@ class Game:
self.config["game_started"] = True self.config["game_started"] = True
self.save_function() self.save_function()
@started_only @game_started
async def delete(self): async def delete(self):
category = self.get_game_category() category = self.get_game_category()
for channel in category.channels: for channel in category.channels: