Added some interfaces to the pankration and db

@@ -2,6 +2,7 @@ import operator
 import sqlite3
 import sqlite3
 import datetime
 import datetime
 import logging
 import logging
+import pickle
 
 
 
 
 def byteify(input):
 def byteify(input):
@@ -236,6 +237,36 @@ def db_get_all_members():
         members_list.append(dict_factory(c, member))
         members_list.append(dict_factory(c, member))
     return members_list
     return members_list
 
 
+def db_get_pankration_record(member_id):
+    # Do a lookup by ID, if it's found but the name doesn't match then add a row to aliases with the previous name and change the member name
+    pan_record = sqlite3.connect('db.sqlite3')
+
+    c = pan_record.cursor()
+    result = None
+    result = c.execute("SELECT * FROM pankration WHERE member_id = ?;", (member_id,)).fetchone()
+    pan_record.close()
+    return dict_factory(c, result)
+
+def db_add_monster(member_id, monster):
+    pan_record = db_get_pankration_record(member_id)
+    log("{}".format(member_id))
+    conn = sqlite3.connect('db.sqlite3')
+    c = conn.cursor()
+    db_state = c.execute("SELECT monster_inventory FROM pankration WHERE member_id = ?;", (member_id,)).fetchone()
+    if not db_state:
+        c.execute("""INSERT INTO pankration(member_id, wins, losses, monster_inventory) 
+                 VALUES(?, ?, ?, ?);""", (member_id, 0, 0, pickle.dumps([monster])))
+        conn.commit()
+    else:
+        log(db_state)
+        inv = str(db_state[0])
+        log(inv)
+        monster_inventory = pickle.loads(inv)
+        monster_inventory.append(monster)
+        c.execute("""UPDATE pankration SET monster_inventory = ? 
+                 WHERE member_id = ?;""", (pickle.dumps(monster_inventory), member_id))
+        conn.commit()
+
 def db_get_member(discord_id=None, username=None):
 def db_get_member(discord_id=None, username=None):
     # Do a lookup by ID, if it's found but the name doesn't match then add a row to aliases with the previous name and change the member name
     # Do a lookup by ID, if it's found but the name doesn't match then add a row to aliases with the previous name and change the member name
     member_conn = sqlite3.connect('db.sqlite3')
     member_conn = sqlite3.connect('db.sqlite3')

@@ -85,6 +85,7 @@ registered_commands = {'!help': 'do_help', '!commands': 'do_help',
                        '!rigged': 'do_rigged',
                        '!rigged': 'do_rigged',
                        '!listzones': 'do_list_zones',
                        '!listzones': 'do_list_zones',
                        '!huntmonster': 'do_hunt_monster',
                        '!huntmonster': 'do_hunt_monster',
+                       '!listmonster': 'do_list_monsters',
                        }
                        }
 
 
 
 
@@ -1014,12 +1015,24 @@ def do_rigged(client, message_parts, message):
     send_message(client, message.channel, ":musical_note: {} :musical_note:".format(random.choice(lines)))
     send_message(client, message.channel, ":musical_note: {} :musical_note:".format(random.choice(lines)))
     return
     return
 
 
+def do_list_monsters(client, message_parts, message):
+    member = data.db_get_member(message.author.id)
+    log(member)
+    monster_data = data.db_get_pankration_record(member['member_id'])
+    if monster_data:
+        monsters = pickle.loads(str(monster_data['monster_inventory']))
+        send_message(client, message.channel, "\n\n".join(str(monster) for monster in monsters))
+    else:
+        send_message(client, message.channel, 'You have no monsters.')
+
 def do_hunt_monster(client, message_parts, message):
 def do_hunt_monster(client, message_parts, message):
     p = Pankration()
     p = Pankration()
     hunt_response = p.hunt_monster(' '.join(message_parts))
     hunt_response = p.hunt_monster(' '.join(message_parts))
     str_out = hunt_response.message + "\n\n"
     str_out = hunt_response.message + "\n\n"
     if hunt_response.result == HuntResponse.SUCCESS:
     if hunt_response.result == HuntResponse.SUCCESS:
         monster = hunt_response.monster
         monster = hunt_response.monster
+        member = data.db_get_member(message.author.id)
+        data.db_add_monster(member['member_id'], monster)
         str_out += str(monster)
         str_out += str(monster)
     send_message(client, message.channel, str_out)
     send_message(client, message.channel, str_out)