Browse Source

Strategická: Lepší enum akcí

master
David Klement 1 year ago
parent
commit
1808d89e14
  1. 21
      klient/play.py

21
klient/play.py

@ -1,18 +1,18 @@
#!/usr/bin/env python3
from __future__ import annotations
from enum import Enum
from typing import Callable, Iterable, Optional, List, Set
import collections
import enum
import json
import sys
class Action(Enum):
UP = "up"
DOWN = "down"
LEFT = "left"
RIGHT = "right"
STAY = None
class Action(enum.Enum):
UP = enum.auto()
DOWN = enum.auto()
LEFT = enum.auto()
RIGHT = enum.auto()
STAY = enum.auto()
def invert(self) -> Action:
if self == Action.UP:
@ -25,6 +25,11 @@ class Action(Enum):
return Action.LEFT
return Action.STAY
def to_json(self) -> Optional[str]:
if self == Action.STAY:
return None
return self.name.lower()
class State:
def __init__(self, state: dict) -> None:
@ -157,7 +162,7 @@ def pathfind(member: Member, goal: Field) -> Action:
def build_turn(members: Iterable[Member]) -> dict:
return {
"members": [
{"id": member.id, "action": member.action}
{"id": member.id, "action": member.action.to_json()}
for member in members
]
}

Loading…
Cancel
Save