make add omega icon to hulls + make omega attr persistent

Author: Skybladev2

db_update.py

@@ -649,6 +649,27 @@ def _hardcodeAttribs(typeID, attrMap):
             else:
                 attr.value = value
 
+    def processCloneGradeRestrictions():
+        print('processing clone grade restrictions')
+        alpha_caps = {}
+        for clone in eos.db.gamedata_session.query(eos.gamedata.AlphaClone).all():
+            for skill in clone.skills:
+                alpha_caps[skill.typeID] = skill.level
+        for item in eos.db.gamedata_session.query(eos.gamedata.Item).all():
+            if not item.reqskills:
+                continue
+            try:
+                reqskills = json.loads(item.reqskills)
+            except (ValueError, TypeError):
+                continue
+            for skill_type_id, required_level in reqskills.items():
+                max_alpha_level = alpha_caps.get(int(skill_type_id))
+                if max_alpha_level is None or required_level > max_alpha_level:
+                    _hardcodeAttribs(item.ID, {"cloneGradeRestriction": 1})
+                    break
+
+    processCloneGradeRestrictions()
+
     def _hardcodeEffects(typeID, effectMap, clearEffects=True):
         item = eos.db.gamedata_session.query(eos.gamedata.Item).filter(eos.gamedata.Item.ID == typeID).one()
         if clearEffects:

gui/builtinShipBrowser/shipItem.py

@@ -93,6 +93,10 @@ def __init__(self, parent, shipID=None, shipFittingInfo=("Test", "TestTrait", 2)
         self.marketInstance = Market.getInstance()
         self.baseItem = self.marketInstance.getItem(self.shipID)
 
+        self.omegaBmp = None
+        if self.baseItem.getAttribute("cloneGradeRestriction", 0) > 0:
+            self.omegaBmp = BitmapLoader.getBitmap("25874", "icons")
+
         # =====================================================================
         # DISABLED - it will be added as an option in PREFERENCES
 
@@ -211,7 +215,12 @@ def UpdateElementsPos(self, mdc):
         self.raceBmpx = self.shipEffx + self.shipEffBk.GetWidth() + self.padding
         self.raceBmpy = (rect.height - self.raceBmp.GetHeight()) / 2
 
-        self.textStartx = self.raceBmpx + self.raceBmp.GetWidth() + self.padding
+        self.omegaBmpx = self.raceBmpx + self.raceBmp.GetWidth() + self.padding
+        omegaWidth = self.omegaBmp.GetWidth() if self.omegaBmp else 0
+        omegaHeight = self.omegaBmp.GetHeight() if self.omegaBmp else 0
+        self.omegaBmpy = (rect.height - omegaHeight) / 2
+
+        self.textStartx = self.omegaBmpx + omegaWidth + self.padding
 
         self.shipNamey = (rect.height - self.shipBmp.GetHeight()) / 2
 
@@ -254,6 +263,9 @@ def DrawItem(self, mdc):
         mdc.DrawBitmap(self.raceDropShadowBmp, round(self.raceBmpx + 1), round(self.raceBmpy + 1))
         mdc.DrawBitmap(self.raceBmp, round(self.raceBmpx), round(self.raceBmpy))
 
+        if self.omegaBmp:
+            mdc.DrawBitmap(self.omegaBmp, round(self.omegaBmpx), round(self.omegaBmpy))
+
         shipName, shipTrait, fittings = self.shipFittingInfo
 
         if fittings < 1: