from csv import reader
from csv import DictReader
from sys import argv
import csv
from cs50 import SQL
import cs50

# Give access to students.db

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

def main():

    if len (argv) != 2:
        print("Error: Argv Length")
        exit(1)

    with open(argv[1],"r") as chars:
        reader = csv.DictReader(chars)

        id_count = 0
        for row in reader:
            splitname = row["name"].split()
            if len(splitname) == 3:
                first = splitname[0]
                middle = splitname[1]
                last = splitname[2]
            if len(splitname) ==2:
                first = splitname[0]
                middle = None
                last = splitname[1]

            house=row["house"]
            birth=int(row["birth"])

            id_count += 1

            print(id_count, first, middle, last, house, birth)

            # empty file ready to recieve data

            open("students.db", "w").close()
            db=cs50.SQL("sqlite:///students.db")
            db.execute("INSERT INTO students (first, middle, last, house, birth) values (?,?,?,?,?)", first, middle, last, house, birth)

main()