got lang rarity populating

source_entries
James Miller 2019-11-11 16:28:05 -06:00
parent 6ed2eea571
commit 6476f4de41
1 changed files with 27 additions and 4 deletions

View File

@ -2,6 +2,7 @@ import yaml
import sqlite3 import sqlite3
import os import os
import pprint import pprint
import sys
DBFILE = 'tmp.db' DBFILE = 'tmp.db'
@ -17,10 +18,34 @@ def main():
# call the functions to input to SQL # call the functions to input to SQL
do_abilityscore(data['abilityscore'], conn) do_abilityscore(data['abilityscore'], conn)
do_actioncost(data['actioncost'], 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): def do_actioncost(data, conn):
print(data)
table = """ table = """
CREATE TABLE actioncost ( CREATE TABLE actioncost (
actioncost_id INTEGER PRIMARY KEY, actioncost_id INTEGER PRIMARY KEY,
@ -45,8 +70,6 @@ CREATE TABLE actioncost (
conn.commit() conn.commit()
def do_abilityscore(data, conn): def do_abilityscore(data, conn):
print(data)
table = """ table = """
CREATE TABLE abilityscore ( CREATE TABLE abilityscore (
abilityscore_id INTEGER PRIMARY KEY, abilityscore_id INTEGER PRIMARY KEY,