Generate WW2 Ship groups, added B17 to allies. Implemented modifiable doctrine to setup flight generator.

gen/fleet/ship_group_generator.py

@@ -1,6 +1,35 @@
+import random
+
 from game import db
 from gen.fleet.carrier_group import CarrierGroupGenerator
 from gen.fleet.lha_group import LHAGroupGenerator
+from dcs.ships import *
+
+from gen.fleet.schnellboot import SchnellbootGroupGenerator
+from gen.fleet.uboat import UBoatGroupGenerator
+from gen.fleet.ww2lst import WW2LSTGroupGenerator
+
+SHIP_MAP = {
+    Schnellboot_type_S130: SchnellbootGroupGenerator,
+    LS_Samuel_Chase: WW2LSTGroupGenerator,
+    Uboat_VIIC_U_flak: UBoatGroupGenerator
+}
+
+
+def generate_ship_group(game, ground_object, faction:str):
+    """
+    This generate a ship group
+    :return: Nothing, but put the group reference inside the ground object
+    """
+    faction = db.FACTIONS[faction]
+    if "boat" in faction.keys():
+        ships = faction["boat"]
+        if len(ships) > 0:
+            sam = random.choice(ships)
+            generator = SHIP_MAP[sam](game, ground_object, faction)
+            generator.generate()
+            return generator.get_generated_group()
+    return None
 
 
 def generate_carrier_group(faction:str, game, ground_object):