From b964003d74fe67d10c5d019e11684158216269db Mon Sep 17 00:00:00 2001 From: James Miller Date: Mon, 11 Nov 2019 16:31:21 -0600 Subject: [PATCH] got alignment working --- data/yaml/basics.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/data/yaml/basics.py b/data/yaml/basics.py index 84da463..29bf222 100644 --- a/data/yaml/basics.py +++ b/data/yaml/basics.py @@ -18,8 +18,34 @@ def main(): # call the functions to input to SQL do_abilityscore(data['abilityscore'], conn) do_actioncost(data['actioncost'], conn) + do_alignment(data['alignment'], conn) do_langrarity(data['lang_rarity'], conn) +def do_alignment(data, conn): + print(data) + table = """ +CREATE TABLE alignment ( + alignment_id INTEGER PRIMARY KEY, + "name" TEXT UNIQUE NOT NULL, -- 'Lawful Good' + abbr TEXT UNIQUE NOT NULL -- 'LG' +); + """ + + c = conn.cursor() + c.execute(table) + + inp_data = [] + for i in data: + inp_data.append((i['name'], i['abbr'])) + + stmt = "INSERT INTO alignment(name, abbr) VALUES (?,?)" + try: + conn.executemany(stmt,inp_data) + except: + print("Error creating alignment") + else: + conn.commit() + def do_langrarity(data, conn): table = """ CREATE TABLE langrarity (