This repository has been archived on 2023-12-31. You can view files and clone it, but cannot push or open issues or pull requests.
Considition-2020/game_layer.py

156 lines
5.9 KiB
Python
Raw Normal View History

2020-09-27 13:46:19 +02:00
from typing import Tuple
import api
from game_state import GameState
class GameLayer:
def __init__(self, api_key):
2020-09-27 14:45:23 +02:00
self.game_state = GameState
self.api_key = api_key
2020-09-27 13:46:19 +02:00
def new_game(self, map_name: str = "training0"):
"""
Create a new game.
"""
if map_name:
game_options = {"mapName": map_name}
else:
game_options = ""
self.game_state = GameState(api.new_game(self.api_key, game_options))
def end_game(self):
"""
End the current game
"""
api.end_game(self.api_key, self.game_state.game_id)
def start_game(self):
"""
Starts the game.
"""
self.game_state.update_state(api.start_game(self.api_key, self.game_state.game_id))
def place_foundation(self, pos: Tuple[int, int], building_name: str):
"""
Places a building with name building_name at the given position.
:param pos: (int, int) - the position
:param building_name: string - the name, check available_residence_buildings or available_residence_utilities for which buildings are available
"""
position = {'X': pos[0], 'Y': pos[1]}
foundation = {'Position': position, 'BuildingName': building_name}
self.game_state.update_state(api.place_foundation(self.api_key, foundation, self.game_state.game_id))
def build(self, pos: Tuple[int, int]):
"""
Continues the construction of a building at the given position.
:param pos: (int, int) - the position
"""
position = {'position': {"X": pos[0], "Y": pos[1]}}
self.game_state.update_state(api.build(self.api_key, position, self.game_state.game_id))
def maintenance(self, pos: Tuple[int, int]):
"""
Performs maintenance on the building at the given position.
:param pos: (int, int) - the position
"""
position = {'position': {"x": pos[0], "y": pos[1]}}
self.game_state.update_state(api.maintenance(self.api_key, position, self.game_state.game_id))
def demolish(self, pos: Tuple[int, int]):
"""
Demolishes the building at the given position.
:param pos: (int, int) - the position
"""
position = {'position': {"x": pos[0], "y": pos[1]}}
self.game_state.update_state(api.demolish(self.api_key, position, self.game_state.game_id))
def adjust_energy_level(self, pos: Tuple[int, int], value: float):
"""
Adjusts the requested energy to value on the building at the given position.
:param pos: (int, int) - the position
:param value: float - the new requested value
"""
position = {"x": pos[0], "y": pos[1]}
self.game_state.update_state(api.adjust_energy(self.api_key, {"position": position, "value": value}, self.game_state.game_id))
def wait(self):
"""
Advances the game by one turn.
"""
self.game_state.update_state(api.wait(self.api_key, self.game_state.game_id))
def buy_upgrade(self, pos: Tuple[int, int], upgrade: str):
"""
Adds the specified upgrade to the building at the given position.
You can find the available upgrades in available_upgrades
Parameters
----------
:param pos: (int, int) - the position
:param upgrade: string - the upgrade to purchase
"""
position = {"x": pos[0], "y": pos[1]}
self.game_state.update_state(api.buy_upgrades(self.api_key, {"position": position, "upgradeAction": upgrade}, self.game_state.game_id))
def get_score(self):
"""
Gets the score for the game.
:return An object with partial and total scores.
"""
return api.get_score(self.api_key, self.game_state.game_id)
def get_game_info(self, game_id: str):
"""
Gets the game info of an already ongoing game and updates the state.
:param game_id: string - the id of the game to get info about.
"""
self.game_state = GameState(api.get_game_info(self.api_key, game_id))
def get_game_state(self, game_id: str):
"""
Gets the game state of an already ongoing game and updates the state. Can be used to resume a game.
:param game_id: string - the id of the game to get the state.
"""
self.game_state.update_state(api.get_game_state(self.api_key, game_id))
def get_blueprint(self, building_name: str):
"""
Returns the matching blueprint for a building
:param building_name: string - the name of the building to get a blueprint.
"""
res_blueprint = self.get_residence_blueprint(building_name)
if res_blueprint:
return res_blueprint
return self.get_utility_blueprint(building_name)
def get_residence_blueprint(self, building_name: str):
"""
Returns the matching blueprint for a residence
:param building_name: string - the name of the building to get a blueprint.
"""
for blueprint in self.game_state.available_residence_buildings:
if blueprint.building_name == building_name:
return blueprint
return None
def get_utility_blueprint(self, building_name: str):
"""
Return the matching blueprint for a utility building
:param building_name: string - the name of the building to get a blueprint.
"""
for blueprint in self.game_state.available_utility_buildings:
if blueprint.building_name == building_name:
return blueprint
return None
def get_effect(self, effect_name: str):
"""
Return the matching effect for an effect name.
:param effect_name: string - the name of the effect to get.
"""
for effect in self.game_state.effects:
if effect.name == effect_name:
return effect
return None