frontline attack ops

game/event/baseattack.py

@@ -0,0 +1,89 @@
+import math
+import random
+
+from dcs.task import *
+
+from game import db
+from game.operation.baseattack import BaseAttackOperation
+from userdata.debriefing import Debriefing
+
+from .event import Event
+
+
+class BaseAttackEvent(Event):
+    silent = True
+    BONUS_BASE = 15
+    STRENGTH_RECOVERY = 0.55
+
+    def __str__(self):
+        return "Attack from {} to {}".format(self.from_cp, self.to_cp)
+
+    def is_successfull(self, debriefing: Debriefing):
+        alive_attackers = sum([v for k, v in debriefing.alive_units[self.attacker_name].items() if db.unit_task(k) == PinpointStrike])
+        alive_defenders = sum([v for k, v in debriefing.alive_units[self.defender_name].items() if db.unit_task(k) == PinpointStrike])
+        attackers_success = alive_attackers >= alive_defenders
+        if self.from_cp.captured:
+            return attackers_success
+        else:
+            return not attackers_success
+
+    def commit(self, debriefing: Debriefing):
+        super(BaseAttackEvent, self).commit(debriefing)
+        if self.is_successfull(debriefing):
+            if self.from_cp.captured:
+                self.to_cp.captured = True
+                self.to_cp.base.filter_units(db.UNIT_BY_COUNTRY[self.attacker_name])
+
+            self.to_cp.base.affect_strength(+self.STRENGTH_RECOVERY)
+        else:
+            if not self.from_cp.captured:
+                self.to_cp.captured = False
+            self.to_cp.base.affect_strength(+self.STRENGTH_RECOVERY)
+
+    def skip(self):
+        if not self.is_player_attacking and self.to_cp.captured:
+            self.to_cp.captured = False
+
+    def player_defending(self, interceptors: db.PlaneDict, clients: db.PlaneDict):
+        cas = self.from_cp.base.scramble_cas(self.game.settings.multiplier)
+        escort = self.from_cp.base.scramble_sweep(self.game.settings.multiplier)
+        attackers = self.from_cp.base.armor
+
+        op = BaseAttackOperation(game=self.game,
+                                 attacker_name=self.attacker_name,
+                                 defender_name=self.defender_name,
+                                 attacker_clients={},
+                                 defender_clients=clients,
+                                 from_cp=self.from_cp,
+                                 to_cp=self.to_cp)
+
+        op.setup(cas=cas,
+                 escort=escort,
+                 attack=attackers,
+                 intercept=interceptors,
+                 defense=self.to_cp.base.armor,
+                 aa=self.to_cp.base.aa)
+
+        self.operation = op
+
+    def player_attacking(self, cas: db.PlaneDict, escort: db.PlaneDict, armor: db.ArmorDict, clients: db.PlaneDict):
+        op = BaseAttackOperation(game=self.game,
+                                 attacker_name=self.attacker_name,
+                                 defender_name=self.defender_name,
+                                 attacker_clients=clients,
+                                 defender_clients={},
+                                 from_cp=self.from_cp,
+                                 to_cp=self.to_cp)
+
+        defenders = self.to_cp.base.scramble_sweep(self.game.settings.multiplier)
+        defenders.update(self.to_cp.base.scramble_cas(self.game.settings.multiplier))
+
+        op.setup(cas=cas,
+                 escort=escort,
+                 attack=armor,
+                 intercept=defenders,
+                 defense=self.to_cp.base.armor,
+                 aa=self.to_cp.base.assemble_aa())
+
+        self.operation = op
+