From 10314042807fca44c46ad993a23e8367042e6e40 Mon Sep 17 00:00:00 2001
From: James Miller <james.ryland.miller@gmail.com>
Date: Tue, 12 Nov 2019 22:26:40 -0600
Subject: [PATCH] got backgrounds working YAML->SQL

---
 data/yaml/backgrounds.yaml |  1 +
 data/yaml/gendb.py         | 88 ++++++++++++++++++++++++++++++++++++--
 2 files changed, 85 insertions(+), 4 deletions(-)

diff --git a/data/yaml/backgrounds.yaml b/data/yaml/backgrounds.yaml
index 7437d24..0c2ce81 100644
--- a/data/yaml/backgrounds.yaml
+++ b/data/yaml/backgrounds.yaml
@@ -1,3 +1,4 @@
+background:
 - descr: You spent your early days in a religious monastery or cloister. You may have
     traveled out into the world to spread the message of your religion or because
     you cast away the teachings of your faith, but deep down you’ll always carry within
diff --git a/data/yaml/gendb.py b/data/yaml/gendb.py
index 1d7ac1d..1d99a2f 100644
--- a/data/yaml/gendb.py
+++ b/data/yaml/gendb.py
@@ -51,6 +51,84 @@ def main():
         data = yaml.full_load(yl)
     do_conditions(data, conn)
 
+    # move on to backgrounds
+    with open('backgrounds.yaml') as yl:
+        data = yaml.full_load(yl)
+    do_backgrounds(data, conn)
+
+
+def do_backgrounds(data, conn):
+    # MAKE THE 2 TABLES
+    table = """
+CREATE TABLE background (
+  background_id INTEGER PRIMARY KEY,
+  "name" TEXT NOT NULL UNIQUE,
+  descr TEXT NOT NULL,
+  is_comty_use BOOLEAN NOT NULL, -- false = no community use policy required
+  is_specific_to_adv BOOLEAN NOT NULL -- means the background is specific to its adventure
+);
+   """
+    c = conn.cursor()
+    c.execute(table)
+    table = """
+CREATE TABLE sourceentry_background (
+  id INTEGER PRIMARY KEY,
+  sourceentry_id INTEGER NOT NULL,
+  background_id INTEGER NOT NULL,
+  UNIQUE (sourceentry_id, background_id), -- prevent duplicates
+  FOREIGN KEY (sourceentry_id) REFERENCES sourceentry(sourceentry_id),
+  FOREIGN KEY (background_id) REFERENCES background(background_id)
+);
+   """
+    c.execute(table)
+
+    print(data)
+    for i in data['background']:
+        srcentrydata = []
+        for j in i['source']:
+            abbr = j['abbr']
+            page_start = j['page_start']
+            # Not all YAML entries have page_stop data
+            if 'page_stop' in j:
+                page_stop = j['page_stop']
+            else:
+                page_stop = page_start
+            srcentrydata.append((abbr, page_start, page_stop))
+        stmt = "INSERT INTO background(name, descr, is_comty_use, is_specific_to_adv) VALUES (?,?,?,?)"
+        try:
+            conn.execute(stmt, (i['name'], i['descr'], i['is_comty_use'], i['is_specific_to_adv']))
+        except:
+            print("Error creating background")
+        else:
+            conn.commit()
+        # print("backgrounds, about to call util insert se: {}".format(srcentrydata))
+        util_insert_into_sourceentry(srcentrydata, conn)
+        # now link the source entries to this table
+        # TODO
+        link_sourceentry_backgrounds(i['name'], srcentrydata, conn)
+
+def link_sourceentry_backgrounds(name, srcentrydata, conn):
+    stmt = """
+INSERT INTO sourceentry_background (sourceentry_id, background_id)
+    SELECT sourceentry_id, background_id
+    FROM sourceentry, background
+    WHERE sourceentry.source_id=(SELECT source_id FROM source WHERE abbr=?)
+    AND sourceentry.page_start=?
+    AND sourceentry.page_stop=?
+    AND background.name=?;
+    """
+    # print(srcentrydata)
+    for i in srcentrydata:
+        # print("i is:{}".format(i))
+        d = (i[0], i[1], i[2], name)
+        # print(d)
+        try:
+            conn.execute(stmt, d)
+        except Exception as e:
+            print("Error linking sourceentry to backgrounds: {}".format(e))
+        else:
+            conn.commit()
+
 def do_conditions(data, conn):
     # MAKE THE 2 TABLES
     table = """
@@ -93,11 +171,12 @@ CREATE TABLE sourceentry_condition (
             print("Error creating condition")
         else:
             conn.commit()
-        print("conditions, about to call util insert se: {}".format(srcentrydata))
+        # print("conditions, about to call util insert se: {}".format(srcentrydata))
         util_insert_into_sourceentry(srcentrydata, conn)
-        # now link the source entries to this table 
+        # now link the source entries to this table
         link_sourceentry_conditions(i['name'], srcentrydata, conn)
 
+
 def link_sourceentry_conditions(name, srcentrydata, conn):
     stmt = """
 INSERT INTO sourceentry_condition (sourceentry_id, condition_id)
@@ -110,9 +189,9 @@ INSERT INTO sourceentry_condition (sourceentry_id, condition_id)
     """
     # print(srcentrydata)
     for i in srcentrydata:
-        print("i is:{}".format(i))
+        # print("i is:{}".format(i))
         d = (i[0], i[1], i[2], name)
-        print(d)
+        # print(d)
         try:
             conn.execute(stmt, d)
         except Exception as e:
@@ -120,6 +199,7 @@ INSERT INTO sourceentry_condition (sourceentry_id, condition_id)
         else:
             conn.commit()
 
+
 def do_damage(data, conn):
     # make the four tables
     do_damage_sub_tables(data, conn)