First build V2

1 files changed, 211 insertions, 0 deletions

diff --git a/sassbot/modules/artifact2-0/Items.py b/sassbot/modules/artifact2-0/Items.py
new file mode 100644
index 0000000..7841d7d
--- /dev/null
+++ b/sassbot/modules/artifact2-0/Items.py
@@ -0,0 +1,211 @@
+import MySQLdb

+import random

+

+def GenerateEquipment(phenny, parameters, itemCode):

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

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

+    intIteration=0

+    itemName=strRaritySpacing='';

+

+    

+    itemCode = itemCode.split(':')

+

+    #Generate Rarity

+    itemGenRarity = CreateEquipmentRarity(phenny, parameters, itemCode[0])

+

+    #Generate Item

+    itemGenEquipment = CreateEquipmentName(phenny, parameters, itemCode[1])

+    itemGenEquipment['FabledPrefix'] = ''

+

+    #Generate Enchantment

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

+

+    #Generate Element

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

+

+    #Attempt creating an Artifact

+    itemGenEquipment = CreateEquipmentArtifact(phenny, parameters, itemGenEquipment, itemCode[4])

+

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

+

+    

+

+    itemGen={}

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

+    itemGen['EquipmentName'] = (itemGenEquipment['FabledPrefix'] +

+                                itemGenRarity['RarityName'] +

+                                strRaritySpacing + 

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

+                                itemGenEquipment['EquipmentName'] + " " +

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

+

+    itemGen['EquipmentSlot'] = itemGenEquipment['EquipmentSlot']

+    itemGen['EquipmentType'] = itemGenEquipment['EquipmentType']

+    itemGen['HandsReq'] = itemGenEquipment['HandsReq']

+    itemGen['Strength'] = itemGenEquipment['Strength']

+    itemGen['Dexterity'] = itemGenEquipment['Dexterity']

+    itemGen['Constitution'] = itemGenEquipment['Constitution']

+    itemGen['Intelligence'] = itemGenEquipment['Intelligence']

+    itemGen['Dodge'] = itemGenEquipment['Dodge']

+    itemGen['Crit'] = itemGenEquipment['Crit']

+    itemGen['Armor'] = itemGenEquipment['Armor']

+    itemGen['Precision']  = itemGenEquipment['Precision']

+    itemGen['Resistance'] = itemGenEquipment['Resistance']

+    itemGen['Luck'] = itemGenEquipment['Luck']

+    itemGen['Block'] = itemGenEquipment['Block']

+    itemGen['Permanent'] = itemGenEquipment['Permanent']

+

+    itemGen['Cost'] = (itemGenEquipment['Cost'] + 

+                        (0 if itemGenElement == {} else itemGenElement['Cost']) +

+                        (0 if itemGenEnchantment == {} else itemGenEnchantment['Cost']) +

+                        (0 if itemGenRarity == {} else itemGenRarity['RarityCost']))

+    

+    return itemGen

+

+def CreateEquipmentName(phenny, parameters, itemCode):

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

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

+    itemGenEquipment = {}

+

+

+    #Item by ID

+    if int(itemCode) is not 0:

+        c.execute("SELECT `EquipmentName`,`EquipmentSlot`,`EquipmentType`,`HandsReq`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`,`Crit`,`Armor`,`Precision`,`Resistance`,`Luck`,`Block` FROM `Items_Master` WHERE `ID` = " + str(itemCode))

+        itemGenEquipment = c.fetchone()

+

+    #Item by Random

+    elif int(itemCode) is 0:

+        #UPDATE: Should be able to generate an available weapon with just one query

+        # Pick random item that is available to the player

+        c.execute("SELECT `Items_PlayerPool`.`Nick`, `Items_PlayerPool`.`EquipmentName`, `Items_Master`.`Locked` FROM `Items_PlayerPool` INNER JOIN `Items_Master` ON `Items_PlayerPool`.`EquipmentName` = `Items_Master`.`EquipmentName` WHERE `Nick`=%s AND `Locked`=0 ORDER BY RAND() LIMIT 1",parameters['Nick'])

+        sqlItemPlayer = c.fetchone()

+

+        # Get information from randomly picked weapon

+        c.execute("SELECT `EquipmentName`, `EquipmentSlot`, `EquipmentType`, `HandsReq`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit`, `Armor`, `Precision`, `Resistance`, `Luck`, `Block` FROM `Items_Master` WHERE `EquipmentName`=%s",sqlItemPlayer['EquipmentName'])

+        itemGenEquipment = c.fetchone()

+

+    itemGenEquipment['Permanent'] = 0

+    itemGenEquipment['Cost'] = 100

+    return itemGenEquipment

+

+def CreateEquipmentElement(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)

+

+    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()

+

+        itemGenElement['Cost'] = 200

+    return itemGenElement

+

+def CreateEquipmentRarity(phenny, parameters, itemCode):

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

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

+    itemGenRarity = {}

+    if int(itemCode) is not 0:

+        c.execute("SELECT `ID`,`RarityName`,`RarityColor`,`RarityModifier`,`RarityChance`,`RarityCost` FROM `GenRarity` WHERE `ID` = " + str(itemCode))

+        itemGenRarity = c.fetchone()

+    elif int(itemCode) is 0:

+        # Obtain sum of all rarities

+        c.execute("SELECT SUM(`RarityChance`) AS RarityTotal FROM `GenRarity`")

+        intRarityTotal = c.fetchone()['RarityTotal']

+

+        # Variable declaration

+        intIteration = 0

+        intRandom = random.randint(1, intRarityTotal)

+

+        while True:

+            intIteration = intIteration + 1

+

+            # Obtain Rarity in order of lowest to highest with each iteration

+            c.execute("SELECT `ID`,`RarityName`,`RarityColor`,`RarityModifier`,`RarityChance`,`RarityCost` FROM `GenRarity` WHERE `ID` = " + str(intIteration))

+            itemGenRarity = c.fetchone()

+

+            # Subtract the current iterations rarity from the random roll

+            # If intRandom becomes negative, our item rarity has been found. End the loop.

+            intRandom = intRandom - itemGenRarity['RarityChance']

+            if intRandom <= 0:

+                break

+    return itemGenRarity

+    db.close()

+

+def CreateEquipmentEnchantment(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)

+

+    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()

+

+        itemGenEnchantment['Cost'] = 200

+    return itemGenEnchantment

+

+def CreateEquipmentArtifact(phenny, parameters, itemGenEquipment, itemCode):

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

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

+

+    # Gen for Artifact item:

+    # 0 - Will attempt to add an artifact randomly and not exceed the standard limit

+    if random.randint(1, 100) == 1 and int(itemCode) == 0:

+        if itemGenEquipment['EquipmentType'] == 'Armor' or itemGenEquipment['EquipmentType'] == 'Weapon':

+            c.execute("SELECT COUNT(`Permanent`) AS `Artifacts` FROM `Items_PlayerInventory` WHERE `Nick`=%s AND `Permanent`=1;",parameters['Nick'])

+            sqlCountArtifacts = c.fetchone()

+            if sqlCountArtifacts['Artifacts'] < 2:

+                itemGenEquipment['FabledPrefix'] = 'Fabled '

+                itemGenEquipment['Permanent'] = 1

+    # 1 - Will not generate item as an artifact

+    elif int(itemCode) == 1:

+        itemGenEquipment['FabledPrefix'] = ''

+        

+    # 2 - Will force the item to become an artifact and exceed the limit of 2 artifacts

+    elif int(itemCode) == 2:

+        if itemGenEquipment['EquipmentType'] == 'Armor' or itemGenEquipment['EquipmentType'] == 'Weapon':

+            c.execute("SELECT COUNT(`Permanent`) AS `Artifacts` FROM `Items_PlayerInventory` WHERE `Nick`=%s AND `Permanent`=1;",parameters['Nick'])

+            sqlCountArtifacts = c.fetchone()

+            itemGenEquipment['FabledPrefix'] = 'Fabled '

+            itemGenEquipment['Permanent'] = 1

+    return itemGenEquipment

+

+def CreateItem(phenny, enemyStats, playerStats, parameters):

+    import random

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

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

+

+    # Get random element

+    c.execute("SELECT `LootName`,`ItemType`,`Potency` FROM `GenLoot` WHERE `Locked`=0 ORDER BY RAND() LIMIT 1")

+    sqlRarity = c.fetchone()

+    strLootName = sqlRarity['LootName']

+    strItemType = sqlRarity['ItemType']

+    intPotency = sqlRarity['Potency']

+

+    if strItemType == 'Quantity':

+        intPotency = enemyStats['GoldCarried'] + int(

+            random.uniform(float(enemyStats['GoldCarried']) * -0.3, float(enemyStats['GoldCarried']) * 0.3))

+

+    db.close()

+    return strLootName, intPotency

+

+def AddToShop(phenny, parameters, item):

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

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

+    c.execute("""INSERT INTO `PlayerShop` (`Nick`,`Cost`,`Equipped`,`EquipmentColor`,`EquipmentName`,`EquipmentSlot`,`EquipmentType`,`Permanent`,`HandsReq`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`,`Crit`,`Armor`) VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s);""", (parameters['Nick'], item['Cost'], 0, item['RarityColor'], item['EquipmentName'], item['EquipmentSlot'], item['EquipmentType'],item['Permanent'], item['HandsReq'], item['Strength'], item['Dexterity'], item['Constitution'], item['Intelligence'], item['Dodge'], item['Crit'], item['Armor']));db.commit();

+    return

+

+def AddToInventory(phenny, parameters, item):

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

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

+    c.execute("""INSERT INTO `Items_PlayerInventory` (`EquipmentColor`,`EquipmentName`,`EquipmentSlot`,`EquipmentType`,`Permanent`,`HandsReq`,`Equipped`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`,`Crit`,`Armor`,`Nick`) VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s);""", (item['RarityColor'], item['EquipmentName'], item['EquipmentSlot'], item['EquipmentType'],item['Permanent'], item['HandsReq'], 0, item['Strength'], item['Dexterity'], item['Constitution'], item['Intelligence'], item['Dodge'], item['Crit'], item['Armor'],parameters['Nick']));db.commit();

+    return
\ No newline at end of file