From 893271b04a9322186a50435913bd43f8ab9179e9 Mon Sep 17 00:00:00 2001
From: James Miller <james.ryland.miller@gmail.com>
Date: Fri, 21 Feb 2020 16:15:54 -0600
Subject: [PATCH] working on armor

---
 data/yaml/gendb.py | 58 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)

diff --git a/data/yaml/gendb.py b/data/yaml/gendb.py
index 2b13823..5c522fe 100644
--- a/data/yaml/gendb.py
+++ b/data/yaml/gendb.py
@@ -92,6 +92,64 @@ def main():
         data = yaml.full_load(yl)
     do_triggers(data, conn)
 
+    # move on to armor
+    with open('armor.yaml') as yl:
+        data = yaml.full_load(yl)
+    do_armor(data, conn)
+
+def do_armor(data, conn):
+    # Create the 3 tables
+    table = """ 
+    CREATE TABLE armorcategory (
+        armor_category_id INTEGER PRIMARY KEY,
+        short_name TEXT NOT NULL UNIQUE);
+    """
+    c = conn.cursor()
+    c.execute(table)
+
+    table = """
+    CREATE TABLE armorgroup (
+        grp_id INTEGER PRIMARY KEY,
+        short_name TEXT NOT NULL UNIQUE,
+        descr TEXT NOT NULL
+    );
+    """
+    c.execute(table)
+
+    table = """
+    CREATE TABLE armor (
+        armor_id INTEGER PRIMARY KEY,
+        armor_category_id INTEGER NOT NULL,
+        short_name TEXT NOT NULL,
+        item_level INTEGER,
+        price_text TEXT NOT NULL,
+        price_gp FLOAT NOT NULL,
+        ac_bonus INTEGER NOT NULL,
+        dex_cap INTEGER,
+        check_penalty INTEGER,
+        speed_penalty INTEGER,
+        strength INTEGER,
+        bulk_id INTEGER NOT NULL,
+        grp_id INTEGER,
+        descr TEXT NOT NULL,
+	FOREIGN KEY (bulk_id) REFERENCES bulks(bulk_id),
+	FOREIGN KEY (grp_id) REFERENCES armorgroup(grp_id)
+    );
+    """
+    c.execute(table)
+    table = """
+    CREATE TABLE trait_armor (
+        id INTEGER PRIMARY KEY,
+        trait_id INTEGER NOT NULL,
+        armor_id INTEGER NOT NULL,
+    FOREIGN KEY (trait_id) REFERENCES traits(trait_id),
+    FOREIGN KEY (armor_id) REFERENCES armor(armor_id)
+    );
+    """
+    c.execute(table)
+
+    pass
+
 def do_triggers(data, conn):
     table = """
 CREATE TABLE trigger (