Considition-2020/game_state.py

from typing import List


class GameState:
    def __init__(self, map_values):
        self.game_id: str = map_values["gameId"]
        self.map_name: str = map_values["mapName"]
        self.max_turns: int = map_values["maxTurns"]
        self.max_temp: float = map_values["maxTemp"]
        self.min_temp: float = map_values["minTemp"]
        self.map: List[List[int]] = map_values["map"]
        self.energy_levels: List[EnergyLevel] = []
        for level in map_values["energyLevels"]:
            self.energy_levels.append(EnergyLevel(level))
        self.available_residence_buildings: List[BlueprintResidenceBuilding] = []
        for building in map_values["availableResidenceBuildings"]:
            self.available_residence_buildings.append(BlueprintResidenceBuilding(building))
        self.available_utility_buildings: List[BlueprintUtilityBuilding] = []
        for building in map_values["availableUtilityBuildings"]:
            self.available_utility_buildings.append(BlueprintUtilityBuilding(building))   
        self.available_upgrades: List[Upgrade] = []
        for upgrade in map_values['availableUpgrades']:
            self.available_upgrades.append(Upgrade(upgrade))
        self.effects: List[Effect] = []
        for effect in map_values['effects']:
            self.effects.append(Effect(effect))

        self.turn: int = 0
        self.funds: float = 0
        self.total_co2: float = 0
        self.total_happiness: float = 0
        self.current_temp: float = 0
        self.queue_happiness: float = 0
        self.housing_queue: int = 0
        self.residences: List[Residence] = []
        self.utilities: List[Utility] = []
        self.errors: List[str] = []
        self.messages: List[str] = []

    def update_state(self, state):
        self.turn = state['turn']
        self.funds = state['funds']
        self.total_co2 = state['totalCo2']
        self.total_happiness = state['totalHappiness']
        self.current_temp = state['currentTemp']
        self.queue_happiness = state['queueHappiness']
        self.housing_queue = state['housingQueue']
        self.residences = []
        for building in state['residenceBuildings']:
            self.residences.append(Residence(building))
        self.utilities = []
        for building in state['utilityBuildings']:
            self.utilities.append(Utility(building))
        self.errors = state['errors']
        self.messages = state['messages']


class EnergyLevel:
    def __init__(self, level_values):
        self.energy_threshold: int = level_values['energyThreshold']
        self.cost_per_mwh: float = level_values['costPerMwh']
        self.co2_per_mwh: float = level_values['tonCo2PerMwh']


class Blueprint:
    def __init__(self, blueprint):
        self.building_name: str = blueprint['buildingName']
        self.cost: int = blueprint['cost']
        self.co2_cost: int = blueprint['co2Cost']
        self.base_energy_need: float = blueprint['baseEnergyNeed']
        self.build_speed: int = blueprint['buildSpeed']
        self.type: str = blueprint['type']
        self.release_tick: int = blueprint['releaseTick']


class BlueprintUtilityBuilding(Blueprint):
    def __init__(self, blueprint_building):
        super().__init__(blueprint_building)
        self.effects: [str] = blueprint_building['effects']
        self.queue_increase: float = blueprint_building['queueIncrease']


class BlueprintResidenceBuilding(Blueprint):
    def __init__(self, blueprint_building):
        super().__init__(blueprint_building)
        self.max_pop: int = blueprint_building['maxPop']
        self.income_per_pop: float = blueprint_building['incomePerPop']
        self.emissivity: float = blueprint_building['emissivity']
        self.maintenance_cost: int = blueprint_building['maintenanceCost']
        self.decay_rate: float = blueprint_building['decayRate']
        self.max_happiness = blueprint_building['maxHappiness']


class Upgrade:
    def __init__(self, upgrade):
        self.name: str = upgrade['name']
        self.effect: str = upgrade['effect']
        self.cost: int = upgrade['cost']


class Effect:
    def __init__(self, effect):
        self.name: str = effect['name']
        self.radius: int = effect['radius']
        self.emissivity_multiplier: float = effect['emissivityMultiplier']
        self.decay_multiplier: float = effect['decayMultiplier']
        self.building_income_increase: float = effect['buildingIncomeIncrease']
        self.max_happiness_increase: float = effect['maxHappinessIncrease']
        self.mwh_production: float = effect['mwhProduction']
        self.base_energy_mwh_increase: float = effect['baseEnergyMwhIncrease']
        self.co2_per_pop_increase: float = effect['co2PerPopIncrease']
        self.decay_increase: float = effect['decayIncrease']


class Building:
    def __init__(self, building):
        self.building_name: str = building['buildingName']
        self.X: int = building['position']['x']
        self.Y: int = building['position']['y']
        self.effective_energy_in: float = building['effectiveEnergyIn']
        self.build_progress: int = building['buildProgress']
        self.can_be_demolished: bool = building['canBeDemolished']
        self.effects: List[str] = building['effects']


class Residence(Building):
    def __init__(self, residence):
        super().__init__(residence)
        self.current_pop: int = residence['currentPop']
        self.temperature: float = residence['temperature']
        self.requested_energy_in: float = residence['requestedEnergyIn']
        self.happiness_per_tick_per_pop: float = residence['happinessPerTickPerPop']
        self.health: int = residence['health']


class Utility(Building):
    def __init__(self, utility):
        super().__init__(utility)