Add character add checker

This commit is contained in:
Tim Van Baak 2020-04-24 00:53:39 -07:00
parent 7d633f5201
commit 081a54c9c3
1 changed files with 30 additions and 1 deletions

View File

@ -63,6 +63,35 @@ def add_player_to_lexicon(
lexicon.log('Player "{0.cfg.username}" joined ({0.uid})'.format(player)) lexicon.log('Player "{0.cfg.username}" joined ({0.uid})'.format(player))
def player_can_create_character(
player: UserModel,
lexicon: LexiconModel,
name: str) -> bool:
"""
Checks whether a player can create a character with the given name
"""
# Trivial failures
if not player or not lexicon or not name:
return False
# User needs to be a player
if player.uid not in lexicon.cfg.join.joined:
return False
# Character can't be a dupe
if any([
char.name for char in lexicon.cfg.character.values()
if char.name == name]):
return False
# Player can't add more characters than the limit
if len([
char for char in lexicon.cfg.character.values()
if char.player == player.uid]) > lexicon.cfg.join.chars_per_player:
return False
# Players can't add characters after the game has started
if lexicon.cfg.turn.current:
return False
return True
def create_character_in_lexicon( def create_character_in_lexicon(
player: UserModel, player: UserModel,
lexicon: LexiconModel, lexicon: LexiconModel,
@ -77,7 +106,7 @@ def create_character_in_lexicon(
raise ArgumentError(f'Invalid player: {player}') raise ArgumentError(f'Invalid player: {player}')
if player.uid not in lexicon.cfg.join.joined: if player.uid not in lexicon.cfg.join.joined:
raise ArgumentError(f'Player {player} not in lexicon {lexicon}') raise ArgumentError(f'Player {player} not in lexicon {lexicon}')
if name is None: if not name:
raise ArgumentError(f'Invalid character name: "{name}"') raise ArgumentError(f'Invalid character name: "{name}"')
if any([ if any([
char.name for char in lexicon.cfg.character.values() char.name for char in lexicon.cfg.character.values()