From 6476f4de41d3697b38a7fef78ede28605685c06e Mon Sep 17 00:00:00 2001
From: James Miller <james.ryland.miller@gmail.com>
Date: Mon, 11 Nov 2019 16:28:05 -0600
Subject: [PATCH] got lang rarity populating

---
 data/yaml/basics.py | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/data/yaml/basics.py b/data/yaml/basics.py
index b832591..84da463 100644
--- a/data/yaml/basics.py
+++ b/data/yaml/basics.py
@@ -2,6 +2,7 @@ import yaml
 import sqlite3
 import os
 import pprint
+import sys
 
 DBFILE = 'tmp.db'
 
@@ -17,10 +18,34 @@ def main():
    # call the functions to input to SQL
    do_abilityscore(data['abilityscore'], conn)
    do_actioncost(data['actioncost'], conn)
+   do_langrarity(data['lang_rarity'], conn)
+
+def do_langrarity(data, conn):
+   table = """
+CREATE TABLE langrarity (
+  rarity_id INTEGER PRIMARY KEY,
+  rarity_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 langrarity(rarity_name) VALUES (?)"
+   try:
+      conn.executemany(stmt,inp_data)
+   except:
+      e = sys.exc_info()[0]
+      print("Error creating langrarity: {}".format(e))
+      print(vars(e))
+   else:
+      conn.commit()
 
 def do_actioncost(data, conn):
-   print(data)
-
    table = """
 CREATE TABLE actioncost (
   actioncost_id INTEGER PRIMARY KEY,
@@ -45,8 +70,6 @@ CREATE TABLE actioncost (
       conn.commit()
 
 def do_abilityscore(data, conn):
-   print(data)
-
    table = """
 CREATE TABLE abilityscore (
    abilityscore_id INTEGER PRIMARY KEY,