Strategická: Načti protected_for_team
This commit is contained in:
parent
00e2a58991
commit
67cd40a3f0
1 changed files with 6 additions and 2 deletions
|
@ -54,12 +54,14 @@ class Field:
|
||||||
|
|
||||||
def __init__(self, i: int, j: int, hill: bool,
|
def __init__(self, i: int, j: int, hill: bool,
|
||||||
home_for_team: Optional[int],
|
home_for_team: Optional[int],
|
||||||
occupied_by_team: Optional[int]) -> None:
|
occupied_by_team: Optional[int],
|
||||||
|
protected_for_team: Optional[int],) -> None:
|
||||||
self.i = i
|
self.i = i
|
||||||
self.j = j
|
self.j = j
|
||||||
self.hill = hill
|
self.hill = hill
|
||||||
self.home_for_team = home_for_team
|
self.home_for_team = home_for_team
|
||||||
self.occupied_by_team = occupied_by_team
|
self.occupied_by_team = occupied_by_team
|
||||||
|
self.protected_for_team = protected_for_team
|
||||||
|
|
||||||
def __eq__(self, other: Field) -> bool:
|
def __eq__(self, other: Field) -> bool:
|
||||||
return (self.i, self.j) == (other.i, other.j)
|
return (self.i, self.j) == (other.i, other.j)
|
||||||
|
@ -93,7 +95,8 @@ class Field:
|
||||||
|
|
||||||
if self.hill:
|
if self.hill:
|
||||||
return False
|
return False
|
||||||
if self.home_for_team is not None and self.home_for_team != team:
|
if (self.protected_for_team is not None and
|
||||||
|
self.protected_for_team != team):
|
||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
@ -178,6 +181,7 @@ def parse_world(world: dict) -> List[List[Field]]:
|
||||||
field["hill"],
|
field["hill"],
|
||||||
field["home_for_team"],
|
field["home_for_team"],
|
||||||
field["occupied_by_team"],
|
field["occupied_by_team"],
|
||||||
|
field["protected_for_team"],
|
||||||
)
|
)
|
||||||
members = []
|
members = []
|
||||||
for member in field["members"]:
|
for member in field["members"]:
|
||||||
|
|
Loading…
Reference in a new issue