From c8bde8dad613ab6a3306725f5cce448d63b99ada Mon Sep 17 00:00:00 2001
From: James Miller <james.ryland.miller@gmail.com>
Date: Mon, 11 Nov 2019 16:39:50 -0600
Subject: [PATCH] got weaponcategory working

---
 data/yaml/basics.py   | 26 ++++++++++++++++++++++++++
 data/yaml/basics.yaml |  2 +-
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/data/yaml/basics.py b/data/yaml/basics.py
index d13dbdf..24aee8b 100644
--- a/data/yaml/basics.py
+++ b/data/yaml/basics.py
@@ -22,6 +22,32 @@ def main():
    do_frequency(data['frequency'], conn)
    do_langrarity(data['lang_rarity'], conn)
    do_movement(data['movement'], conn)
+   do_weaponcategory(data['weaponcategory'], conn)
+
+def do_weaponcategory(data, conn):
+   table = """
+CREATE TABLE weaponcategory (
+  weaponcategory_id INTEGER PRIMARY KEY,
+  "name" TEXT NOT NULL UNIQUE
+);
+   """
+
+   c = conn.cursor()
+   c.execute(table)
+
+   inp_data = []
+   for i in data:
+      inp_data.append((i,)) # trailing comma necessary for one-item tuple
+
+   stmt = "INSERT INTO weaponcategory(name) VALUES (?)"
+   try:
+      conn.executemany(stmt,inp_data)
+   except:
+      e = sys.exc_info()[0]
+      print("Error creating weaponcategory: {}".format(e))
+      print(vars(e))
+   else:
+      conn.commit()
 
 def do_movement(data, conn):
    table = """
diff --git a/data/yaml/basics.yaml b/data/yaml/basics.yaml
index 4e7c5eb..2e9f1c1 100644
--- a/data/yaml/basics.yaml
+++ b/data/yaml/basics.yaml
@@ -98,7 +98,7 @@ size:
     space_in_ft: 20
     reach_tall_ft: 20
     reach_long_Ft: 15
-weapon_category:
+weaponcategory:
   - Unarmed
   - Simple
   - Martial