importing stuff from csv to db

characters.csv

import.py

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

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

if len(argv) < 2:
    print("Usage error.use: import.py characters.csv")
    exit()

with open(argv[1], newline='') as charactersFile:
    characters = reader(charactersFile)
    for character in characters:
        if character[0] == 'name':
            continue

        # split the full name into first, last and middle name if present
        fullName = character[0].split()
        # insert the name and the other infos in the database, the database will show NULL if the student has no middle name
        if len(fullName) < 3:
            db.execute("INSERT INTO students(first, middle, last, house, birth) VALUES(?, ?, ?, ?, ?)",
                       fullName[0], None, fullName[1], character[1], character[2])

        else:
            db.execute("INSERT INTO students(first, middle, last, house, birth) VALUES(?, ?, ?, ?, ?)",
                       fullName[0], fullName[1], fullName[2], character[1], character[2])