From 0bb51ed2410ceebc71185f8c3bb3a1d74c22128a Mon Sep 17 00:00:00 2001
From: James Miller <james.ryland.miller@gmail.com>
Date: Mon, 24 Feb 2020 22:26:59 -0600
Subject: [PATCH] fixes #77; gear new yaml to new sql in gendb.py
---
data/yaml/gendb.py | 121 +++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 121 insertions(+)
diff --git a/data/yaml/gendb.py b/data/yaml/gendb.py
index e828dbc..1ed5597 100644
--- a/data/yaml/gendb.py
+++ b/data/yaml/gendb.py
@@ -105,6 +105,127 @@ def main():
data = yaml.full_load(yl)
do_ammo(data, conn)
+ # move on to ammo
+ with open('gear.yaml') as yl:
+ data = yaml.full_load(yl)
+ do_gear(data, conn)
+
+def do_gear(data, conn):
+ table = """
+ CREATE TABLE gear(
+ "gear_id" INTEGER PRIMARY KEY,
+ "name" TEXT NOT NULL UNIQUE,
+ "level" INTEGER,
+ "price_gp" REAL,
+ "bulk" REAL,
+ "hands" INTEGER,
+ "descr" TEXT
+ );
+ """
+ c = conn.cursor()
+ c.execute(table)
+ table = """
+ CREATE TABLE gear_traits(
+ id INTEGER PRIMARY KEY,
+ gear_id INTEGER NOT NULL,
+ trait_id INTEGER NOT NULL,
+ FOREIGN KEY (gear_id) REFERENCES gear(gear_id),
+ FOREIGN KEY (trait_id) REFERENCES trait(trait_id)
+ );
+ """
+ c.execute(table)
+ table = """
+ CREATE TABLE sourceentry_gear (
+ id INTEGER PRIMARY KEY,
+ sourceentry_id INTEGER NOT NULL,
+ gear_id INTEGER NOT NULL,
+ UNIQUE (sourceentry_id, gear_id), -- prevent duplicates
+ FOREIGN KEY (sourceentry_id) REFERENCES sourceentry(sourceentry_id),
+ FOREIGN KEY (gear_id) REFERENCES gear(gear_id)
+ );
+ """
+ c.execute(table)
+
+ # insert basics into gear table
+ inp_data = []
+ for i in data['gear']:
+ # print(i)
+ inp_data.append((i['bulk'], i['descr'], i['hands'], i['level'], i['name'], i['price_gp']))
+
+ stmt = "INSERT INTO gear(bulk, descr, hands, level, name, price_gp) VALUES (?,?,?,?,?,?)"
+ try:
+ conn.executemany(stmt, inp_data)
+ except sqlite3.Error as e:
+ print("Error creating gear: {}".format(e))
+ except:
+ print("Error creating gear something other than sqlite3 error")
+ else:
+ conn.commit()
+
+ # link the traits to the gear_traits table
+ traitslist = []
+ for i in data['gear']:
+ if i['traits'] == None:
+ continue
+ else:
+ for j in i['traits']:
+ traitslist.append((i['name'], j))
+ # print(traitslist)
+ stmt = """
+ INSERT INTO gear_traits(gear_id, trait_id)
+ VALUES (
+ (SELECT gear_id FROM gear WHERE name=?),
+ (SELECT trait_id FROM trait WHERE short_name=?)
+ );
+ """
+ try:
+ conn.executemany(stmt, traitslist)
+ except sqlite3.Error as e:
+ print("Error creating gear_traits data: {}".format(e))
+ except:
+ print("Error creating gear_traits data something other than sqlite3 error")
+ else:
+ conn.commit()
+
+
+
+ # do the sourceentry linking
+ for i in data['gear']:
+ 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))
+ util_insert_into_sourceentry(srcentrydata, conn)
+ link_sourceentry_gear(i['name'], srcentrydata, conn)
+
+def link_sourceentry_gear(name, srcentrydata, conn):
+ stmt = """
+ INSERT INTO sourceentry_gear (sourceentry_id, gear_id)
+ SELECT sourceentry_id, gear_id
+ FROM sourceentry, gear
+ WHERE sourceentry.source_id=(SELECT source_id FROM source WHERE abbr=?)
+ AND sourceentry.page_start=?
+ AND sourceentry.page_stop=?
+ AND gear.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 gear: {}".format(e))
+ else:
+ conn.commit()
+
def do_ammo(data, conn):
table = """
CREATE TABLE ammunition (