shortdeck1.1
This commit is contained in:
@@ -12,13 +12,13 @@ BASE_PATH = "/get_game_state"
|
||||
APPLY_PATH = "/apply_action"
|
||||
|
||||
|
||||
def fetch_game_state(base_url: str, player_id: int) -> Dict[str, Any]:
|
||||
def fetch_game_state(base_url, player_id) -> Dict[str, Any]:
|
||||
resp = requests.get(f"{base_url.rstrip('/')}{BASE_PATH}", params={"player_id": player_id}, timeout=5)
|
||||
resp.raise_for_status()
|
||||
return resp.json()
|
||||
|
||||
|
||||
def post_action(base_url: str, player_id: int, action: str, amount: Optional[int]):
|
||||
def post_action(base_url, player_id, action, amount):
|
||||
payload = {"player_id": player_id, "action": action}
|
||||
if amount is not None:
|
||||
payload["amount"] = amount
|
||||
@@ -27,7 +27,7 @@ def post_action(base_url: str, player_id: int, action: str, amount: Optional[int
|
||||
return resp.json()
|
||||
|
||||
|
||||
def choose_random_action(info: Dict[str, Any]) -> Optional[tuple[str, Optional[int]]]:
|
||||
def choose_random_action(info) -> Optional[tuple[str, Optional[int]]]:
|
||||
actions = info.get("actions")
|
||||
if not actions:
|
||||
return None
|
||||
@@ -48,7 +48,7 @@ def choose_random_action(info: Dict[str, Any]) -> Optional[tuple[str, Optional[i
|
||||
return random.choice(choices)
|
||||
|
||||
|
||||
def run_loop(base_url: str, player_id: int, interval: float = 2.0, seed: Optional[int] = None):
|
||||
def run_loop(base_url, player_id, interval, seed):
|
||||
if seed is not None:
|
||||
random.seed(seed)
|
||||
|
||||
@@ -77,7 +77,7 @@ def run_loop(base_url: str, player_id: int, interval: float = 2.0, seed: Optiona
|
||||
time.sleep(interval)
|
||||
|
||||
|
||||
def main(argv: list[str] | None = None) -> int:
|
||||
def main(argv) -> int:
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--server", default="http://localhost:8000")
|
||||
parser.add_argument("--player_id", type=int, default=1)
|
||||
|
||||
Reference in New Issue
Block a user