From bc27dc4621a86c204430b81d86a50c1a9402fd78 Mon Sep 17 00:00:00 2001 From: MetalStormGhost Date: Sun, 17 Sep 2023 16:06:17 +0300 Subject: [PATCH] Cleaned up some code. --- game/pretense/pretenseluagenerator.py | 2 +- game/pretense/pretensetgogenerator.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/game/pretense/pretenseluagenerator.py b/game/pretense/pretenseluagenerator.py index 7c5dc964..b10f6ab6 100644 --- a/game/pretense/pretenseluagenerator.py +++ b/game/pretense/pretenseluagenerator.py @@ -56,7 +56,7 @@ class PretenseLuaGenerator(LuaGenerator): self.mission.triggerrules.triggers.remove(t) self.mission.triggerrules.triggers.append(t) - def generate_pretense_land_upgrade_supply(self, cp_name: str, cp_side: int): + def generate_pretense_land_upgrade_supply(self, cp_name: str, cp_side: int) -> str: lua_string_zones = "" cp_name_trimmed = "".join([i for i in cp_name.lower() if i.isalnum()]) cp_side_str = "blue" if cp_side == PRETENSE_BLUE_SIDE else "red" diff --git a/game/pretense/pretensetgogenerator.py b/game/pretense/pretensetgogenerator.py index bcb9a0c2..74f99a6e 100644 --- a/game/pretense/pretensetgogenerator.py +++ b/game/pretense/pretensetgogenerator.py @@ -450,8 +450,8 @@ class PretenseTgoGenerator(TgoGenerator): else: other_coalition = coalition country = self.m.country(other_coalition.faction.country.name) + new_ground_object: TheaterGroundObject for ground_object in cp.ground_objects: - generator: GroundObjectGenerator if isinstance(ground_object, BuildingGroundObject): new_ground_object = BuildingGroundObject( name=ground_object.name,