Hardcode sov upgrade buffs onto their carrying items

Author: DarkFenX

db_update.py

@@ -643,13 +643,17 @@ def _hardcodeAttribs(typeID, attrMap):
             else:
                 attr.value = value
 
-    def _hardcodeEffects(typeID, effectMap):
+    def _hardcodeEffects(typeID, effectMap, clearEffects=True):
         item = eos.db.gamedata_session.query(eos.gamedata.Item).filter(eos.gamedata.Item.ID == typeID).one()
-        item.effects.clear()
+        if clearEffects:
+            item.effects.clear()
         for effectID, effectName in effectMap.items():
-            effect = eos.gamedata.Effect()
-            effect.effectID = effectID
-            effect.effectName = effectName
+            try:
+                effect = eos.db.gamedata_session.query(eos.gamedata.Effect).filter(eos.gamedata.Effect.ID == effectID).one()
+            except sqlalchemy.orm.exc.NoResultFound:
+                effect = eos.gamedata.Effect()
+                effect.effectID = effectID
+                effect.effectName = effectName
             item.effects[effectName] = effect
 
     def hardcodeSuppressionTackleRange():
@@ -660,6 +664,43 @@ def hardcodeSuppressionTackleRange():
         effectMap = {100000: 'pyfaCustomSuppressionTackleRange'}
         _hardcodeAttribs(beaconTypeID, attrMap)
         _hardcodeEffects(beaconTypeID, effectMap)
+        eos.db.gamedata_session.flush()
+
+    def hardcodeSovUpgradeBuffs():
+        typeBuffMap = {
+            # Gamma
+            87815: {
+                'warfareBuff1ID': 2433,
+                'warfareBuff1Value': 5,
+                'warfareBuff2ID': 2434,
+                'warfareBuff2Value': 10,
+                'warfareBuff3ID': 2441,
+                'warfareBuff3Value': 5},
+            # Plasma
+            87949: {
+                'warfareBuff1ID': 2442,
+                'warfareBuff1Value': 5,
+                'warfareBuff2ID': 2435,
+                'warfareBuff2Value': 5,
+                'warfareBuff3ID': 2436,
+                'warfareBuff3Value': 10},
+            # Electric
+            87950: {
+                'warfareBuff1ID': 2437,
+                'warfareBuff1Value': -25,
+                'warfareBuff2ID': 2438,
+                'warfareBuff2Value': 25},
+            # Exotic
+            87951: {
+                'warfareBuff1ID': 2440,
+                'warfareBuff1Value': 2,
+                'warfareBuff2ID': 2439,
+                'warfareBuff2Value': 25}}
+        effectMap = {100001: 'pyfaCustomSovUpgradeBuffEffect'}
+        for typeID, attrMap in typeBuffMap.items():
+            _hardcodeAttribs(typeID, attrMap)
+            _hardcodeEffects(typeID, effectMap, clearEffects=False)
+            eos.db.gamedata_session.flush()
 
 
     def hardcodeShapash():
@@ -806,6 +847,7 @@ def hardcodeCybele():
         _hardcodeEffects(cybeleTypeID, effectMap)
 
     hardcodeSuppressionTackleRange()
+    hardcodeSovUpgradeBuffs()
 
     eos.db.gamedata_session.commit()
     eos.db.gamedata_engine.execute('VACUUM')

eos/effects.py

@@ -50,13 +50,26 @@ class Effect100000(BaseEffect):
 
     @staticmethod
     def handler(fit, beacon, context, projectionRange, **kwargs):
-        if beacon.getModifiedItemAttr('warfareBuff1ID'):
+        if id := beacon.getModifiedItemAttr('warfareBuff1ID'):
             value = beacon.getModifiedItemAttr('warfareBuff1Value')
-            id = beacon.getModifiedItemAttr('warfareBuff1ID')
             if id:
                 fit.addCommandBonus(id, value, beacon, kwargs['effect'], 'early')
 
 
+class Effect100001(BaseEffect):
+
+    runTime = 'early'
+    type = ('projected', 'passive', 'gang')
+
+    @staticmethod
+    def handler(fit, beacon, context, projectionRange, **kwargs):
+        for i in (1, 4):
+            if id := beacon.getModifiedItemAttr(f'warfareBuff{i}ID'):
+                value = beacon.getModifiedItemAttr(f'warfareBuff{i}Value')
+                if id:
+                    fit.addCommandBonus(id, value, beacon, kwargs['effect'], 'early')
+
+
 class Effect4(BaseEffect):
     """
     shieldBoosting