#import.py

import cs50
import csv
from sys import argv


studb = cs50.SQL("sqlite:///students.db")

with open("characters.csv", "r") as students:
    reader = csv.DictReader(students, delimiter = ",")
    nmesplt = ""
    frstnme = ""
    mdlnme = ""
    lastnme = ""
    for row in reader:
        index = 0
        nmesplt = row['name'].split(" ")

        if len(nmesplt) == 3:
            frstnme = nmesplt[0]
            mdlenme = nmesplt[1]
            lastnme = nmesplt[2]

            studb.execute("INSERT INTO students(first, middle, last, house, birth) VALUES(?, ?, ?, ?, ?)", frstnme, mdlenme, lastnme, row['house'], row['birth'])
        if len(nmesplt) == 2:
            frstnme = nmesplt[0]
            mdlenme = ''
            lastnme = nmesplt[1]


            studb.execute("INSERT INTO students(first, middle, last, house, birth) VALUES(?, ?, ?, ?, ?)", frstnme, mdlenme, lastnme, row['house'], int(row['birth']))

        index += 1