From 6ce72e4fb3b9ac7f7163128db250babac53ef8b2 Mon Sep 17 00:00:00 2001 From: DarkPhoenix Date: Sun, 7 Jul 2019 19:48:50 +0300 Subject: [PATCH] Add methods which will be used to access temporarily modified values --- eos/modifiedAttributeDict.py | 26 ++++++++++++++------------ 1 file changed, 14 insertions(+), 12 deletions(-) diff --git a/eos/modifiedAttributeDict.py b/eos/modifiedAttributeDict.py index f73ef08e4..e44d9b6cf 100644 --- a/eos/modifiedAttributeDict.py +++ b/eos/modifiedAttributeDict.py @@ -28,32 +28,34 @@ cappingAttrKeyCache = {} class ItemAttrShortcut: + def getModifiedItemAttr(self, key, default=0): return_value = self.itemModifiedAttributes.get(key) + return return_value or default + def getModifiedItemAttrWithExtraMods(self, key, multipliers=(), boosts=(), default=0): + """Returns attribute value with passed modifiers applied to it.""" + return_value = self.itemModifiedAttributes.get(key) return return_value or default def getItemBaseAttrValue(self, key, default=0): - """ - Gets base value in this order: - Mutated value > override value > attribute value - """ return_value = self.itemModifiedAttributes.getOriginal(key) return return_value or default - def getChargeBaseAttrValue(self, key, default=0): - """ - Gets base value in this order: - Mutated value > override value > attribute value - """ - return_value = self.chargeModifiedAttributes.getOriginal(key) - return return_value or default - class ChargeAttrShortcut: + def getModifiedChargeAttr(self, key, default=0): return_value = self.chargeModifiedAttributes.get(key) + return return_value or default + def getModifiedChargeAttrWithExtraMods(self, key, multipliers=(), boosts=(), default=0): + """Returns attribute value with passed modifiers applied to it.""" + return_value = self.itemModifiedAttributes.get(key) + return return_value or default + + def getChargeBaseAttrValue(self, key, default=0): + return_value = self.chargeModifiedAttributes.getOriginal(key) return return_value or default