mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import logging
|
|
import random
|
|
from typing import Optional
|
|
|
|
from dcs.unitgroup import VehicleGroup
|
|
|
|
from game import db, Game
|
|
from game.theater.theatergroundobject import MissileSiteGroundObject
|
|
from gen.missiles.scud_site import ScudGenerator
|
|
from gen.missiles.v1_group import V1GroupGenerator
|
|
|
|
MISSILES_MAP = {"V1GroupGenerator": V1GroupGenerator, "ScudGenerator": ScudGenerator}
|
|
|
|
|
|
def generate_missile_group(
|
|
game: Game, ground_object: MissileSiteGroundObject, faction_name: str
|
|
) -> Optional[VehicleGroup]:
|
|
"""
|
|
This generate a missiles group
|
|
:return: Nothing, but put the group reference inside the ground object
|
|
"""
|
|
faction = db.FACTIONS[faction_name]
|
|
if len(faction.missiles) > 0:
|
|
generators = faction.missiles
|
|
if len(generators) > 0:
|
|
gen = random.choice(generators)
|
|
if gen in MISSILES_MAP.keys():
|
|
generator = MISSILES_MAP[gen](game, ground_object, faction)
|
|
generator.generate()
|
|
return generator.get_generated_group()
|
|
else:
|
|
logging.info(
|
|
"Unable to generate missile group, generator : "
|
|
+ str(gen)
|
|
+ "does not exists"
|
|
)
|
|
return None
|