Equipment rolls working as desired

1 files changed, 35 insertions, 20 deletions

diff --git a/sassbot/modules/artifact2-0/EquipmentCreate.py b/sassbot/modules/artifact2-0/EquipmentCreate.py
index e1520d7..c7f1c8c 100644
--- a/sassbot/modules/artifact2-0/EquipmentCreate.py
+++ b/sassbot/modules/artifact2-0/EquipmentCreate.py
@@ -14,19 +14,28 @@ def GenerateItem(phenny, parameters, itemCode):
     itemGenEquipment = CreateItemName(phenny, parameters, itemCode[1])

 

     #Generate Enchantment

-    itemGenEnchantment = CreateItemEnchantment(phenny, parameters, itemCode[2])

+    itemGenEnchantment = CreateItemEnchantment(phenny, parameters, itemCode[2], int(itemGenRarity["ID"]))

 

     #Generate Element

-    itemGenElement = CreateItemElement(phenny, parameters, itemCode[3])

+    itemGenElement = CreateItemElement(phenny, parameters, itemCode[3], int(itemGenRarity["ID"]))

 

     #Attempt creating an Artifact

     itemGenEquipment = CreateItemArtifact(phenny, parameters, itemGenEquipment, itemCode[4])

 

     strRaritySpacing = "" if len(itemGenRarity['RarityName']) == 0 else " "

 

+

+    

+

     itemGen={}

-    itemGen['EquipmentName'] = "\x03"+itemGenRarity['RarityColor'] + itemGenEquipment['FabledPrefix'] + itemGenRarity['RarityName'] + strRaritySpacing + itemGenElement['EffectName'] + " " + itemGenEquipment['EquipmentName'] + " " + itemGenEnchantment['Name']+"\x0399"

-    #itemGen['EquipmentName']

+    itemGen['EquipmentName'] = ("\x03"+itemGenRarity['RarityColor'] + 

+                                itemGenEquipment['FabledPrefix'] + 

+                                itemGenRarity['RarityName'] + 

+                                strRaritySpacing + 

+                                ("" if itemGenElement == {} else itemGenElement['EffectName'] + " ") + 

+                                itemGenEquipment['EquipmentName'] + " " + 

+                                ("" if itemGenEnchantment == {} else itemGenEnchantment['Name'] + " ") +

+                                "\x0399")

     #itemGen['EquipmentSlot']

     #itemGen['EquipmentType']

     #itemGen['HandsReq']

@@ -68,16 +77,19 @@ def CreateItemName(phenny, parameters, itemCode):
         itemGenEquipment['Permanent'] = 0

         return itemGenEquipment

 

-def CreateItemElement(phenny, parameters, itemCode):

+def CreateItemElement(phenny, parameters, itemCode, RarityID):

     db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])

     c = db.cursor(MySQLdb.cursors.DictCursor)

-    itemGenElement = {}

-    if int(itemCode) is not 0:

-        c.execute("SELECT `EffectName`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit`, `Armor` FROM `GenElement` WHERE `ID` = %s",itemCode)

-        itemGenElement = c.fetchone()

-    elif int(itemCode) is 0:

-        c.execute("SELECT `EffectName`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit`, `Armor` FROM `GenElement` ORDER BY RAND() LIMIT 1")

-        itemGenElement = c.fetchone()

+

+    if random.randint(1, 3) is 1 or RarityID <= 3:

+        itemGenElement = {}

+    else:

+        if int(itemCode) is not 0:

+            c.execute("SELECT `EffectName`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit`, `Armor` FROM `GenElement` WHERE `ID` = %s",itemCode)

+            itemGenElement = c.fetchone()

+        elif int(itemCode) is 0:

+            c.execute("SELECT `EffectName`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit`, `Armor` FROM `GenElement` ORDER BY RAND() LIMIT 1")

+            itemGenElement = c.fetchone()

     return itemGenElement

 

 def CreateItemRarity(phenny, parameters, itemCode):

@@ -111,16 +123,19 @@ def CreateItemRarity(phenny, parameters, itemCode):
     return itemGenRarity

     db.close()

 

-def CreateItemEnchantment(phenny, parameters, itemCode):

+def CreateItemEnchantment(phenny, parameters, itemCode, RarityID):

     db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])

     c = db.cursor(MySQLdb.cursors.DictCursor)

-    itemGenElement = {}

-    if int(itemCode) is not 0:

-        c.execute("SELECT `Name`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`, `Crit`, `Armor` FROM `GenEnchantment` WHERE `ID` = %s", str(itemCode))

-        itemGenEnchantment = c.fetchone()

-    elif int(itemCode) is 0:

-        c.execute("SELECT `Name`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`, `Crit`, `Armor` FROM `GenEnchantment` ORDER BY RAND() LIMIT 1")

-        itemGenEnchantment = c.fetchone()

+

+    if random.randint(1, 3) is 1 or RarityID <= 2:

+        itemGenEnchantment = {}

+    else:

+        if int(itemCode) is not 0:

+            c.execute("SELECT `Name`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`, `Crit`, `Armor` FROM `GenEnchantment` WHERE `ID` = %s", str(itemCode))

+            itemGenEnchantment = c.fetchone()

+        elif int(itemCode) is 0:

+            c.execute("SELECT `Name`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`, `Crit`, `Armor` FROM `GenEnchantment` ORDER BY RAND() LIMIT 1")

+            itemGenEnchantment = c.fetchone()

     return itemGenEnchantment

 

 def CreateItemArtifact(phenny, parameters, itemGenEquipment, itemCode):