Added the start of weapon skills calc and lots of little fixes.

Barry
Showing 4 changed files with 61 additions and 17 deletions
data.py
db.sqlite3
hellsbot.py
pankration.py
--- a/data.py
View file @226fb77
+++ b/data.py
View file @226fb77
@@ -339,6 +339,14 @@ def db_remove_soul_plate(member_id, idx):
        conn.commit()
    return True
+def db_update_pankration_record(member_id, column_name, value):
+    conn = sqlite3.connect('db.sqlite3')
+    c = conn.cursor()
+    c.execute("""UPDATE pankration SET """ + column_name + """ = ? 
+             WHERE member_id = ?;""", (value, member_id))
+    conn.commit()
+    conn.close()
 def db_add_reflector(member_id, reflector):
    pan_record = db_get_pankration_record(member_id)
    log("{}".format(member_id))
@@ -394,13 +402,20 @@ def db_convert_soul_plate_to_reflector(member_id, soul_plate, idx):
    return True
-def db_register_battle(member_id, reflector, idx, target_member=None):
+def db_register_battle(member_id, reflector, idx, target_member_id=None):
    pan_record = db_get_pankration_record(member_id)
    conn = sqlite3.connect('db.sqlite3')
    c = conn.cursor()
-    db_state = c.execute("SELECT battle_id, reflector_primary member_id FROM pankration_arena WHERE battle_status = 'waiting';").fetchall()
+    if target_member_id:
+        db_state = c.execute("SELECT battle_id, reflector_primary FROM pankration_arena WHERE battle_status = 'waiting' AND direct_challenge = ?;", (member_id,)).fetchall()
+    else:
+        db_state = c.execute("SELECT battle_id, reflector_primary FROM pankration_arena WHERE battle_status = 'waiting' AND direct_challenge = 0;").fetchall()
    if not db_state:
        log("No battles available: {}".format(idx,))
+        if target_member_id:
+            c.execute("""INSERT INTO pankration_arena(primary_member_id, reflector_primary, direct_challenge) 
+                      VALUES(?, ?, ?);""", (member_id, pickle.dumps(reflector), target_member_id))
+        else:
            c.execute("""INSERT INTO pankration_arena(primary_member_id, reflector_primary) 
                      VALUES(?, ?);""", (member_id, pickle.dumps(reflector)))
        conn.commit()
@@ -440,16 +455,17 @@ def db_complete_battle(battle_id, primary_member_id, primary_reflector, secondar
    else:
        c.execute("""UPDATE pankration SET reflectors = ?, losses = losses + 1
         WHERE member_id = ?;""", (pickle.dumps(reflectors), primary_member_id))
+    conn.commit()
-    reflectors = pickle.loads(str(db_get_pankration_record(secondary_member_id)['reflectors']))
+    reflectors2 = pickle.loads(str(db_get_pankration_record(secondary_member_id)['reflectors']))
-    reflectors.append(secondary_reflector)
+    reflectors2.append(secondary_reflector)
    log('Completing battle - secondary')
    if winner == 2:
        c.execute("""UPDATE pankration SET reflectors = ?, wins = wins + 1
-         WHERE member_id = ?;""", (pickle.dumps(reflectors), secondary_member_id))
+         WHERE member_id = ?;""", (pickle.dumps(reflectors2), secondary_member_id))
    else:
        c.execute("""UPDATE pankration SET reflectors = ?, losses = losses + 1
-         WHERE member_id = ?;""", (pickle.dumps(reflectors), secondary_member_id))
+         WHERE member_id = ?;""", (pickle.dumps(reflectors2), secondary_member_id))
    conn.commit()
    conn.close()
    log('Completing battle - done')
--- a/db.sqlite3
View file @226fb77
+++ b/db.sqlite3
View file @226fb77
--- a/hellsbot.py
View file @226fb77
+++ b/hellsbot.py
View file @226fb77
@@ -93,6 +93,7 @@ registered_commands = {'!help': 'do_help', '!commands': 'do_help',
                       '!convertplate': 'do_convert_plate', '!convertsoulplate': 'do_convert_plate',
                       '!assignskill': 'do_assign_skill',
                       '!registerbattle': 'do_register_battle',
+                       '!setreflectorname': 'do_set_reflector_name',
                       '!pankration': 'do_pankration',
                       }
@@ -1041,6 +1042,7 @@ def do_pankration(client, message_parts, message):
        Ex:  **!convertplate 1 reflector**
             **!convertplate 4 skill**
    **!assignskill <reflector number> <skill num>** - Assigns a skill to a reflector. Each skill is worth a certain amount of feral points, each monster has a maximum of feral points available.
+    **!setreflectorname <reflector number> <name>** - This will rename a monster in your collection to a custom name.
 *Arena Battle:*
    **!registerbattle <reflector number>** - Adds your reflector to the queue in the arena.  When the arena is available your monster will be paired with either another players monster or a similarly matched arena monster.
@@ -1048,6 +1050,28 @@ def do_pankration(client, message_parts, message):
 """.format(message.author.mention()))
+def do_set_reflector_name(client, message_parts, message):
+    member = data.db_get_member(message.author.id)
+    if len(message_parts) < 2:
+        send_message(client, message.channel, 'You must provide the reflector number and the new name.')
+        return
+    reflector_num = message_parts[0]
+    new_name = ' '.join(message_parts[1:])
+    pankration_data = data.db_get_pankration_record(member['member_id'])
+    if pankration_data and pankration_data['reflectors']:
+        reflectors = pickle.loads(str(pankration_data['reflectors']))
+        if not reflector_num.isdigit() or int(reflector_num) < 1 or int(reflector_num) > len(reflectors):
+            send_message(client, message.channel, 'The requested reflector is invalid. Please provide the number from !listreflectors')
+            return
+        reflector_num = int(reflector_num) - 1
+        reflectors[reflector_num].set_monster_name(new_name)
+        data.db_update_pankration_record(member['member_id'], 'reflectors', pickle.dumps(reflectors))
+        send_message(client, message.channel, 'The reflector (monster) name has been updated.')        
+    else:
+        send_message(client, message.channel, 'Unable to find the reflector data.')
 def do_register_battle(client, message_parts, message):
    member = data.db_get_member(message.author.id)
@@ -1067,13 +1091,17 @@ def do_register_battle(client, message_parts, message):
        reflector_num = int(reflector_num) - 1
        if len(message_parts) > 1:
-            user = ' '.join(message_parts[1:])
+            username = ' '.join(message_parts[1:])
-            data.db_register_battle(member['member_id'], reflectors[reflector_num], reflector_num, user)
+            target_member = data.db_get_member(username=username)
-            send_message(client, message.channel, "Your *{} level {}* has been registered for battle with {}. The battle will begin when the arena is availble with the next challenger".format(reflectors[reflector_num].monster_type, reflectors[reflector_num].level, user))
+            if target_member and 'member_id' in target_member:
+                data.db_register_battle(member['member_id'], reflectors[reflector_num], reflector_num, target_member['member_id'])
+                send_message(client, message.channel, "Your *{} level {}* has been registered for battle with {}. The battle will begin when the arena is availble with the next challenger".format(reflectors[reflector_num].get_monster_name(), reflectors[reflector_num].level, username))
+            else:
+                send_message(client, message.channel, "There was a problem looking up that user.")
        else:
            data.db_register_battle(member['member_id'], reflectors[reflector_num], reflector_num)
-            send_message(client, message.channel, "Your *{} level {}* has been registered for battle. The battle will begin when the arena is availble with the next challenger".format(reflectors[reflector_num].monster_type, reflectors[reflector_num].level))
+            send_message(client, message.channel, "Your *{} level {}* has been registered for battle. The battle will begin when the arena is availble with the next challenger".format(reflectors[reflector_num].get_monster_name(), reflectors[reflector_num].level))
    else:
        send_message(client, message.channel, 'You have no available reflectors. You can get reflectors by converting soul plates with !convertplate')
        return
@@ -1221,15 +1249,15 @@ def check_arena():
        send_message(client, arena_channel, "Ladies and gentlemen!\nFor our next match...")
        time.sleep(5)
-        send_message(client, arena_channel, "In the red corner we have...\n  **{}**!".format(monster.monster_type))
+        send_message(client, arena_channel, "In the red corner we have...\n  **{}**!".format(monster.get_monster_name()))
        time.sleep(5)
        send_message(client, arena_channel, "Hmmm... This monster seems {} and {}.".format(monster.get_current_posture()["name"], monster.get_current_attitude()["name"]))
        time.sleep(5)
-        send_message(client, arena_channel, "And in the blue corner is...\n  **{}**!".format(monster2.monster_type))
+        send_message(client, arena_channel, "And in the blue corner is...\n  **{}**!".format(monster2.get_monster_name()))
        time.sleep(5)
        send_message(client, arena_channel, "Hmmm... This monster seems {} and {}.".format(monster2.get_current_posture()["name"], monster2.get_current_attitude()["name"]))
        time.sleep(5)
-        send_message(client, arena_channel, "Alright, Pankration fans, the match is about to begin!\n  *Chaaaaaarge!*".format(monster2.monster_type))
+        send_message(client, arena_channel, "Alright, Pankration fans, the match is about to begin!\n  *Chaaaaaarge!*".format(monster2.get_monster_name()))
        time.sleep(5)
        data.db_start_battle(battle['battle_id'])
@@ -1244,12 +1272,12 @@ def check_arena():
            out_str = ""
            for action in actions:
                if isinstance(action, MissAction):
-                    out_str += "{} {} {}.\n".format(action.attacker.monster_type, action.message, action.target.monster_type)
+                    out_str += "{} {} {}.\n".format(action.attacker.get_monster_name(), action.message, action.target.get_monster_name())
                if isinstance(action, AttackAction):
-                    out_str += "{} {} {} for {}.\n".format(action.attacker.monster_type, action.message, action.target.monster_type, int(action.damage))
+                    out_str += "{} {} {} for {}.\n".format(action.attacker.get_monster_name(), action.message, action.target.get_monster_name(), int(action.damage))
                if isinstance(action, DefeatAction):
-                    out_str += "\n\n**{}** {}. {} gains {} xp.\n\n".format(action.target.monster_type, action.message, action.attacker.monster_type, action.xp)
+                    out_str += "\n\n**{}** {}. {} gains {} xp.\n\n".format(action.target.get_monster_name(), action.message, action.attacker.get_monster_name(), action.xp)
                    fighting = False
                    if action.attacker == monster:
                        result, xp_msg = monster.add_xp(action.xp)
@@ -1266,7 +1294,7 @@ def check_arena():
                        monster2.wins += 1
                        monster.losses += 1
                    break
-            out_str += "\n{} {}%    -    {}  {}%\n".format(monster.monster_type, monster.get_hp_percent(), monster2.monster_type, monster2.get_hp_percent())
+            out_str += "\n{} {}%    -    {}  {}%\n".format(monster.get_monster_name(), monster.get_hp_percent(), monster2.get_monster_name(), monster2.get_hp_percent())
            log(out_str)
            send_message(client, arena_channel, byteify(out_str))
        # Heal the monsters before they are returned to the player inventory
--- a/pankration.py
View file @226fb77
+++ b/pankration.py
View file @226fb77