import sys
from cs50 import SQL
import csv


if len(sys.argv) == 2 and sys.argv[1] == "characters.csv":

    db = SQL("sqlite:///students.db")
    with open(sys.argv[1]) as csv_file:
        csv_reader = csv.DictReader(csv_file, delimiter = ',')
        next(csv_reader)
        for row in csv_reader:
            name = row['name'].split()
            house = row['house']
            birth = row['birth']

            if len(name) == 3:
                first = name[0]
                middle = name[1]
                last = name[2]
            else:
                first = name[0]
                middle = None
                last = name[1]

            sql = "INSERT INTO students (first, middle, last, house, birth) VALUES (?, ?, ?, ?, ?)"
            val = (first, middle, last, house, birth)
            db.execute(sql, val)
else:
    sys.exit("Error")