From 8d8dc8a2f9c16b72cac2fc00c4418c635fc236b5 Mon Sep 17 00:00:00 2001 From: ErfanFaravani Date: Fri, 14 Feb 2020 13:31:06 +0330 Subject: [PATCH] get king by id added --- world.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/world.py b/world.py index a1ef7fe..0801fcc 100644 --- a/world.py +++ b/world.py @@ -24,7 +24,7 @@ def __init__(self, world=None, queue=None): self.player_friend = None self.player_first_enemy = None self.player_second_enemy = None - self.spells = None + self.spells = [] self.cast_spells = [] if world is not None: @@ -503,13 +503,13 @@ def get_damage_upgrade_number(self): # returns the spell given in that turn def get_received_spell(self): spell_id = self.turn_updates.received_spell - spell = self.get_spell_by_id(spell_id) + spell = self.get_spell_by_type_id(spell_id) return spell # returns the spell given in that turn to friend def get_friend_received_spell(self): spell_id = self.turn_updates.friend_received_spell - spell = self.get_spell_by_id(spell_id) + spell = self.get_spell_by_type_id(spell_id) return spell def upgrade_unit_range(self, unit=None, unit_id=None): @@ -546,10 +546,10 @@ def get_all_base_unit(self): def get_all_spells(self): return copy.deepcopy(self.spells) - def get_spell_by_id(self, spell_id): - for i in self.spells: - if spell_id == i.type_id: - return i + def get_king_by_id(self, player_id): + for p in self.players: + if p.player_id == player_id: + return p.king return None